Browse Source

fix: commit to database and get username through jwt in `/register`

master
kdxcxs 4 years ago
parent
commit
e6d84cbd39
  1. 7
      app.py
  2. 6
      dao.py

7
app.py

@ -26,12 +26,13 @@ def login():
@app.route('/register', methods=['POST'])
def register():
cookie_jwt = request.cookies.get('jwt')
if cookie_jwt is not None:
username = jwt.check_jwt(cookie_jwt)
else:
return make_response(json.dumps({'result': 'fail'}))
rdata = json.loads(request.data.decode()) # request data
username = rdata['username']
program_type = rdata['program_type']
program_name = rdata['program_name']
if cookie_jwt is not None:
if jwt.check_jwt(cookie_jwt):
if dao.register(username, program_type, program_name):
return make_response(json.dumps({'result': 'success'}))
return make_response(json.dumps({'result': 'fail'}))

6
dao.py

@ -7,8 +7,10 @@ def valid(username, password):
password_cursor = base_cursor.execute(f'SELECT password FROM user WHERE username=="{username}";')
db_password = password_cursor.fetchone()
if db_password is None:
db_connection.close()
return False
if db_password[0] == password:
db_connection.close()
return True
@ -16,5 +18,7 @@ def register(username, program_type, program_name):
db_connection = sqlite3.connect('sports-registration.sqlite')
base_cursor = db_connection.cursor()
base_cursor.execute(f'INSERT INTO list (username, program_type, program_name) '
f'VALUES ("{username}", "{program_type}", "{program_name}")')
f'VALUES ("{username}", "{program_type}", "{program_name}");')
db_connection.commit()
db_connection.close()
return True

Loading…
Cancel
Save