实现虎牙密码登录

web/backend/services/huya_service.py

@@ -97,6 +97,50 @@ def parse_huya_cookie_line(line: str) -> dict | None:
     }
 
 
+def _upsert_huya_account(db: Session, parsed: dict, tag: str = "", status: str | None = None) -> HuyaAccount:
+    """按 uid/yyuid 新增或更新虎牙账号。"""
+    account = None
+    if parsed["uid"]:
+        account = db.query(HuyaAccount).filter(HuyaAccount.uid == parsed["uid"]).first()
+    if account is None and parsed["yyuid"]:
+        account = db.query(HuyaAccount).filter(HuyaAccount.yyuid == parsed["yyuid"]).first()
+
+    if account is None:
+        account = HuyaAccount(
+            uid=parsed["uid"],
+            yyuid=parsed["yyuid"],
+            username=parsed["username"],
+            cookie=parsed["cookie"],
+            game_phone=parsed["game_phone"],
+            tag=tag,
+            status=status or "imported",
+        )
+        db.add(account)
+    else:
+        account.uid = parsed["uid"] or account.uid
+        account.yyuid = parsed["yyuid"] or account.yyuid
+        account.username = parsed["username"] or account.username
+        account.cookie = parsed["cookie"]
+        account.game_phone = parsed["game_phone"] or account.game_phone
+        if tag:
+            account.tag = tag
+        account.status = status or "updated"
+        account.updated_at = datetime.now(timezone.utc)
+    return account
+
+
+def upsert_huya_cookie(db: Session, cookie: str, tag: str = "", username_hint: str = "") -> HuyaAccount:
+    """保存单条登录得到的虎牙 Cookie。"""
+    line = f"{username_hint}----{cookie}" if username_hint else cookie
+    parsed = parse_huya_cookie_line(line)
+    if not parsed:
+        raise ValueError("登录成功但 Cookie 中没有识别到虎牙 uid")
+    account = _upsert_huya_account(db, parsed, tag=(tag or "").strip(), status="login_success")
+    db.commit()
+    db.refresh(account)
+    return account
+
+
 def import_huya_cookies(db: Session, text: str, tag: str = "") -> tuple[int, int]:
     """导入虎牙 Cookie，返回 (成功数, 跳过数)。"""
     created_or_updated = 0
@@ -109,34 +153,7 @@ def import_huya_cookies(db: Session, text: str, tag: str = "") -> tuple[int, int
             skipped += 1
             continue
 
-        account = None
-        if parsed["uid"]:
-            account = db.query(HuyaAccount).filter(HuyaAccount.uid == parsed["uid"]).first()
-        if account is None and parsed["yyuid"]:
-            account = db.query(HuyaAccount).filter(HuyaAccount.yyuid == parsed["yyuid"]).first()
-
-        if account is None:
-            account = HuyaAccount(
-                uid=parsed["uid"],
-                yyuid=parsed["yyuid"],
-                username=parsed["username"],
-                cookie=parsed["cookie"],
-                game_phone=parsed["game_phone"],
-                tag=tag,
-                status="imported",
-            )
-            db.add(account)
-        else:
-            account.uid = parsed["uid"] or account.uid
-            account.yyuid = parsed["yyuid"] or account.yyuid
-            account.username = parsed["username"] or account.username
-            account.cookie = parsed["cookie"]
-            account.game_phone = parsed["game_phone"] or account.game_phone
-            if tag:
-                account.tag = tag
-            account.status = "updated"
-            account.updated_at = datetime.now(timezone.utc)
-
+        _upsert_huya_account(db, parsed, tag=tag)
         created_or_updated += 1
 
     if created_or_updated: