diff --git a/DB/Database.py b/DB/Database.py index 8e3e3c452584ad467ddbd2419f64a4bbd5cacc52..23a246e0de69227d50057b6b00ce2a033f0c78e3 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 0e7a306a204a285e964b5003a496a6d4969a0e26..866ff59ef7da17def8518d892ebca4f3dfac9920 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])