test: 全面迁移 pytest 并加入格式门禁

web/backend/models.py

@@ -2,7 +2,15 @@
 
 from datetime import datetime, timezone
 from sqlalchemy import (
-    Column, Integer, String, Boolean, Text, DateTime, ForeignKey, JSON, UniqueConstraint,
+    Column,
+    Integer,
+    String,
+    Boolean,
+    Text,
+    DateTime,
+    ForeignKey,
+    JSON,
+    UniqueConstraint,
 )
 from sqlalchemy.orm import relationship
 from .database import Base
@@ -23,27 +31,39 @@ def _utcnow():
 
 class User(Base):
     """系统用户"""
+
     __tablename__ = "users"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
     username = Column(String(64), unique=True, nullable=False, index=True)
     password_hash = Column(String(256), nullable=False)
-    role = Column(String(32), nullable=False, default="support")  # super_admin / operation / support
+    role = Column(
+        String(32), nullable=False, default="support"
+    )  # super_admin / operation / support
     is_active = Column(Boolean, default=True)
     remark = Column(String(256), default="")
-    custom_permissions = Column(JSON, nullable=True, comment="自定义权限列表，null表示使用角色默认权限")
+    custom_permissions = Column(
+        JSON, nullable=True, comment="自定义权限列表，null表示使用角色默认权限"
+    )
     created_at = Column(DateTime, default=_utcnow)
     updated_at = Column(DateTime, default=_utcnow, onupdate=_utcnow)
     deleted_at = Column(DateTime, nullable=True, index=True)
     deleted_username = Column(String(64), default="")
 
     # 客服被分配的账号
-    assigned_accounts = relationship("Account", back_populates="assigned_user", foreign_keys="Account.assigned_to")
-    huya_accounts = relationship("HuyaAccount", back_populates="assigned_user", foreign_keys="HuyaAccount.assigned_to")
+    assigned_accounts = relationship(
+        "Account", back_populates="assigned_user", foreign_keys="Account.assigned_to"
+    )
+    huya_accounts = relationship(
+        "HuyaAccount",
+        back_populates="assigned_user",
+        foreign_keys="HuyaAccount.assigned_to",
+    )
 
 
 class Account(Base):
     """斗鱼账号"""
+
     __tablename__ = "accounts"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -83,19 +103,24 @@ class Account(Base):
     created_at = Column(DateTime, default=_utcnow)
     updated_at = Column(DateTime, default=_utcnow, onupdate=_utcnow)
 
-    assigned_user = relationship("User", back_populates="assigned_accounts", foreign_keys=[assigned_to])
+    assigned_user = relationship(
+        "User", back_populates="assigned_accounts", foreign_keys=[assigned_to]
+    )
     login_tasks = relationship("LoginTask", back_populates="account")
     douyu_tasks = relationship("DouyuTask", back_populates="account")
 
 
 class LoginTask(Base):
     """单次登录任务"""
+
     __tablename__ = "login_tasks"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
     batch_id = Column(String(64), nullable=False, index=True)  # 批次ID
     account_id = Column(Integer, ForeignKey("accounts.id"), nullable=False)
-    status = Column(String(32), default="pending")  # pending / running / success / failed / error
+    status = Column(
+        String(32), default="pending"
+    )  # pending / running / success / failed / error
     cookie = Column(EncryptedText(), default="")
     message = Column(String(512), default="")
     created_by = Column(Integer, ForeignKey("users.id"), nullable=False)
@@ -103,7 +128,9 @@ class LoginTask(Base):
     finished_at = Column(DateTime, nullable=True)
     # Cookie 有效性检测结果（持久化，刷新不丢失）
     ck_check_status = Column(String(16), default="")  # valid / invalid / 空=未检测
-    ck_check_result = Column(JSON, nullable=True)  # {fish_ball, nickname, level, message}
+    ck_check_result = Column(
+        JSON, nullable=True
+    )  # {fish_ball, nickname, level, message}
     ck_checked_at = Column(DateTime, nullable=True)
 
     account = relationship("Account", back_populates="login_tasks")
@@ -111,6 +138,7 @@ class LoginTask(Base):
 
 class DouyuTask(Base):
     """斗鱼业务任务"""
+
     __tablename__ = "douyu_tasks"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -133,6 +161,7 @@ class DouyuTask(Base):
 
 class DouyuWorkbenchAccount(Base):
     """用户在指定斗鱼工作台中启用的账号，跨浏览器同步。"""
+
     __tablename__ = "douyu_workbench_accounts"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -142,12 +171,15 @@ class DouyuWorkbenchAccount(Base):
     created_at = Column(DateTime, default=_utcnow)
 
     __table_args__ = (
-        UniqueConstraint("user_id", "handbook_scope", "account_id", name="uq_douyu_workbench_account"),
+        UniqueConstraint(
+            "user_id", "handbook_scope", "account_id", name="uq_douyu_workbench_account"
+        ),
     )
 
 
 class DouyuWorkbench(Base):
     """工作台配置哨兵，令空账号集合也能跨浏览器同步。"""
+
     __tablename__ = "douyu_workbenches"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -165,6 +197,7 @@ class YybRechargeTask(Base):
 
     YYB 登录身份与斗鱼账号无关，因此任务只关联创建者，不复用 Account。
     """
+
     __tablename__ = "yyb_recharge_tasks"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -197,6 +230,7 @@ class YybRechargeTask(Base):
 
 class DouyuConfig(Base):
     """斗鱼业务配置"""
+
     __tablename__ = "douyu_config"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -228,6 +262,7 @@ class DouyuConfig(Base):
 
 class DouyuGoodsSnapshot(Base):
     """斗鱼兑换商品快照"""
+
     __tablename__ = "douyu_goods_snapshot"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -241,6 +276,7 @@ class DouyuGoodsSnapshot(Base):
 
 class DouyuXpdGoodsSnapshot(Base):
     """斗鱼和平小店商品快照"""
+
     __tablename__ = "douyu_xpd_goods_snapshot"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -256,6 +292,7 @@ class DouyuXpdGoodsSnapshot(Base):
 
 class DouyuEsportsGoodsSnapshot(Base):
     """斗鱼电竞手册皮肤商城商品快照"""
+
     __tablename__ = "douyu_esports_goods_snapshot"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -269,6 +306,7 @@ class DouyuEsportsGoodsSnapshot(Base):
 
 class HuyaAccount(Base):
     """虎牙账号"""
+
     __tablename__ = "huya_accounts"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -292,12 +330,15 @@ class HuyaAccount(Base):
     created_at = Column(DateTime, default=_utcnow)
     updated_at = Column(DateTime, default=_utcnow, onupdate=_utcnow)
 
-    assigned_user = relationship("User", back_populates="huya_accounts", foreign_keys=[assigned_to])
+    assigned_user = relationship(
+        "User", back_populates="huya_accounts", foreign_keys=[assigned_to]
+    )
     tasks = relationship("HuyaTask", back_populates="account")
 
 
 class HuyaRegisterBatch(Base):
     """虎牙自动注册批次（持久化元数据）"""
+
     __tablename__ = "huya_register_batches"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -320,15 +361,20 @@ class HuyaRegisterBatch(Base):
     started_at = Column(DateTime, nullable=True)
     finished_at = Column(DateTime, nullable=True)
 
-    items = relationship("HuyaRegisterItem", back_populates="batch", cascade="all, delete-orphan")
+    items = relationship(
+        "HuyaRegisterItem", back_populates="batch", cascade="all, delete-orphan"
+    )
 
 
 class HuyaRegisterItem(Base):
     """虎牙自动注册批次内单条手机号状态"""
+
     __tablename__ = "huya_register_items"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
-    batch_db_id = Column(Integer, ForeignKey("huya_register_batches.id"), nullable=False, index=True)
+    batch_db_id = Column(
+        Integer, ForeignKey("huya_register_batches.id"), nullable=False, index=True
+    )
     batch_id = Column(String(64), nullable=False, index=True)
     line = Column(Integer, nullable=False)
     phone = Column(String(64), default="", index=True)
@@ -354,6 +400,7 @@ class HuyaRegisterItem(Base):
 
 class HuyaRegisterSuccessLog(Base):
     """虎牙自动注册成功流水（只追加，供随时导出）"""
+
     __tablename__ = "huya_register_success_logs"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -373,6 +420,7 @@ class HuyaRegisterSuccessLog(Base):
 
 class HuyaTask(Base):
     """虎牙业务任务"""
+
     __tablename__ = "huya_tasks"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -391,6 +439,7 @@ class HuyaTask(Base):
 
 class HuyaConfig(Base):
     """虎牙业务配置"""
+
     __tablename__ = "huya_config"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -404,6 +453,7 @@ class HuyaConfig(Base):
 
 class HuyaGoodsSnapshot(Base):
     """虎牙兑换商品快照"""
+
     __tablename__ = "huya_goods_snapshot"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -417,6 +467,7 @@ class HuyaGoodsSnapshot(Base):
 
 class HuyaRechargeGoodsSnapshot(Base):
     """虎牙充值商品快照"""
+
     __tablename__ = "huya_recharge_goods_snapshot"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -436,6 +487,7 @@ class HuyaRechargeGoodsSnapshot(Base):
 
 class ProxyConfig(Base):
     """代理配置（全局单条记录）"""
+
     __tablename__ = "proxy_config"
 
     id = Column(Integer, primary_key=True, autoincrement=True)
@@ -454,6 +506,7 @@ class ProxyConfig(Base):
 
 class AuditLog(Base):
     """操作审计日志"""
+
     __tablename__ = "audit_logs"
 
     id = Column(Integer, primary_key=True, autoincrement=True)