from func import register, login
def main():
cmd = input("Enter command (r - register, l - login): ")
if cmd == 'r':
username = input("Enter username: ")
password = input("Enter password: ")
ans = register(login=username, password=password)
if ans == True:
print('Registration successful')
else:
print('Invalid data')
elif cmd == 'l':
username = input("Enter username: ")
password = input("Enter password: ")
ans = login(username, password)
if ans == True:
print('Login successful')
else:
print('Invalid data')
if __name__ == '__main__':
main()
Объяснение: