yml2213
dd56de9bd4
- proxy_service.py: 将requests和WhitelistManager改为函数内延迟import， 避免启动时加载不需要的依赖；移除未使用的get_exit_ip_via_proxy导入 - account_service.py: 移除未使用的func、joinedload、AuditLog、 user_has_permission顶层导入
75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
"""账号管理服务层：封装账号导入、查询等业务逻辑，隔离 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) -> tuple[list[Account], int]:
|
|
"""
|
|
解析批量导入文本，构建 Account ORM 对象列表。
|
|
|
|
Returns:
|
|
(accounts, skipped_count)
|
|
"""
|
|
from core.douyu.email_verifier import get_email_config_for_account
|
|
|
|
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]]
|
|
tag = parts[4].strip() if len(parts) > 4 else ""
|
|
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
|