From 11c0b622aac04c65495f12544d509e19f1e3c656 Mon Sep 17 00:00:00 2001 From: cnlimiter Date: Thu, 19 Mar 2026 19:01:13 +0800 Subject: [PATCH] =?UTF-8?q?fix(team=20manager):=20=E4=BF=AE=E5=A4=8D?= =?UTF-8?q?=E4=B8=8A=E4=BC=A0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 98 ++++++++++++++------------ src/core/upload/team_manager_upload.py | 75 ++++++++++++++++---- 2 files changed, 115 insertions(+), 58 deletions(-) diff --git a/README.md b/README.md index 2b70647..1b9bcf3 100644 --- a/README.md +++ b/README.md @@ -31,19 +31,21 @@ - 降级轮询备用方案 - **代理管理** - - 静态代理配置 - 动态代理（通过 API 每次获取新 IP） - - 代理列表（随机选取，记录使用时间） + - 代理列表（随机选取，支持设置默认代理，记录使用时间） - **账号管理** - 查看、删除、批量操作 - Token 刷新与验证 - - 导出（JSON / CSV / CPA 格式） + - 订阅状态管理（手动标记 / 自动检测 plus/team/free） + - 导出格式：JSON / CSV / CPA 格式 / Sub2API 格式 - 单个账号导出为独立 `.json` 文件 - - 多个账号打包为 `.zip`，每个账号一个独立文件 - - CPA 上传（Codex Protocol API，直连不走代理） - - 订阅状态管理（手动标记 / 自动检测 plus/team） - - Team Manager 上传（直连不走代理） + - 多个 CPA 账号打包为 `.zip`，每个账号一个独立文件 + - Sub2API 格式所有账号合并为单个 JSON + - 上传目标（直连不走代理）： + - **CPA**：支持多服务配置，上传时选择目标服务 + - **Sub2API**：支持多服务配置，标准 sub2api-data 格式 + - **Team Manager**：支持多服务配置 - **支付升级** - 为账号生成 ChatGPT Plus 或 Team 订阅支付链接 @@ -51,12 +53,13 @@ - Team 套餐支持自定义工作区名称、座位数、计费周期 - **系统设置** - - 代理配置（静态 + 动态） + - 代理配置（动态代理 + 代理列表，支持设默认） + - CPA 服务列表管理（多服务，连接测试） + - Sub2API 服务列表管理（多服务，连接测试） + - Team Manager 服务列表管理（多服务，连接测试） - Outlook OAuth 参数 - 注册参数（超时、重试、密码长度等） - 验证码等待配置 - - CPA 上传配置 - - Team Manager 配置（API URL + API Key） - 数据库管理（备份、清理） - 支持远程 PostgreSQL @@ -152,14 +155,16 @@ codex-register-v2/ ├── build.sh # Linux/macOS 打包脚本 ├── src/ │ ├── config/ # 配置管理（Pydantic Settings） -│ ├── core/ # 核心功能（注册引擎、HTTP 客户端、CPA 上传、支付、TM 上传） -│ ├── database/ # 数据库（SQLAlchemy + SQLite） +│ ├── core/ +│ │ ├── openai/ # OAuth、Token 刷新、支付核心 +│ │ └── upload/ # CPA / Sub2API / Team Manager 上传模块 +│ ├── database/ # 数据库（SQLAlchemy + SQLite/PostgreSQL） │ ├── services/ # 邮箱服务实现 -│ └── web/ # FastAPI Web 应用 +│ └── web/ │ ├── app.py # 应用入口、路由挂载 -│ ├── routes/ # API 路由 │ ├── task_manager.py # 任务/日志/WebSocket 管理 -│ └── routes/websocket.py # WebSocket 处理 +│ └── routes/ # API 路由 +│ └── upload/ # CPA / Sub2API / TM 服务管理路由 ├── templates/ # Jinja2 HTML 模板 ├── static/ # 静态资源（CSS / JS） └── data/ # 运行时数据目录（数据库、日志） @@ -184,44 +189,40 @@ codex-register-v2/ | 方法 | 路径 | 说明 | |------|------|------| -| POST | `/api/registration/start` | 启动单次注册 | -| POST | `/api/registration/batch` | 启动批量注册（支持 `concurrency`、`mode` 参数） | -| GET | `/api/registration/batch/{id}` | 批量任务状态 | -| POST | `/api/registration/batch/{id}/cancel` | 取消批量任务 | -| POST | `/api/registration/outlook-batch` | 启动 Outlook 批量注册 | -| GET | `/api/registration/outlook-batch/{id}` | Outlook 批量状态 | +| POST | `/api/registration/start` | 启动注册任务 | | GET | `/api/registration/tasks` | 任务列表 | -| GET | `/api/registration/tasks/{uuid}` | 任务详情 | | GET | `/api/registration/tasks/{uuid}/logs` | 任务日志 | | POST | `/api/registration/tasks/{uuid}/cancel` | 取消任务 | -| DELETE | `/api/registration/tasks/{uuid}` | 删除任务 | | GET | `/api/registration/available-services` | 可用邮箱服务 | -| GET | `/api/registration/outlook-accounts` | 可用 Outlook 账户 | ### 账号管理 | 方法 | 路径 | 说明 | |------|------|------| -| GET | `/api/accounts` | 账号列表 | +| GET | `/api/accounts` | 账号列表（支持分页、筛选、搜索） | | GET | `/api/accounts/{id}` | 账号详情 | +| PATCH | `/api/accounts/{id}` | 更新账号（状态/cookies） | | DELETE | `/api/accounts/{id}` | 删除账号 | | POST | `/api/accounts/batch-delete` | 批量删除 | | POST | `/api/accounts/export/json` | 导出 JSON | | POST | `/api/accounts/export/csv` | 导出 CSV | | POST | `/api/accounts/export/cpa` | 导出 CPA 格式（单文件或 ZIP） | +| POST | `/api/accounts/export/sub2api` | 导出 Sub2API 格式 | | POST | `/api/accounts/{id}/refresh` | 刷新 Token | | POST | `/api/accounts/batch-refresh` | 批量刷新 Token | | POST | `/api/accounts/{id}/validate` | 验证 Token | | POST | `/api/accounts/batch-validate` | 批量验证 Token | -| POST | `/api/accounts/{id}/upload-cpa` | 上传到 CPA | +| POST | `/api/accounts/{id}/upload-cpa` | 上传单账号到 CPA | | POST | `/api/accounts/batch-upload-cpa` | 批量上传到 CPA | +| POST | `/api/accounts/{id}/upload-sub2api` | 上传单账号到 Sub2API | +| POST | `/api/accounts/batch-upload-sub2api` | 批量上传到 Sub2API | ### 支付升级 | 方法 | 路径 | 说明 | |------|------|------| -| POST | `/api/payment/generate-link` | 生成 Plus/Team 支付链接 | -| POST | `/api/payment/open-incognito` | 后端无痕模式打开浏览器 | +| POST | `/api/payment/generate` | 生成 Plus/Team 支付链接 | +| POST | `/api/payment/open` | 后端无痕模式打开浏览器 | | POST | `/api/payment/accounts/{id}/mark-subscription` | 手动标记订阅类型 | | POST | `/api/payment/accounts/batch-check-subscription` | 批量检测订阅状态 | | POST | `/api/payment/accounts/{id}/upload-tm` | 上传单账号到 Team Manager | @@ -233,31 +234,40 @@ codex-register-v2/ |------|------|------| | GET | `/api/email-services` | 服务列表 | | POST | `/api/email-services` | 添加服务 | -| GET | `/api/email-services/{id}` | 服务详情 | | PATCH | `/api/email-services/{id}` | 更新服务 | | DELETE | `/api/email-services/{id}` | 删除服务 | | POST | `/api/email-services/{id}/test` | 测试服务 | | POST | `/api/email-services/outlook/batch-import` | 批量导入 Outlook | +### 上传服务管理 + +| 方法 | 路径 | 说明 | +|------|------|------| +| GET/POST | `/api/cpa-services` | CPA 服务列表/创建 | +| PUT/DELETE | `/api/cpa-services/{id}` | 更新/删除 CPA 服务 | +| POST | `/api/cpa-services/{id}/test` | 测试 CPA 连接 | +| GET/POST | `/api/sub2api-services` | Sub2API 服务列表/创建 | +| PUT/DELETE | `/api/sub2api-services/{id}` | 更新/删除 Sub2API 服务 | +| POST | `/api/sub2api-services/{id}/test` | 测试 Sub2API 连接 | +| GET/POST | `/api/tm-services` | Team Manager 服务列表/创建 | +| PUT/DELETE | `/api/tm-services/{id}` | 更新/删除 TM 服务 | +| POST | `/api/tm-services/{id}/test` | 测试 TM 连接 | + ### 设置 | 方法 | 路径 | 说明 | |------|------|------| | GET | `/api/settings` | 获取所有设置 | -| POST | `/api/settings/proxy` | 更新代理设置 | -| POST | `/api/settings/dynamic-proxy` | 更新动态代理设置 | -| POST | `/api/settings/cpa` | 更新 CPA 设置 | -| POST | `/api/settings/cpa/test` | 测试 CPA 连接 | -| GET/POST | `/api/settings/team-manager` | Team Manager 设置 | -| POST | `/api/settings/team-manager/test` | 测试 Team Manager 连接 | +| POST | `/api/settings/proxy/dynamic` | 更新动态代理设置 | +| GET/POST/DELETE | `/api/settings/proxies` | 代理列表管理 | +| POST | `/api/settings/proxies/{id}/set-default` | 设为默认代理 | | GET | `/api/settings/database` | 数据库信息 | ### WebSocket | 路径 | 说明 | |------|------| -| `ws://host/api/ws/task/{uuid}` | 单任务实时日志 | -| `ws://host/api/ws/batch/{id}` | 批量任务实时状态与日志 | +|| `ws://host/api/ws/logs/{uuid}` | 实时日志流 | ## Docker 部署 @@ -290,11 +300,12 @@ volumes: - ./logs:/app/logs ``` -**代理配置**： +**环境变量配置**： ```yaml environment: - - HTTP_PROXY=http://your-proxy:port - - HTTPS_PROXY=http://your-proxy:port + - APP_ACCESS_PASSWORD=mypassword + - APP_HOST=0.0.0.0 + - APP_PORT=8000 ``` ### 常用命令 @@ -315,13 +326,12 @@ docker-compose build --no-cache - 首次运行会自动创建 `data/` 目录和 SQLite 数据库 - 所有账号和设置数据存储在 `data/register.db` - 日志文件写入 `logs/` 目录 -- 代理设置优先级：动态代理 > 代理列表（随机） > 静态默认代理 +- 代理优先级：动态代理 > 代理列表（随机/默认） > 直连 +- CPA / Sub2API / Team Manager 上传始终直连，不走代理 - 注册时自动随机生成用户名和生日（年龄范围 18-45 岁） -- CPA 上传始终直连，不经过代理 -- Team Manager 上传始终直连，不经过代理 - 支付链接生成使用账号 access_token 鉴权，走全局代理配置 - 无痕浏览器优先使用 playwright（注入 cookie 直达支付页）；未安装时降级为系统 Chrome/Edge 无痕模式 -- 安装完整支付功能：`pip install playwright && playwright install chromium`（可选） +- 安装完整支付功能：`pip install ".[payment]" && playwright install chromium`（可选） - 订阅状态自动检测调用 `chatgpt.com/backend-api/me`，走全局代理 - 批量注册并发数上限为 50，线程池大小已相应调整 diff --git a/src/core/upload/team_manager_upload.py b/src/core/upload/team_manager_upload.py index 267e2a2..d42d587 100644 --- a/src/core/upload/team_manager_upload.py +++ b/src/core/upload/team_manager_upload.py @@ -5,13 +5,11 @@ Team Manager 上传功能 import logging from typing import List, Tuple -from datetime import datetime from curl_cffi import requests as cffi_requests -from ...database.session import get_db from ...database.models import Account -from ...config.settings import get_settings +from ...database.session import get_db logger = logging.getLogger(__name__) @@ -34,7 +32,7 @@ def upload_to_team_manager( if not account.access_token: return False, "账号缺少 access_token" - url = api_url.rstrip("/") + "/api/accounts/import" + url = api_url.rstrip("/") + "/admin/teams/import" headers = { "X-API-Key": api_key, "Content-Type": "application/json", @@ -46,6 +44,7 @@ def upload_to_team_manager( "session_token": account.session_token or "", "refresh_token": account.refresh_token or "", "client_id": account.client_id or "", + "account_id": account.account_id or "", } try: @@ -78,7 +77,7 @@ def batch_upload_to_team_manager( api_key: str, ) -> dict: """ - 批量上传账号到 Team Manager + 批量上传账号到 Team Manager（使用 batch 模式，一次请求提交所有账号） Returns: 包含成功/失败统计和详情的字典 @@ -91,6 +90,8 @@ def batch_upload_to_team_manager( } with get_db() as db: + lines = [] + valid_accounts = [] for account_id in account_ids: account = db.query(Account).filter(Account.id == account_id).first() if not account: @@ -99,24 +100,70 @@ def batch_upload_to_team_manager( {"id": account_id, "email": None, "success": False, "error": "账号不存在"} ) continue - if not account.access_token: results["skipped_count"] += 1 results["details"].append( {"id": account_id, "email": account.email, "success": False, "error": "缺少 Token"} ) continue + # 格式：邮箱,AT,RT,ST,ClientID + lines.append(",".join([ + account.email or "", + account.access_token or "", + account.refresh_token or "", + account.session_token or "", + account.client_id or "", + ])) + valid_accounts.append(account) - success, message = upload_to_team_manager(account, api_url, api_key) - if success: - results["success_count"] += 1 - results["details"].append( - {"id": account_id, "email": account.email, "success": True, "message": message} - ) + if not valid_accounts: + return results + + url = api_url.rstrip("/") + "/admin/teams/import" + headers = { + "X-API-Key": api_key, + "Content-Type": "application/json", + } + payload = { + "import_type": "batch", + "content": "\n".join(lines), + } + + try: + resp = cffi_requests.post( + url, + headers=headers, + json=payload, + proxies=None, + timeout=60, + impersonate="chrome110", + ) + if resp.status_code in (200, 201): + for account in valid_accounts: + results["success_count"] += 1 + results["details"].append( + {"id": account.id, "email": account.email, "success": True, "message": "批量上传成功"} + ) else: + error_msg = f"批量上传失败: HTTP {resp.status_code}" + try: + detail = resp.json() + if isinstance(detail, dict): + error_msg = detail.get("message", error_msg) + except Exception: + error_msg = f"{error_msg} - {resp.text[:200]}" + for account in valid_accounts: + results["failed_count"] += 1 + results["details"].append( + {"id": account.id, "email": account.email, "success": False, "error": error_msg} + ) + except Exception as e: + logger.error(f"Team Manager 批量上传异常: {e}") + error_msg = f"上传异常: {str(e)}" + for account in valid_accounts: results["failed_count"] += 1 results["details"].append( - {"id": account_id, "email": account.email, "success": False, "error": message} + {"id": account.id, "email": account.email, "success": False, "error": error_msg} ) return results @@ -134,7 +181,7 @@ def test_team_manager_connection(api_url: str, api_key: str) -> Tuple[bool, str] if not api_key: return False, "API Key 不能为空" - url = api_url.rstrip("/") + "/api/accounts/import" + url = api_url.rstrip("/") + "/admin/teams/import" headers = {"X-API-Key": api_key} try: