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!')