更新

backend/auth/account_quota.py

@@ -0,0 +1,68 @@
+"""账号额度与 quota_disabled 同步。"""
+
+from __future__ import annotations
+
+from collections.abc import Awaitable, Callable
+from typing import Any
+
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from models.models import Account, User
+from .account_limits import account_limit_for_user
+
+QUOTA_DISABLED_MESSAGE = "账号额度不足，该账号已被系统停用"
+
+
+async def sync_user_account_quota(
+    db: AsyncSession,
+    user: User,
+    *,
+    stop_worker: Callable[[int], Awaitable[None]] | None = None,
+) -> dict[str, Any]:
+    """按 max_accounts 同步停用状态：保留最早创建的账号，超额账号禁用并停止托管。"""
+    limit = account_limit_for_user(user)
+
+    result = await db.execute(
+        select(Account)
+        .where(Account.owner_id == user.id)
+        .order_by(Account.created_at.asc(), Account.id.asc())
+    )
+    accounts = list(result.scalars().all())
+
+    disabled_ids: list[int] = []
+    enabled_ids: list[int] = []
+
+    if limit is None:
+        for acc in accounts:
+            if acc.quota_disabled:
+                acc.quota_disabled = False
+                if (acc.error_message or "").strip() == QUOTA_DISABLED_MESSAGE:
+                    acc.error_message = None
+                enabled_ids.append(acc.id)
+        return {"disabled_ids": disabled_ids, "enabled_ids": enabled_ids, "limit": None}
+
+    for idx, acc in enumerate(accounts):
+        if idx < limit:
+            if acc.quota_disabled:
+                acc.quota_disabled = False
+                if (acc.error_message or "").strip() == QUOTA_DISABLED_MESSAGE:
+                    acc.error_message = None
+                enabled_ids.append(acc.id)
+        else:
+            if not acc.quota_disabled:
+                acc.quota_disabled = True
+            acc.error_message = QUOTA_DISABLED_MESSAGE
+            acc.status = "offline"
+            acc.qr_code_base64 = None
+            disabled_ids.append(acc.id)
+            if stop_worker:
+                await stop_worker(acc.id)
+
+    return {"disabled_ids": disabled_ids, "enabled_ids": enabled_ids, "limit": limit}
+
+
+async def default_stop_worker(account_id: int) -> None:
+    from main import manager
+
+    await manager.stop_worker(account_id)