增加了账号管理的分组功能

web/backend/database.py

@@ -29,9 +29,22 @@ def get_db():
 def init_db():
     """建表 + 写入初始数据。"""
     Base.metadata.create_all(bind=engine)
+    _migrate()
     _seed()
 
 
+def _migrate():
+    """数据库迁移：为已有表添加新列。"""
+    from sqlalchemy import text
+    with engine.connect() as conn:
+        # 检查 accounts.tag 列是否存在
+        result = conn.execute(text("PRAGMA table_info(accounts)"))
+        columns = [row[1] for row in result]
+        if 'tag' not in columns:
+            conn.execute(text("ALTER TABLE accounts ADD COLUMN tag VARCHAR(64) DEFAULT ''"))
+            conn.commit()
+
+
 def _seed():
     """写入默认超管账号和角色。"""
     from .models import User

web/backend/models.py

@@ -37,6 +37,7 @@ class Account(Base):
     email_imap_server = Column(String(128), default="")
     email_imap_port = Column(Integer, default=993)
     assigned_to = Column(Integer, ForeignKey("users.id"), nullable=True, index=True)
+    tag = Column(String(64), default="")
     remark = Column(String(256), default="")
     created_at = Column(DateTime, default=datetime.utcnow)
 

web/backend/routers/accounts.py

@@ -7,7 +7,7 @@ from sqlalchemy.orm import Session
 
 from ..database import get_db
 from ..models import User, Account, AuditLog
-from ..schemas import AccountImport, AccountAssign, AccountOut
+from ..schemas import AccountImport, AccountAssign, AccountTag, AccountOut
 from ..deps import get_current_user, require_permission
 from ..permissions import has_permission
 
@@ -29,6 +29,7 @@ def _split_account_line(line: str) -> list[str]:
 @router.get("", response_model=list[AccountOut])
 def list_accounts(
     assigned_only: bool = Query(False),
+    tag: str = Query(None),
     db: Session = Depends(get_db),
     current: User = Depends(get_current_user),
 ):
@@ -45,11 +46,15 @@ def list_accounts(
     if assigned_only and has_permission(current.role, "account:view_all"):
         query = query.filter(Account.assigned_to.isnot(None))
 
+    if tag:
+        query = query.filter(Account.tag == tag)
+
     accounts = query.order_by(Account.id).all()
     result = []
     for acc in accounts:
         item = AccountOut(
             id=acc.id, username=acc.username, remark=acc.remark or "",
+            tag=acc.tag or "",
             assigned_to=acc.assigned_to,
             assigned_username=acc.assigned_user.username if acc.assigned_user else None,
             created_at=acc.created_at,
@@ -69,7 +74,7 @@ def import_accounts(
     db: Session = Depends(get_db),
     current: User = Depends(require_permission("account:import")),
 ):
-    """批量导入账号。格式：用户名|密码|邮箱|邮箱密码"""
+    """批量导入账号。格式：用户名|密码|邮箱|邮箱密码|标签(可选)"""
     from core.douyu.email_verifier import get_email_config_for_account
 
     accounts = []
@@ -79,11 +84,12 @@ def import_accounts(
         if not line or line.startswith('#'):
             continue
         parts = _split_account_line(line)
-        if len(parts) != 4:
+        if len(parts) < 4:
             skipped += 1
             continue
 
-        username, password, email, email_password = [p.strip() for p in parts]
+        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
@@ -99,6 +105,7 @@ def import_accounts(
             email_password=email_password,
             email_imap_server=email_cfg['server'],
             email_imap_port=email_cfg.get('port', 993),
+            tag=tag,
         ))
 
     if accounts:
@@ -135,6 +142,49 @@ def assign_account(
     return {"message": "已分配", "success": True}
 
 
+@router.put("/{account_id}/tag")
+def set_account_tag(
+    account_id: int,
+    req: AccountTag,
+    db: Session = Depends(get_db),
+    current: User = Depends(require_permission("account:import")),
+):
+    """设置单个账号标签。"""
+    acc = db.query(Account).filter(Account.id == account_id).first()
+    if not acc:
+        raise HTTPException(status_code=404, detail="账号不存在")
+    acc.tag = (req.tag or "").strip()
+    db.commit()
+    return {"message": "标签已更新", "success": True}
+
+
+@router.put("/batch-tag")
+def batch_tag(
+    req: AccountTag,
+    db: Session = Depends(get_db),
+    current: User = Depends(require_permission("account:import")),
+):
+    """批量设置账号标签。"""
+    if not req.account_ids:
+        raise HTTPException(status_code=400, detail="请选择账号")
+    tag = (req.tag or "").strip()
+    count = db.query(Account).filter(Account.id.in_(req.account_ids)).update(
+        {Account.tag: tag}, synchronize_session=False
+    )
+    db.commit()
+    return {"message": f"已为 {count} 个账号设置标签", "success": True}
+
+
+@router.get("/tags/list")
+def list_tags(
+    db: Session = Depends(get_db),
+    current: User = Depends(get_current_user),
+):
+    """获取所有标签列表。"""
+    tags = db.query(Account.tag).filter(Account.tag != "", Account.tag.isnot(None)).distinct().all()
+    return [t[0] for t in tags if t[0]]
+
+
 @router.delete("/{account_id}")
 def delete_account(
     account_id: int,

web/backend/schemas.py

@@ -57,6 +57,11 @@ class AccountAssign(BaseModel):
     assigned_to: Optional[int] = None
 
 
+class AccountTag(BaseModel):
+    tag: Optional[str] = None
+    account_ids: Optional[list[int]] = None
+
+
 class AccountOut(BaseModel):
     id: int
     username: str
@@ -64,6 +69,7 @@ class AccountOut(BaseModel):
     password: Optional[str] = None
     email: Optional[str] = None
     email_password: Optional[str] = None
+    tag: str = ""
     assigned_to: Optional[int] = None
     assigned_username: Optional[str] = None
     remark: str = ""