codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bug

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app/api/endpoints/login.py
View File

@ -13,6 +13,7 @@ from app.core.config import settings
from app.core.security import get_password_hash from app.core.security import get_password_hash
from app.db import get_db from app.db import get_db
from app.db.models.user import User from app.db.models.user import User
from app.helper.sites import SitesHelper
from app.log import logger from app.log import logger
from app.utils.web import WebUtils from app.utils.web import WebUtils
@ -58,17 +59,20 @@ async def login_access_token(
elif user and not user.is_active: elif user and not user.is_active:
raise HTTPException(status_code=403, detail="用户未启用") raise HTTPException(status_code=403, detail="用户未启用")
logger.info(f"用户 {user.name} 登录成功！") logger.info(f"用户 {user.name} 登录成功！")
level = SitesHelper().auth_level
return schemas.Token( return schemas.Token(
access_token=security.create_access_token( access_token=security.create_access_token(
userid=user.id, userid=user.id,
username=user.name, username=user.name,
super_user=user.is_superuser, super_user=user.is_superuser,
expires_delta=timedelta(minutes=settings.ACCESS_TOKEN_EXPIRE_MINUTES) expires_delta=timedelta(minutes=settings.ACCESS_TOKEN_EXPIRE_MINUTES),
level=level
), ),
token_type="bearer", token_type="bearer",
super_user=user.is_superuser, super_user=user.is_superuser,
user_name=user.name, user_name=user.name,
avatar=user.avatar avatar=user.avatar,
level=level
) )

5
app/core/security.py
View File

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

1
app/schemas/token.py
View File

@ -9,6 +9,7 @@ class Token(BaseModel):
super_user: bool super_user: bool
user_name: str user_name: str
avatar: Optional[str] = None avatar: Optional[str] = None
level: int = 1
class TokenPayload(BaseModel): class TokenPayload(BaseModel):