"""账号管理服务层:封装账号导入、查询等业务逻辑,隔离 core/ 依赖。""" import csv import re from sqlalchemy.orm import Session from ..models import Account, LoginTask EMAIL_PATTERN = re.compile(r'^[^\s@|]+@[^\s@|]+\.[^\s@|]+$') def split_account_line(line: str) -> list[str]: """拆分一行账号文本,支持 |、tab、逗号、空格分隔。""" if '|' in line: return line.split('|') if '\t' in line: return line.split('\t') if ',' in line: return next(csv.reader([line])) return line.split() def cookie_account_ids_query(db: Session): """返回有成功登录记录(cookie非空)的账号ID子查询。""" return db.query(LoginTask.account_id).filter( LoginTask.status == 'success', LoginTask.cookie != '', LoginTask.cookie.isnot(None), ).distinct() def parse_and_build_accounts(text: str, default_tag: str = "") -> tuple[list[Account], int]: """ 解析批量导入文本,构建 Account ORM 对象列表。 Args: text: 批量导入文本,每行格式:用户名|密码|邮箱|邮箱密码|标签(可选) default_tag: 统一标签兜底。某行未单独写标签时使用该值, 行内第5列标签优先。 Returns: (accounts, skipped_count) """ from core.douyu.email_verifier import get_email_config_for_account default_tag = (default_tag or "").strip() accounts = [] skipped = 0 for line in text.strip().split('\n'): line = line.strip() if not line or line.startswith('#'): continue parts = split_account_line(line) if len(parts) < 4: skipped += 1 continue username, password, email, email_password = [p.strip() for p in parts[:4]] line_tag = parts[4].strip() if len(parts) > 4 else "" # 行内标签优先,未写则用统一兜底标签 tag = line_tag or default_tag if not all([username, password, email, email_password]): skipped += 1 continue if not EMAIL_PATTERN.match(email): skipped += 1 continue email_cfg = get_email_config_for_account(email) accounts.append(Account( username=username, password=password, email=email, email_password=email_password, email_imap_server=email_cfg['server'], email_imap_port=email_cfg.get('port', 993), email_imap_ssl=email_cfg.get('ssl', True), tag=tag, )) return accounts, skipped