Merge pull request #2725 from InfinityPacer/feature/auth

app/api/endpoints/login.py

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

database/versions/294b007932ef_2_0_0.py

@@ -40,7 +40,7 @@ def upgrade() -> None:
                 hashed_password=get_password_hash(random_password),
                 email="admin@movie-pilot.org",
                 is_superuser=True,
-                avatar="src/assets/images/avatars/avatar-1.png"
+                avatar=""
             )
             _user.create(db)
         # 初始化本地存储