This commit is contained in:
王思川 2022-11-10 15:41:15 +08:00
parent b1a5fbe63b
commit 1bb1b326df
2 changed files with 4 additions and 4 deletions

View File

@ -13,7 +13,7 @@ def get_user_info(token: str):
""" """
user_info = decode_token(token).get("user_info") user_info = decode_token(token).get("user_info")
if not user_info: if not user_info:
raise HTTPException(status_code=400, detail="Invalid Token") raise HTTPException(status_code=401, detail="Invalid Token")
return user_info return user_info
@ -33,7 +33,7 @@ def ac(token: Optional[str] = Header(...)):
return True return True
def rbac(request: Request, token: Optional[str] = Header(...)): def rbac(request: Request, token: Optional[str] = Header(None)):
# 获取用户角色 # 获取用户角色
user_info = get_user_info(token) user_info = get_user_info(token)
@ -55,4 +55,4 @@ def rbac(request: Request, token: Optional[str] = Header(...)):
return True return True
# 权限匹配失败 # 权限匹配失败
raise HTTPException(status_code=400, detail="No Access") raise HTTPException(status_code=202, detail="No Access")

View File

@ -11,7 +11,7 @@ def decode_token(token: str):
try: try:
payload = jwt.decode(token, Config.SECRET_KEY, algorithms=[Config.ALGORITHM]) payload = jwt.decode(token, Config.SECRET_KEY, algorithms=[Config.ALGORITHM])
except jwt.exceptions.ExpiredSignatureError: except jwt.exceptions.ExpiredSignatureError:
raise HTTPException(status_code=201, detail="Token Has Expired") raise HTTPException(status_code=401, detail="Token Has Expired")
except PyJWTError: except PyJWTError:
raise HTTPException(status_code=401, detail="Invalid Token") raise HTTPException(status_code=401, detail="Invalid Token")
return payload return payload