更新

backend/auth/account_limits.py

@@ -0,0 +1,70 @@
+"""用户可添加抖音账号数量限制。"""
+
+from __future__ import annotations
+
+from fastapi import HTTPException
+from sqlalchemy import func, select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from models.models import Account, User
+from .roles import is_admin
+
+UNLIMITED_ACCOUNTS = -1
+
+
+def normalize_max_accounts(value: int | None, role: str = "operator") -> int:
+    if is_admin(role):
+        return UNLIMITED_ACCOUNTS
+    if value is None:
+        return 3
+    try:
+        parsed = int(value)
+    except (TypeError, ValueError):
+        return 3
+    if parsed < 0:
+        return UNLIMITED_ACCOUNTS
+    return parsed
+
+
+def account_limit_for_user(user: User) -> int | None:
+    """返回 None 表示不限制。"""
+    if is_admin(user.role):
+        return None
+    limit = user.max_accounts if user.max_accounts is not None else 3
+    if limit < 0:
+        return None
+    return limit
+
+
+async def count_user_accounts(db: AsyncSession, user_id: int) -> int:
+    result = await db.execute(
+        select(func.count()).select_from(Account).where(Account.owner_id == user_id)
+    )
+    return int(result.scalar() or 0)
+
+
+async def count_user_account_breakdown(db: AsyncSession, user_id: int) -> dict[str, int]:
+    """统计用户名下托管账号：总数 / 可用 / 额度停用。"""
+    total = await count_user_accounts(db, user_id)
+    if total <= 0:
+        return {"total": 0, "active": 0, "disabled": 0}
+    disabled_result = await db.execute(
+        select(func.count())
+        .select_from(Account)
+        .where(Account.owner_id == user_id, Account.quota_disabled.is_(True))
+    )
+    disabled = int(disabled_result.scalar() or 0)
+    active = max(0, total - disabled)
+    return {"total": total, "active": active, "disabled": disabled}
+
+
+async def ensure_can_add_account(db: AsyncSession, user: User) -> None:
+    limit = account_limit_for_user(user)
+    if limit is None:
+        return
+    count = await count_user_accounts(db, user.id)
+    if count >= limit:
+        raise HTTPException(
+            status_code=400,
+            detail=f"已达到可添加抖音账号上限（{limit} 个），请购买额度或联系管理员",
+        )