Function add_tokens

app/admin/api.py:630–702

添加 Token

(request: Request)

@router.post("/tokens/add")
async def add_tokens(request: Request):
    """添加 Token"""
    from app.services.token_dao import get_token_dao
    from app.utils.token_pool import get_token_pool

    form_data = await request.form()
    single_token = form_data.get("single_token", "").strip()
    bulk_tokens = form_data.get("bulk_tokens", "").strip()

    dao = get_token_dao()
    added_count = 0
    failed_count = 0

    # 添加单个 Token（带验证）
    if single_token:
        token_id = await dao.add_token(
            DEFAULT_TOKEN_NAMESPACE,
            single_token,
            validate=True,
        )
        if token_id:
            added_count += 1
        else:
            failed_count += 1

    # 批量添加 Token（带验证）
    if bulk_tokens:
        # 支持换行和逗号分隔
        tokens = []
        for line in bulk_tokens.split('\n'):
            line = line.strip()
            if ',' in line:
                tokens.extend([t.strip() for t in line.split(',') if t.strip()])
            elif line:
                tokens.append(line)

        success, failed = await dao.bulk_add_tokens(
            DEFAULT_TOKEN_NAMESPACE,
            tokens,
            validate=True,
        )
        added_count += success
        failed_count += failed

    # 同步 Token 池状态（如果有新增成功的 Token）
    if added_count > 0:
        pool = get_token_pool()
        if pool:
            await pool.sync_from_database(DEFAULT_TOKEN_NAMESPACE)
            logger.info(f"✅ Token 池已同步，新增 {added_count} 个 Token")

    # 生成响应
    if added_count > 0 and failed_count == 0:
        return HTMLResponse(f"""
        <div class="bg-green-100 border border-green-400 text-green-700 px-4 py-3 rounded relative" role="alert">
            <strong class="font-bold">成功！</strong>
            <span class="block sm:inline">已添加 {added_count} 个有效 Token</span>
        </div>