changes
This commit is contained in:
parent
b1a5fbe63b
commit
1bb1b326df
|
@ -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")
|
||||||
|
|
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue