DigitalMgnt
/
user-wsc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

changes

This commit is contained in:
王思川 2022-11-01 14:18:09 +08:00
parent 6cb0ea93e5
commit f5a8c18261
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Utils/AccessControl/AccessUtil.py
View File

@ -7,7 +7,12 @@ from Utils.Authentication.TokenUtil import decode_token
def ac_admin(token: Optional[str] = Header(...)):
role = decode_token(token).get("user_info").get("role")
user_info = decode_token(token).get("user_info")
if not user_info:
raise HTTPException(status_code=400, detail="Invalid Token")
role = user_info.get("role")
if role != "admin":
raise HTTPException(status_code=400, detail="No Access")