ort psycopg2

conn = psycopg2.connect("host=localhost dbname=postgres user=postgres password=postgres")
cur = conn.cursor()


def main():
    login()


def login():
    print("***********************************")
    print("*       ALUGUER DE BICICLETAS     *")
    print("*           by Amazing Lda        *")
    print("***********************************")
    print("[1] Fazer Login")
    print("[2] Criar Conta")
    print("[3] Menu de Administrador")
    print("[0] Sair")
    print("Opção: ")
    option = "1"

    while option != "0":
        option = input()
        if option == "1":
            print("E-mai: ")
            user = input()
            print("Password: ")
            password = input()

            # Condição de que se o utilizador existe e a passe funciona pode ir para a funçao menu

            cur.execute("SELECT count (*) from cliente where mail =%s and password=%s", (user, password))
            if cur.fetchone()[0] == 1:
                menuClient()
            else:
                print("USER NAO ENCONTRADO")
                login()


        elif option == "2":
            print("Digite o e-mail que pretende registar:")
            NewUser = input()
            print("Digite o seu nome:")
            name = input()
            print("Digite a password pretendida:")
            NewPass = input()
            # inserir o novo user e a pass e mandar para a funçao menu
            # como meter o Auto increment???
            # Ta a dar erro nao sei porque
            cur.execute("INSERT into cliente values (DEFAULT,%s, %s, %s, 10);", (name, NewUser, NewPass))

        elif option == "3":  # fazer menu administrador
            print("oli")
        elif option == "0":
            print("PROGRAMA ENCERRADO")


def menuClient():
    option = "1"
    print("            M E N U    C L I E N T E")
    print("")
    print("                Bem Vindo!          ")
    print("")
    print("[1]Ver Bicicletas")  # Aqui dentro precisa de opçao para ver detalhes
    print("[2]Alugar Bicicletas")
    print("[3]Devolver Bicicleta")
    print("[4]Historico de Alugueres")  # aqui meter também quanto gastou em cada tipo de bicicleta
    print("[5]Mensagens")  # distinguir lidas de nao lidas
    print("[6]Consultar Bicicletas")
    print("[0]Sair")
    print("Opção: ")

    while option != "0":

        option = input()
        if option == "1":

            cur.execute("SELECT * from bicicleta")
            # meter a printar em condiçoes
            for linha in cur.fetchall():
                print(linha)
            a = input()
            if a == "0":
                menuClient()

        elif option == "2":
            print("2")
        elif option == "3":
            print("3")

        elif option == "4":
            print("4")


        elif option == "5":
            print("5")


        elif option == "6":
            print("6")
        elif option == "0":
            login()


# Fazer Menu de Admin aqui
# def menuAdmin():


main()