diff --git a/app.py b/app.py index 5099399..0e63cdc 100644 --- a/app.py +++ b/app.py @@ -16,7 +16,9 @@ def login(): username = rdata['username'] password = rdata['password'] if dao.valid(username, password): - return make_response(json.dumps({'result': 'success'})).set_cookie('jwt', jwt.generate_jwt(username)) + response = make_response(json.dumps({'result': 'success'})) + response.set_cookie('jwt', jwt.generate_jwt(username)) + return response else: return make_response(json.dumps({'result': 'fail'})) diff --git a/dao.py b/dao.py index bb9593b..ff918fd 100644 --- a/dao.py +++ b/dao.py @@ -6,7 +6,7 @@ def valid(username, password): base_cursor = db_connection.cursor() password_cursor = base_cursor.execute(f'SELECT password FROM user WHERE username=="{username}";') db_password = password_cursor.fetchone() - if len(db_password != 1): + if db_password is None: return False if db_password[0] == password: return True