nas/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-04-04 19:28:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(auth): handle scenario where the user is null

Browse Source
This commit is contained in:
InfinityPacer
2024-09-16 15:41:13 +08:00
parent 4efc80e35a
commit 36142b97bf
1 changed files with 17 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -38,21 +38,25 @@ async def login_access_token(
otp_password=otp_password
)
if not success:
# 认证不成功且开启了辅助认证
if not user and settings.AUXILIARY_AUTH_ENABLE:
# 未找到用户，请求协助认证
logger.warn(f"登录用户 {form_data.username} 本地不存在，尝试辅助认证 ...")
token = UserChain().user_authenticate(form_data.username, form_data.password)
if not token:
# 认证不成功
if not user:
if not settings.AUXILIARY_AUTH_ENABLE:
logger.warn(f"用户 {form_data.username} 登录失败！")
raise HTTPException(status_code=401, detail="用户名、密码、二次校验码不正确")
raise HTTPException(status_code=401, detail="用户名、密码或二次校验码不正确")
else:
logger.info(f"用户 {form_data.username} 辅助认证成功，用户信息: {token}，以普通用户登录...")
# 加入用户信息表
logger.info(f"创建用户: {form_data.username}")
user = User(name=form_data.username, is_active=True,
is_superuser=False, hashed_password=get_password_hash(token))
user.create(db)
# 如果找不到用户并开启了辅助认证
logger.warn(f"登录用户 {form_data.username} 本地不存在，尝试辅助认证 ...")
token = UserChain().user_authenticate(form_data.username, form_data.password)
if not token:
logger.warn(f"用户 {form_data.username} 登录失败！")
raise HTTPException(status_code=401, detail="用户名、密码、二次校验码不正确")
else:
logger.info(f"用户 {form_data.username} 辅助认证成功，用户信息: {token}，以普通用户登录...")
# 加入用户信息表
logger.info(f"创建用户: {form_data.username}")
user = User(name=form_data.username, is_active=True,
is_superuser=False, hashed_password=get_password_hash(token))
user.create(db)
else:
# 用户存在，但认证失败
logger.warn(f"用户 {user.name} 登录失败！")