This commit is contained in:
jxxghp
2024-06-19 18:04:00 +08:00
parent 631df4c9f8
commit 0fb12c77eb
3 changed files with 10 additions and 4 deletions

View File

@ -30,7 +30,7 @@ reusable_oauth2 = OAuth2PasswordBearer(
def create_access_token(
userid: Union[str, Any], username: str, super_user: bool = False,
expires_delta: timedelta = None
expires_delta: timedelta = None, level: int = 1
) -> str:
if expires_delta:
expire = datetime.utcnow() + expires_delta
@ -42,7 +42,8 @@ def create_access_token(
"exp": expire,
"sub": str(userid),
"username": username,
"super_user": super_user
"super_user": super_user,
"level": level
}
encoded_jwt = jwt.encode(to_encode, settings.SECRET_KEY, algorithm=ALGORITHM)
return encoded_jwt