From e8b395faed536546f106e58a6948e1ad0ed0e881 Mon Sep 17 00:00:00 2001 From: Bob The Burger <yofihzilif@sch.bme.hu> Date: Fri, 3 Apr 2020 13:38:07 -0700 Subject: [PATCH] Created Create DB and Save DB --- DB/Database.py | 24 +++++++++++++++++++++++- program.py | 9 +++++++-- 2 files changed, 30 insertions(+), 3 deletions(-) diff --git a/DB/Database.py b/DB/Database.py index 8e3e3c4..23a246e 100644 --- a/DB/Database.py +++ b/DB/Database.py @@ -1,11 +1,13 @@ """ -ez file lenne ami ossyekoti a db-t a programunkal. hogy egyszerubb legyen +ez file lenne ami osszekoti a db-t a programunkal. hogy egyszerubb legyen az eletunk, ezert most csak egyszeru dictekbe, meg setekbe tarolunk mindent """ # import pprint fake_db = {} +# ez garantalja hogy egy username csak egyszer lehet benne fake_db['u_names'] = set() +# ezek objektumok nev-jelszo paros listaja lesz, tehat csak sima array fake_db['users'] = [] @@ -52,3 +54,23 @@ def login_user(username, password): return 'Succes', username else: return 'Nein' + + +def save_db(): + with open('db.txt', 'w') as f: + for user in fake_db['users']: + f.write('username {} - password {}'.format(user['Username'], + user['Password'])) + return 'Succes' + + +def valid_db(): + if len(fake_db['users']) != len(set([u['Username'] for u in fake_db['users']])): + return ['Error', len(fake_db['users']) != len(set([u['Username'] for u in fake_db['users']]))] + elif set([u['Username'] for u in fake_db['users']]) - fake_db['u_names'] == set(): + return 'Succes' + return ['error', set([u['Username'] for u in fake_db['users']]) - fake_db['u_names']] + + +if __name__ != "__main__": + print(__name__) diff --git a/program.py b/program.py index 0e7a306..866ff59 100644 --- a/program.py +++ b/program.py @@ -18,7 +18,7 @@ def login_user(username=None, password=None): if __name__ == "__main__": while 1: text_input = input("Press enter to start registration, anything else \ - + enter if login") + + enter if login") if text_input == '': name = input("Give us a Username ") psw = input("Give us a Password ") @@ -29,8 +29,9 @@ if __name__ == "__main__": name = input("Give us a Username ") psw = input("Give us a Password ") state = Register(name, psw) + db.save_db() # print('Noice') - else: + elif text_input != '0': name = input("Give us a Username ") psw = input("Give us a Password ") state = login_user(name, psw) @@ -40,4 +41,8 @@ if __name__ == "__main__": name = input("Give us a Username ") psw = input("Give us a Password ") state = login_user(name, psw) + else: + print('Db validation: ', db.valid_db()) + db.save_db() + exit() # print('Noice: ', state[1]) -- GitLab