From e22436bfd6cac7266c5043cc650cd36bf0951fd2 Mon Sep 17 00:00:00 2001 From: yml2213 Date: Sat, 1 Aug 2026 13:27:06 +0800 Subject: [PATCH] =?UTF-8?q?fix(=E8=B4=A6=E5=8F=B7):=20=E6=89=B9=E9=87=8F?= =?UTF-8?q?=E5=AF=BC=E5=85=A5=E8=B7=B3=E8=BF=87=E5=B7=B2=E5=AD=98=E5=9C=A8?= =?UTF-8?q?=E7=9A=84=E9=87=8D=E5=A4=8D=E8=B4=A6=E5=8F=B7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 导入前按用户名(大小写不敏感)查库去重,同批文本内重复也跳过 - 返回消息区分格式跳过与重复跳过数量 --- web/backend/routers/accounts.py | 10 +++++++-- web/backend/services/account_service.py | 27 +++++++++++++++++++++---- 2 files changed, 31 insertions(+), 6 deletions(-) diff --git a/web/backend/routers/accounts.py b/web/backend/routers/accounts.py index be91e59..9141c94 100644 --- a/web/backend/routers/accounts.py +++ b/web/backend/routers/accounts.py @@ -189,8 +189,9 @@ def import_accounts( """批量导入账号。格式:用户名|密码|邮箱|邮箱密码|标签(可选) req.tag 为统一标签兜底:某行未单独写标签时使用该值,行内标签优先。 + 库中已存在的用户名(大小写不敏感)与同批重复行会自动跳过。 """ - accounts, skipped = parse_and_build_accounts(req.text, req.tag) + accounts, skipped, duplicated = parse_and_build_accounts(db, req.text, req.tag) if accounts: db.add_all(accounts) @@ -198,7 +199,12 @@ def import_accounts( action="account:import", target=f"导入{len(accounts)}个")) db.commit() - return {"message": f"导入成功 {len(accounts)} 个,跳过 {skipped} 个", "success": True, "count": len(accounts)} + duplicated_note = f",重复跳过 {duplicated} 个" if duplicated else "" + return { + "message": f"导入成功 {len(accounts)} 个,跳过 {skipped} 个{duplicated_note}", + "success": True, + "count": len(accounts), + } @router.put("/{account_id}/assign") diff --git a/web/backend/services/account_service.py b/web/backend/services/account_service.py index 107c454..3079b0b 100644 --- a/web/backend/services/account_service.py +++ b/web/backend/services/account_service.py @@ -31,23 +31,36 @@ def cookie_account_ids_query(db: Session): ).distinct() -def parse_and_build_accounts(text: str, default_tag: str = "") -> tuple[list[Account], int]: +def parse_and_build_accounts( + db: Session, + text: str, + default_tag: str = "", +) -> tuple[list[Account], int, int]: """ - 解析批量导入文本,构建 Account ORM 对象列表。 + 解析批量导入文本,构建 Account ORM 对象列表,并过滤已存在的重复账号。 Args: + db: 数据库会话,用于查询已有账号 text: 批量导入文本,每行格式:用户名|密码|邮箱|邮箱密码|标签(可选) default_tag: 统一标签兜底。某行未单独写标签时使用该值, 行内第5列标签优先。 Returns: - (accounts, skipped_count) + (accounts, skipped_count, duplicated_count) + skipped_count 为格式非法的行数;duplicated_count 为库中已存在或同批重复被跳过的行数。 """ from core.douyu.email_verifier import get_email_config_for_account default_tag = (default_tag or "").strip() + # 库中已有用户名(大小写不敏感),用于导入去重 + existing_usernames = { + row[0].lower() + for row in db.query(Account.username).filter(Account.username != "").all() + } accounts = [] skipped = 0 + duplicated = 0 + seen_in_batch: set[str] = set() for line in text.strip().split('\n'): line = line.strip() if not line or line.startswith('#'): @@ -68,6 +81,12 @@ def parse_and_build_accounts(text: str, default_tag: str = "") -> tuple[list[Acc skipped += 1 continue + username_key = username.lower() + if username_key in existing_usernames or username_key in seen_in_batch: + duplicated += 1 + continue + seen_in_batch.add(username_key) + email_cfg = get_email_config_for_account(email) accounts.append(Account( username=username, @@ -79,4 +98,4 @@ def parse_and_build_accounts(text: str, default_tag: str = "") -> tuple[list[Acc email_imap_ssl=email_cfg.get('ssl', True), tag=tag, )) - return accounts, skipped + return accounts, skipped, duplicated