interface.py

"""
API Interface to the modules in this project. To be used by the UI application.
"""

import os

import authenticate_user as authenticate
import common
import register_user as register


def authenticate_user(base_dir: str = os.getcwd()) -> str:
    """
    Interface for authenticating a user with the given base directory location.
    :param base_dir: The base directory, from which all default paths are relative. Defaults to current working directory.
    :return: The user_id of the recognized user, or empty string if none was recognized.
    """
    recognized = authenticate.recognize_user_from_database(
        database_loc=os.path.join(base_dir, common.DATABASE_LOC))
    return recognized if recognized is not None else ""


def register_user(base_dir: str = os.getcwd()) -> None:
    """
    Registers new added users in the database.
    :param base_dir: The base directory, from which all default paths are relative. Defaults to current working directory.
    :return: None
    """
    register.register_users_from_dir_and_save(directory_location=os.path.join(base_dir, common.DATABASE_LOC),
                                              delete_images_on_complete=True, overwrite_data=False)


if __name__ == "__main__":
    import argparse, sys

    parser = argparse.ArgumentParser(description="API Interface Options")
    parser.add_argument("--base-dir", type=str, required=False, default="./",
                        help="The directory from which the save locations should be relative.")
    parser.add_argument("--authenticate", required=False, default=False, action="store_true",
                        help="Present this argument to authenticate a user.")
    parser.add_argument("--register", required=False, default=False, action="store_true",
                        help="Present this argument to register a user.")

    if len(sys.argv) <= 1:
        args = parser.parse_args(input().split(" "))
    else:
        args = parser.parse_args()

    kwargs = dict(base_dir=args.base_dir)
    if args.register:
        register_user(**{arg_name: arg for (arg_name, arg)
                         in kwargs.items() if args is not None})
    elif args.authenticate:
        user_id: str = authenticate_user(
            **{arg_name: arg for (arg_name, arg) in kwargs.items() if arg is not None})
        if user_id:
            print(user_id)