diff --git a/database/main.py b/database/main.py index 9537bff..54c2ada 100644 --- a/database/main.py +++ b/database/main.py @@ -29,6 +29,9 @@ def create_token(): refresh_token = data.get('refresh_token') expiration_seconds = data.get('expiration_seconds') + if not access_token or not refresh_token or not expiration_seconds: + return 'Missing required fields', 400 + new_token = Token(access_token=access_token, refresh_token=refresh_token, expiration_seconds=expiration_seconds) db.session.add(new_token) db.session.commit()