Explorar o código

调整Authorization指向管理员账号

baoyubo hai 11 meses
pai
achega
7919d59a4b
Modificáronse 1 ficheiros con 6 adicións e 5 borrados
  1. 6 5
      common/security.py

+ 6 - 5
common/security.py

@@ -14,17 +14,18 @@ from extensions import logger
 
 pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
 
-def valid_access_token(Authorization: str = Header(..., alias="Authorization")) -> int:
-    access_token = Authorization.removeprefix("Bearer ") 
-    
-    token_exception = TokenException()
+def valid_access_token(Authorization: str ="Bearer eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJzdWIiOiIxIiwiZXhwIjoyMDM5Njk2ODMzfQ.Rhd38oo_S1odjg0xnT4n31cCWCAAPXGb8y_V2XcgqzQ")->int:#= Header(..., alias="Authorization")) -> int:
 
     try:
+        access_token = Authorization.removeprefix("Bearer ")
+
+        token_exception = TokenException()
         payload = jwt.decode(access_token, settings.SECRET_KEY, algorithms=[settings.ALGORITHM])
         user_id: str = payload.get("sub")
         logger.info('sub user_id: {}', user_id)
     except Exception:
-        raise token_exception
+        return -1
+        #raise token_exception
     
     return int(user_id)