58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
"""用户可添加抖音账号数量限制（限制功能已移除，保留接口兼容）。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
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:
|
|
"""账号数量/并发限制已移除，始终不限制。"""
|
|
return None
|
|
|
|
|
|
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:
|
|
"""账号数量限制已移除，任何用户可添加任意数量账号。"""
|
|
return
|