auth.py

import utils.input
import users
import actions


session = None


def register(email: str = None):
    global session

    name = (
        utils.input.prompt(
            "First Name: ", lambda i: i.isalpha() and len(i) < 16,
            "Please use alphabetical decent name."
        ),
        utils.input.prompt(
            "Last Name: ", lambda i: i.isalpha() and len(i) < 16,
            "Please use alphabetical decent name."
        )
    )
    if email is None:
        email = utils.input.prompt_email()
    phone = utils.input.prompt(
        "Mobile Phone (+20): ", r'^01[0125][0-9]{8}$',
        "Please use an Egyptian mobile phone number (ex. 015xxxxxxxx)."
    )
    password = utils.input.prompt(
        "Password: ", r'^[^\f\n]{8,}$',
        "The password must be 8 characters."
    )

    # TODO: review behaviour, may be return to the menu instead?
    while input("Confirm Password: ") != password:
        print("Passwords don't match.")

    try:
        users.add(email, password, name, phone)
        session = email
        print("Registered successfully.")
        actions.menu()
    except Exception as exception:
        print(exception)  # TODO: handle better
        menu()


def login():
    global session

    email = utils.input.prompt_email()
    password = users.find_password(email)

    if password is None:
        print("You are not registered, forwarding...")
        register(email)
        return

    attempts = 3
    while attempts > 0:
        if input("Password: ") == password:
            session = email
            print("Logged in successfully.")
            break
        print("Wrong password.")
        attempts -= 1

    if session is not None:
        actions.menu()
    else:
        menu()


def menu():
    print("Authorization:-")
    utils.input.select(('Register', register), ('Login', login))