import sqlite3

def register():
    with conn:
        cur = conn.cursor()
        try:
            EmployeeID = int(input("Enter New Employee ID: "))
            #checkExisting = cur.execute(
            #    'select EmployeeID from Employee table where EmployeeID = '
            #)
            #if EmployeeID ==

            FirstName = input("What's your first name:")
            FirstName = FirstName.title()


            LastName = input("What's your last name: ")
            LastName = LastName.title()

            Email = input("Email: ")
            Email = Email.lower()

            Password = input("Enter a password: ")
            Password = Password.lower()

            cur.execute(
                'insert into employee values(?,?,?,?,?)',
                (EmployeeID, FirstName, LastName, Email, Password))
            cur.execute(
                'select * from employee where EmployeeId=?',
                (EmployeeID,))

            results = cur.fetchall()
            print(results)
        except sqlite3.IntegrityError:
            print("Connection Failed")


def login():
    with conn:
        cur = conn.cursor()
        try:
            loginTest = False  # main condition to loop if email and password not met
            while not loginTest:  # wrong email loopy
                userEmail = input("Email please: ")
                userEmail = userEmail.lower()
                userEmail = userEmail.replace(" ", "")
                userPassword = input("Password: ")
                userPassword = userPassword.strip()
                cur.execute(
                    "SELECT COUNT (*) FROM Employee WHERE(Email= '" + userEmail.lower() + "' AND Password= '" + userPassword + "')")
                results = cur.fetchone()  # return very first thing it finds that matches
                print(results[0])  # print first thing
                if results[0] == 1:
                    print("Login successful")
                    loginTest = True
                else:
                    print("Login Unsucessful")
                    existingUser = input("Existing user?[yes/no]")
                    if existingUser == "no":
                        register()
        except:
            print("connection failed")


conn = sqlite3.connect('OS_employee.db')
with conn:
    cur = conn.cursor()
    print("successfully connected")

    existingUser = input("Existing user?[yes/no]")
    existingUser = existingUser.lower()
    while existingUser != "yes" or existingUser != "no":
        if existingUser == "no":
            register()
            break
        elif existingUser == "yes":
            login()
            break
        else:
            input("Invalid input. Please answer [yes/no]")
            existingUser = input("Existing user?[yes/no]")