From 65a910ff90687c9a99c8211d9eac128502e1c672 Mon Sep 17 00:00:00 2001 From: Tiziano Zito Date: Mon, 26 Aug 2024 12:08:10 +0300 Subject: [PATCH] implement name-main trick --- auth.py | 27 +++++++++++++++------------ 1 file changed, 15 insertions(+), 12 deletions(-) diff --git a/auth.py b/auth.py index 67064e5..9c68472 100644 --- a/auth.py +++ b/auth.py @@ -25,18 +25,21 @@ def write_pwdb(pwdb, pwdb_path): pwdb_file = open(pwdb_path, 'wt') json.dump(pwdb, pwdb_file) -pwdb_path = 'pwdb.json' -pwdb = read_pwdb(pwdb_path) -if len(sys.argv) > 1: - pwdb = add_user(sys.argv[1], pwdb) - write_pwdb(pwdb, pwdb_path) -else: - username, password = get_credentials() - if username not in pwdb: - print('Wrong username!') +if __name__ == "__main__": + pwdb_path = 'pwdb.json' + pwdb = read_pwdb(pwdb_path) + + + if len(sys.argv) > 1: + pwdb = add_user(sys.argv[1], pwdb) + write_pwdb(pwdb, pwdb_path) else: - if authenticate(username, password, pwdb): - print('Successfully authenticated!') + username, password = get_credentials() + if username not in pwdb: + print('Wrong username!') else: - print('Wrong password!') + if authenticate(username, password, pwdb): + print('Successfully authenticated!') + else: + print('Wrong password!')