token

api/main.py

@@ -130,11 +130,12 @@ async def make_video_slide(request: Request, response: Response, Authorize: Auth
     return templates.TemplateResponse("make_video_slide.html", {"request": request, "response": response})
 
 @app.post('/user_profile', response_class=HTMLResponse)
-async def user_profile(token_obj: models.jwt_token_only):
+async def user_profile(token: str = Depends(oauth2_scheme)):
     db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/AI_anchor?charset=utf8mb4')
-    
-    user_obj = first(db.query('SELECT * FROM users where token ="'+token_obj.token+'"'))
-    if token_obj.token != user_obj['token']:
+    user_id = get_user_id(token)
+    user_obj = first(db.query('SELECT * FROM users where id ="'+user_id+'"'))
+
+    if user_obj is None:
         raise HTTPException(
             status_code=status.HTTP_401_UNAUTHORIZED,
             detail="Missing token",

api/models.py

@@ -63,5 +63,3 @@ class register_req(BaseModel):
     email: str
     password: str
 
-class jwt_token_only(BaseModel):
-    token: str