feat(cpa): 支持多cpa服务

This commit is contained in:
cnlimiter
2026-03-18 14:01:44 +08:00
parent 7ce403ece3
commit 931ea798cc
14 changed files with 778 additions and 51 deletions

View File

@@ -39,33 +39,48 @@ def generate_token_json(account: Account) -> dict:
} }
def upload_to_cpa(token_data: dict, proxy: str = None) -> Tuple[bool, str]: def upload_to_cpa(
token_data: dict,
proxy: str = None,
api_url: str = None,
api_token: str = None,
) -> Tuple[bool, str]:
""" """
上传单个账号到 CPA 管理平台(不走代理) 上传单个账号到 CPA 管理平台(不走代理)
Args: Args:
token_data: Token JSON 数据 token_data: Token JSON 数据
proxy: 保留参数不使用CPA 上传始终直连) proxy: 保留参数不使用CPA 上传始终直连)
api_url: 指定 CPA API URL优先于全局配置
api_token: 指定 CPA API Token优先于全局配置
Returns: Returns:
(成功标志, 消息或错误信息) (成功标志, 消息或错误信息)
""" """
settings = get_settings() settings = get_settings()
if not settings.cpa_enabled: # 优先使用传入的参数,否则退回全局配置
effective_url = api_url or settings.cpa_api_url
effective_token = api_token or (settings.cpa_api_token.get_secret_value() if settings.cpa_api_token else "")
# 仅当未指定服务时才检查全局启用开关
if not api_url and not settings.cpa_enabled:
return False, "CPA 上传未启用" return False, "CPA 上传未启用"
if not settings.cpa_api_url: if not effective_url:
return False, "CPA API URL 未配置" return False, "CPA API URL 未配置"
api_url = settings.cpa_api_url.rstrip("/") if not effective_token:
return False, "CPA API Token 未配置"
api_url = effective_url.rstrip("/")
upload_url = f"{api_url}/v0/management/auth-files" upload_url = f"{api_url}/v0/management/auth-files"
filename = f"{token_data['email']}.json" filename = f"{token_data['email']}.json"
file_content = json.dumps(token_data, ensure_ascii=False, indent=2).encode("utf-8") file_content = json.dumps(token_data, ensure_ascii=False, indent=2).encode("utf-8")
headers = { headers = {
"Authorization": f"Bearer {settings.cpa_api_token.get_secret_value()}", "Authorization": f"Bearer {effective_token}",
} }
try: try:
@@ -103,13 +118,20 @@ def upload_to_cpa(token_data: dict, proxy: str = None) -> Tuple[bool, str]:
return False, f"上传异常: {str(e)}" return False, f"上传异常: {str(e)}"
def batch_upload_to_cpa(account_ids: List[int], proxy: str = None) -> dict: def batch_upload_to_cpa(
account_ids: List[int],
proxy: str = None,
api_url: str = None,
api_token: str = None,
) -> dict:
""" """
批量上传账号到 CPA 管理平台 批量上传账号到 CPA 管理平台
Args: Args:
account_ids: 账号 ID 列表 account_ids: 账号 ID 列表
proxy: 可选的代理 URL proxy: 可选的代理 URL
api_url: 指定 CPA API URL优先于全局配置
api_token: 指定 CPA API Token优先于全局配置
Returns: Returns:
包含成功/失败统计和详情的字典 包含成功/失败统计和详情的字典
@@ -150,7 +172,7 @@ def batch_upload_to_cpa(account_ids: List[int], proxy: str = None) -> dict:
token_data = generate_token_json(account) token_data = generate_token_json(account)
# 上传 # 上传
success, message = upload_to_cpa(token_data, proxy) success, message = upload_to_cpa(token_data, proxy, api_url=api_url, api_token=api_token)
if success: if success:
# 更新数据库状态 # 更新数据库状态

View File

@@ -7,7 +7,7 @@ from datetime import datetime, timedelta
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from sqlalchemy import and_, or_, desc, asc, func from sqlalchemy import and_, or_, desc, asc, func
from .models import Account, EmailService, RegistrationTask, Setting, Proxy from .models import Account, EmailService, RegistrationTask, Setting, Proxy, CpaService
# ============================================================================ # ============================================================================
@@ -497,4 +497,73 @@ def get_proxies_count(db: Session, enabled: Optional[bool] = None) -> int:
query = db.query(func.count(Proxy.id)) query = db.query(func.count(Proxy.id))
if enabled is not None: if enabled is not None:
query = query.filter(Proxy.enabled == enabled) query = query.filter(Proxy.enabled == enabled)
return query.scalar() return query.scalar()
# ============================================================================
# CPA 服务 CRUD
# ============================================================================
def create_cpa_service(
db: Session,
name: str,
api_url: str,
api_token: str,
enabled: bool = True,
priority: int = 0
) -> CpaService:
"""创建 CPA 服务配置"""
db_service = CpaService(
name=name,
api_url=api_url,
api_token=api_token,
enabled=enabled,
priority=priority
)
db.add(db_service)
db.commit()
db.refresh(db_service)
return db_service
def get_cpa_service_by_id(db: Session, service_id: int) -> Optional[CpaService]:
"""根据 ID 获取 CPA 服务"""
return db.query(CpaService).filter(CpaService.id == service_id).first()
def get_cpa_services(
db: Session,
enabled: Optional[bool] = None
) -> List[CpaService]:
"""获取 CPA 服务列表"""
query = db.query(CpaService)
if enabled is not None:
query = query.filter(CpaService.enabled == enabled)
return query.order_by(asc(CpaService.priority), asc(CpaService.id)).all()
def update_cpa_service(
db: Session,
service_id: int,
**kwargs
) -> Optional[CpaService]:
"""更新 CPA 服务配置"""
db_service = get_cpa_service_by_id(db, service_id)
if not db_service:
return None
for key, value in kwargs.items():
if hasattr(db_service, key):
setattr(db_service, key, value)
db.commit()
db.refresh(db_service)
return db_service
def delete_cpa_service(db: Session, service_id: int) -> bool:
"""删除 CPA 服务配置"""
db_service = get_cpa_service_by_id(db, service_id)
if not db_service:
return False
db.delete(db_service)
db.commit()
return True

View File

@@ -130,6 +130,20 @@ class Setting(Base):
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow) updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
class CpaService(Base):
"""CPA 服务配置表"""
__tablename__ = 'cpa_services'
id = Column(Integer, primary_key=True, autoincrement=True)
name = Column(String(100), nullable=False) # 服务名称
api_url = Column(String(500), nullable=False) # API URL
api_token = Column(Text, nullable=False) # API Token
enabled = Column(Boolean, default=True)
priority = Column(Integer, default=0) # 优先级
created_at = Column(DateTime, default=datetime.utcnow)
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
class Proxy(Base): class Proxy(Base):
"""代理列表表""" """代理列表表"""
__tablename__ = 'proxies' __tablename__ = 'proxies'

View File

@@ -112,6 +112,9 @@ class DatabaseSessionManager:
("accounts", "cookies", "TEXT"), ("accounts", "cookies", "TEXT"),
] ]
# 确保新表存在create_tables 已处理,此处兜底)
Base.metadata.create_all(bind=self.engine)
with self.engine.connect() as conn: with self.engine.connect() as conn:
for table_name, column_name, column_type in migrations: for table_name, column_name, column_type in migrations:
try: try:

View File

@@ -9,6 +9,7 @@ from .registration import router as registration_router
from .settings import router as settings_router from .settings import router as settings_router
from .email_services import router as email_services_router from .email_services import router as email_services_router
from .payment import router as payment_router from .payment import router as payment_router
from .cpa_services import router as cpa_services_router
api_router = APIRouter() api_router = APIRouter()
@@ -18,3 +19,4 @@ api_router.include_router(registration_router, prefix="/registration", tags=["re
api_router.include_router(settings_router, prefix="/settings", tags=["settings"]) api_router.include_router(settings_router, prefix="/settings", tags=["settings"])
api_router.include_router(email_services_router, prefix="/email-services", tags=["email-services"]) api_router.include_router(email_services_router, prefix="/email-services", tags=["email-services"])
api_router.include_router(payment_router, prefix="/payment", tags=["payment"]) api_router.include_router(payment_router, prefix="/payment", tags=["payment"])
api_router.include_router(cpa_services_router, prefix="/cpa-services", tags=["cpa-services"])

View File

@@ -700,6 +700,7 @@ async def batch_validate_tokens(request: BatchValidateRequest):
class CPAUploadRequest(BaseModel): class CPAUploadRequest(BaseModel):
"""CPA 上传请求""" """CPA 上传请求"""
proxy: Optional[str] = None proxy: Optional[str] = None
cpa_service_id: Optional[int] = None # 指定 CPA 服务 ID不传则使用全局配置
class BatchCPAUploadRequest(BaseModel): class BatchCPAUploadRequest(BaseModel):
@@ -710,6 +711,7 @@ class BatchCPAUploadRequest(BaseModel):
status_filter: Optional[str] = None status_filter: Optional[str] = None
email_service_filter: Optional[str] = None email_service_filter: Optional[str] = None
search_filter: Optional[str] = None search_filter: Optional[str] = None
cpa_service_id: Optional[int] = None # 指定 CPA 服务 ID不传则使用全局配置
@router.post("/{account_id}/upload-cpa") @router.post("/{account_id}/upload-cpa")
@@ -717,8 +719,19 @@ async def upload_account_to_cpa(account_id: int, request: CPAUploadRequest = Non
"""上传单个账号到 CPA""" """上传单个账号到 CPA"""
from ...core.cpa_upload import upload_to_cpa, generate_token_json from ...core.cpa_upload import upload_to_cpa, generate_token_json
# 使用传入的代理或全局代理配置
proxy = request.proxy if request and request.proxy else get_settings().proxy_url proxy = request.proxy if request and request.proxy else get_settings().proxy_url
cpa_service_id = request.cpa_service_id if request else None
# 解析指定的 CPA 服务
cpa_api_url = None
cpa_api_token = None
if cpa_service_id:
with get_db() as db:
svc = crud.get_cpa_service_by_id(db, cpa_service_id)
if not svc:
raise HTTPException(status_code=404, detail="指定的 CPA 服务不存在")
cpa_api_url = svc.api_url
cpa_api_token = svc.api_token
with get_db() as db: with get_db() as db:
account = crud.get_account_by_id(db, account_id) account = crud.get_account_by_id(db, account_id)
@@ -735,23 +748,15 @@ async def upload_account_to_cpa(account_id: int, request: CPAUploadRequest = Non
token_data = generate_token_json(account) token_data = generate_token_json(account)
# 上传 # 上传
success, message = upload_to_cpa(token_data, proxy) success, message = upload_to_cpa(token_data, proxy, api_url=cpa_api_url, api_token=cpa_api_token)
if success: if success:
# 更新数据库状态
account.cpa_uploaded = True account.cpa_uploaded = True
account.cpa_uploaded_at = datetime.utcnow() account.cpa_uploaded_at = datetime.utcnow()
db.commit() db.commit()
return {"success": True, "message": message}
return {
"success": True,
"message": message
}
else: else:
return { return {"success": False, "error": message}
"success": False,
"error": message
}
@router.post("/batch-upload-cpa") @router.post("/batch-upload-cpa")
@@ -759,15 +764,24 @@ async def batch_upload_accounts_to_cpa(request: BatchCPAUploadRequest):
"""批量上传账号到 CPA""" """批量上传账号到 CPA"""
from ...core.cpa_upload import batch_upload_to_cpa from ...core.cpa_upload import batch_upload_to_cpa
# 使用传入的代理或全局代理配置
proxy = request.proxy if request.proxy else get_settings().proxy_url proxy = request.proxy if request.proxy else get_settings().proxy_url
# 解析指定的 CPA 服务
cpa_api_url = None
cpa_api_token = None
if request.cpa_service_id:
with get_db() as db:
svc = crud.get_cpa_service_by_id(db, request.cpa_service_id)
if not svc:
raise HTTPException(status_code=404, detail="指定的 CPA 服务不存在")
cpa_api_url = svc.api_url
cpa_api_token = svc.api_token
with get_db() as db: with get_db() as db:
ids = resolve_account_ids( ids = resolve_account_ids(
db, request.ids, request.select_all, db, request.ids, request.select_all,
request.status_filter, request.email_service_filter, request.search_filter request.status_filter, request.email_service_filter, request.search_filter
) )
results = batch_upload_to_cpa(ids, proxy) results = batch_upload_to_cpa(ids, proxy, api_url=cpa_api_url, api_token=cpa_api_token)
return results return results

View File

@@ -0,0 +1,171 @@
"""
CPA 服务管理 API 路由
"""
from typing import List, Optional
from fastapi import APIRouter, HTTPException
from pydantic import BaseModel
from ...database import crud
from ...database.session import get_db
from ...core.cpa_upload import test_cpa_connection
router = APIRouter()
# ============== Pydantic Models ==============
class CpaServiceCreate(BaseModel):
name: str
api_url: str
api_token: str
enabled: bool = True
priority: int = 0
class CpaServiceUpdate(BaseModel):
name: Optional[str] = None
api_url: Optional[str] = None
api_token: Optional[str] = None
enabled: Optional[bool] = None
priority: Optional[int] = None
class CpaServiceResponse(BaseModel):
id: int
name: str
api_url: str
has_token: bool
enabled: bool
priority: int
created_at: Optional[str] = None
updated_at: Optional[str] = None
class Config:
from_attributes = True
class CpaServiceTestRequest(BaseModel):
api_url: Optional[str] = None
api_token: Optional[str] = None
def _to_response(svc) -> CpaServiceResponse:
return CpaServiceResponse(
id=svc.id,
name=svc.name,
api_url=svc.api_url,
has_token=bool(svc.api_token),
enabled=svc.enabled,
priority=svc.priority,
created_at=svc.created_at.isoformat() if svc.created_at else None,
updated_at=svc.updated_at.isoformat() if svc.updated_at else None,
)
# ============== API Endpoints ==============
@router.get("", response_model=List[CpaServiceResponse])
async def list_cpa_services(enabled: Optional[bool] = None):
"""获取 CPA 服务列表"""
with get_db() as db:
services = crud.get_cpa_services(db, enabled=enabled)
return [_to_response(s) for s in services]
@router.post("", response_model=CpaServiceResponse)
async def create_cpa_service(request: CpaServiceCreate):
"""新增 CPA 服务"""
with get_db() as db:
service = crud.create_cpa_service(
db,
name=request.name,
api_url=request.api_url,
api_token=request.api_token,
enabled=request.enabled,
priority=request.priority,
)
return _to_response(service)
@router.get("/{service_id}", response_model=CpaServiceResponse)
async def get_cpa_service(service_id: int):
"""获取单个 CPA 服务详情"""
with get_db() as db:
service = crud.get_cpa_service_by_id(db, service_id)
if not service:
raise HTTPException(status_code=404, detail="CPA 服务不存在")
return _to_response(service)
@router.get("/{service_id}/full")
async def get_cpa_service_full(service_id: int):
"""获取 CPA 服务完整配置(含 token"""
with get_db() as db:
service = crud.get_cpa_service_by_id(db, service_id)
if not service:
raise HTTPException(status_code=404, detail="CPA 服务不存在")
return {
"id": service.id,
"name": service.name,
"api_url": service.api_url,
"api_token": service.api_token,
"enabled": service.enabled,
"priority": service.priority,
}
@router.patch("/{service_id}", response_model=CpaServiceResponse)
async def update_cpa_service(service_id: int, request: CpaServiceUpdate):
"""更新 CPA 服务配置"""
with get_db() as db:
service = crud.get_cpa_service_by_id(db, service_id)
if not service:
raise HTTPException(status_code=404, detail="CPA 服务不存在")
update_data = {}
if request.name is not None:
update_data["name"] = request.name
if request.api_url is not None:
update_data["api_url"] = request.api_url
# api_token 留空则保持原值
if request.api_token:
update_data["api_token"] = request.api_token
if request.enabled is not None:
update_data["enabled"] = request.enabled
if request.priority is not None:
update_data["priority"] = request.priority
service = crud.update_cpa_service(db, service_id, **update_data)
return _to_response(service)
@router.delete("/{service_id}")
async def delete_cpa_service(service_id: int):
"""删除 CPA 服务"""
with get_db() as db:
service = crud.get_cpa_service_by_id(db, service_id)
if not service:
raise HTTPException(status_code=404, detail="CPA 服务不存在")
crud.delete_cpa_service(db, service_id)
return {"success": True, "message": f"CPA 服务 {service.name} 已删除"}
@router.post("/{service_id}/test")
async def test_cpa_service(service_id: int):
"""测试 CPA 服务连接"""
with get_db() as db:
service = crud.get_cpa_service_by_id(db, service_id)
if not service:
raise HTTPException(status_code=404, detail="CPA 服务不存在")
success, message = test_cpa_connection(service.api_url, service.api_token)
return {"success": success, "message": message}
@router.post("/test-connection")
async def test_cpa_connection_direct(request: CpaServiceTestRequest):
"""直接测试 CPA 连接(用于添加前验证)"""
if not request.api_url or not request.api_token:
raise HTTPException(status_code=400, detail="api_url 和 api_token 不能为空")
success, message = test_cpa_connection(request.api_url, request.api_token)
return {"success": success, "message": message}

View File

@@ -72,6 +72,7 @@ class RegistrationTaskCreate(BaseModel):
email_service_config: Optional[dict] = None email_service_config: Optional[dict] = None
email_service_id: Optional[int] = None # 使用数据库中已配置的邮箱服务 ID email_service_id: Optional[int] = None # 使用数据库中已配置的邮箱服务 ID
auto_upload_cpa: bool = False # 注册成功后自动上传到 CPA auto_upload_cpa: bool = False # 注册成功后自动上传到 CPA
cpa_service_id: Optional[int] = None # 指定 CPA 服务 ID不传则使用全局配置
class BatchRegistrationRequest(BaseModel): class BatchRegistrationRequest(BaseModel):
@@ -86,6 +87,7 @@ class BatchRegistrationRequest(BaseModel):
concurrency: int = 1 # 并发线程数 (1-50) concurrency: int = 1 # 并发线程数 (1-50)
mode: str = "pipeline" # 执行模式: "parallel" 或 "pipeline" mode: str = "pipeline" # 执行模式: "parallel" 或 "pipeline"
auto_upload_cpa: bool = False # 注册成功后自动上传到 CPA auto_upload_cpa: bool = False # 注册成功后自动上传到 CPA
cpa_service_id: Optional[int] = None # 指定 CPA 服务 ID不传则使用全局配置
class RegistrationTaskResponse(BaseModel): class RegistrationTaskResponse(BaseModel):
@@ -149,6 +151,7 @@ class OutlookBatchRegistrationRequest(BaseModel):
concurrency: int = 1 # 并发线程数 (1-50) concurrency: int = 1 # 并发线程数 (1-50)
mode: str = "pipeline" # 执行模式: "parallel" 或 "pipeline" mode: str = "pipeline" # 执行模式: "parallel" 或 "pipeline"
auto_upload_cpa: bool = False # 注册成功后自动上传到 CPA auto_upload_cpa: bool = False # 注册成功后自动上传到 CPA
cpa_service_id: Optional[int] = None # 指定 CPA 服务 ID不传则使用全局配置
class OutlookBatchRegistrationResponse(BaseModel): class OutlookBatchRegistrationResponse(BaseModel):
@@ -179,7 +182,7 @@ def task_to_response(task: RegistrationTask) -> RegistrationTaskResponse:
) )
def _run_sync_registration_task(task_uuid: str, email_service_type: str, proxy: Optional[str], email_service_config: Optional[dict], email_service_id: Optional[int] = None, log_prefix: str = "", batch_id: str = "", auto_upload_cpa: bool = False): def _run_sync_registration_task(task_uuid: str, email_service_type: str, proxy: Optional[str], email_service_config: Optional[dict], email_service_id: Optional[int] = None, log_prefix: str = "", batch_id: str = "", auto_upload_cpa: bool = False, cpa_service_id: Optional[int] = None):
""" """
在线程池中执行的同步注册任务 在线程池中执行的同步注册任务
@@ -344,7 +347,19 @@ def _run_sync_registration_task(task_uuid: str, email_service_type: str, proxy:
saved_account = db.query(AccountModel).filter_by(email=result.email).first() saved_account = db.query(AccountModel).filter_by(email=result.email).first()
if saved_account and saved_account.access_token: if saved_account and saved_account.access_token:
token_data = generate_token_json(saved_account) token_data = generate_token_json(saved_account)
cpa_success, cpa_msg = upload_to_cpa(token_data) # 解析指定 CPA 服务
_cpa_api_url = None
_cpa_api_token = None
if cpa_service_id:
try:
_svc = crud.get_cpa_service_by_id(db, cpa_service_id)
if _svc:
_cpa_api_url = _svc.api_url
_cpa_api_token = _svc.api_token
log_callback(f"[CPA] 使用服务: {_svc.name}")
except Exception:
pass
cpa_success, cpa_msg = upload_to_cpa(token_data, api_url=_cpa_api_url, api_token=_cpa_api_token)
if cpa_success: if cpa_success:
saved_account.cpa_uploaded = True saved_account.cpa_uploaded = True
saved_account.cpa_uploaded_at = datetime.utcnow() saved_account.cpa_uploaded_at = datetime.utcnow()
@@ -399,7 +414,7 @@ def _run_sync_registration_task(task_uuid: str, email_service_type: str, proxy:
pass pass
async def run_registration_task(task_uuid: str, email_service_type: str, proxy: Optional[str], email_service_config: Optional[dict], email_service_id: Optional[int] = None, log_prefix: str = "", batch_id: str = "", auto_upload_cpa: bool = False): async def run_registration_task(task_uuid: str, email_service_type: str, proxy: Optional[str], email_service_config: Optional[dict], email_service_id: Optional[int] = None, log_prefix: str = "", batch_id: str = "", auto_upload_cpa: bool = False, cpa_service_id: Optional[int] = None):
""" """
异步执行注册任务 异步执行注册任务
@@ -426,7 +441,8 @@ async def run_registration_task(task_uuid: str, email_service_type: str, proxy:
email_service_id, email_service_id,
log_prefix, log_prefix,
batch_id, batch_id,
auto_upload_cpa auto_upload_cpa,
cpa_service_id
) )
except Exception as e: except Exception as e:
logger.error(f"线程池执行异常: {task_uuid}, 错误: {e}") logger.error(f"线程池执行异常: {task_uuid}, 错误: {e}")
@@ -473,7 +489,8 @@ async def run_batch_parallel(
email_service_config: Optional[dict], email_service_config: Optional[dict],
email_service_id: Optional[int], email_service_id: Optional[int],
concurrency: int, concurrency: int,
auto_upload_cpa: bool = False auto_upload_cpa: bool = False,
cpa_service_id: Optional[int] = None
): ):
""" """
并行模式所有任务同时提交Semaphore 控制最大并发数 并行模式所有任务同时提交Semaphore 控制最大并发数
@@ -489,7 +506,8 @@ async def run_batch_parallel(
async with semaphore: async with semaphore:
await run_registration_task( await run_registration_task(
uuid, email_service_type, proxy, email_service_config, email_service_id, uuid, email_service_type, proxy, email_service_config, email_service_id,
log_prefix=prefix, batch_id=batch_id, auto_upload_cpa=auto_upload_cpa log_prefix=prefix, batch_id=batch_id, auto_upload_cpa=auto_upload_cpa,
cpa_service_id=cpa_service_id
) )
with get_db() as db: with get_db() as db:
t = crud.get_registration_task(db, uuid) t = crud.get_registration_task(db, uuid)
@@ -531,7 +549,8 @@ async def run_batch_pipeline(
interval_min: int, interval_min: int,
interval_max: int, interval_max: int,
concurrency: int, concurrency: int,
auto_upload_cpa: bool = False auto_upload_cpa: bool = False,
cpa_service_id: Optional[int] = None
): ):
""" """
流水线模式:每隔 interval 秒启动一个新任务Semaphore 限制最大并发数 流水线模式:每隔 interval 秒启动一个新任务Semaphore 限制最大并发数
@@ -547,7 +566,8 @@ async def run_batch_pipeline(
try: try:
await run_registration_task( await run_registration_task(
uuid, email_service_type, proxy, email_service_config, email_service_id, uuid, email_service_type, proxy, email_service_config, email_service_id,
log_prefix=pfx, batch_id=batch_id, auto_upload_cpa=auto_upload_cpa log_prefix=pfx, batch_id=batch_id, auto_upload_cpa=auto_upload_cpa,
cpa_service_id=cpa_service_id
) )
with get_db() as db: with get_db() as db:
t = crud.get_registration_task(db, uuid) t = crud.get_registration_task(db, uuid)
@@ -613,21 +633,22 @@ async def run_batch_registration(
interval_max: int, interval_max: int,
concurrency: int = 1, concurrency: int = 1,
mode: str = "pipeline", mode: str = "pipeline",
auto_upload_cpa: bool = False auto_upload_cpa: bool = False,
cpa_service_id: Optional[int] = None
): ):
"""根据 mode 分发到并行或流水线执行""" """根据 mode 分发到并行或流水线执行"""
if mode == "parallel": if mode == "parallel":
await run_batch_parallel( await run_batch_parallel(
batch_id, task_uuids, email_service_type, proxy, batch_id, task_uuids, email_service_type, proxy,
email_service_config, email_service_id, concurrency, email_service_config, email_service_id, concurrency,
auto_upload_cpa=auto_upload_cpa auto_upload_cpa=auto_upload_cpa, cpa_service_id=cpa_service_id
) )
else: else:
await run_batch_pipeline( await run_batch_pipeline(
batch_id, task_uuids, email_service_type, proxy, batch_id, task_uuids, email_service_type, proxy,
email_service_config, email_service_id, email_service_config, email_service_id,
interval_min, interval_max, concurrency, interval_min, interval_max, concurrency,
auto_upload_cpa=auto_upload_cpa auto_upload_cpa=auto_upload_cpa, cpa_service_id=cpa_service_id
) )
@@ -674,7 +695,8 @@ async def start_registration(
request.email_service_id, request.email_service_id,
"", "",
"", "",
request.auto_upload_cpa request.auto_upload_cpa,
request.cpa_service_id
) )
return task_to_response(task) return task_to_response(task)
@@ -746,7 +768,8 @@ async def start_batch_registration(
request.interval_max, request.interval_max,
request.concurrency, request.concurrency,
request.mode, request.mode,
request.auto_upload_cpa request.auto_upload_cpa,
request.cpa_service_id
) )
return BatchRegistrationResponse( return BatchRegistrationResponse(
@@ -1075,7 +1098,8 @@ async def run_outlook_batch_registration(
interval_max: int, interval_max: int,
concurrency: int = 1, concurrency: int = 1,
mode: str = "pipeline", mode: str = "pipeline",
auto_upload_cpa: bool = False auto_upload_cpa: bool = False,
cpa_service_id: Optional[int] = None
): ):
""" """
异步执行 Outlook 批量注册任务,复用通用并发逻辑 异步执行 Outlook 批量注册任务,复用通用并发逻辑
@@ -1113,7 +1137,8 @@ async def run_outlook_batch_registration(
interval_max=interval_max, interval_max=interval_max,
concurrency=concurrency, concurrency=concurrency,
mode=mode, mode=mode,
auto_upload_cpa=auto_upload_cpa auto_upload_cpa=auto_upload_cpa,
cpa_service_id=cpa_service_id
) )
@@ -1212,7 +1237,8 @@ async def start_outlook_batch_registration(
request.interval_max, request.interval_max,
request.concurrency, request.concurrency,
request.mode, request.mode,
request.auto_upload_cpa request.auto_upload_cpa,
request.cpa_service_id
) )
return OutlookBatchRegistrationResponse( return OutlookBatchRegistrationResponse(

View File

@@ -740,11 +740,86 @@ function escapeHtml(text) {
return div.innerHTML; return div.innerHTML;
} }
// ============== CPA 服务选择 ==============
// 弹出 CPA 服务选择框,返回 Promise<{cpa_service_id: number|null}|null>
// null 表示用户取消,{cpa_service_id: null} 表示使用全局配置
function selectCpaService() {
return new Promise(async (resolve) => {
const modal = document.getElementById('cpa-service-modal');
const listEl = document.getElementById('cpa-service-list');
const closeBtn = document.getElementById('close-cpa-modal');
const cancelBtn = document.getElementById('cancel-cpa-modal-btn');
const globalBtn = document.getElementById('cpa-use-global-btn');
// 加载服务列表
listEl.innerHTML = '<div style="text-align:center;color:var(--text-muted)">加载中...</div>';
modal.classList.add('active');
let services = [];
try {
services = await api.get('/cpa-services?enabled=true');
} catch (e) {
services = [];
}
if (services.length === 0) {
listEl.innerHTML = '<div style="text-align:center;color:var(--text-muted);padding:12px;">暂无已启用的 CPA 服务,将使用全局配置</div>';
} else {
listEl.innerHTML = services.map(s => `
<div class="cpa-service-item" data-id="${s.id}" style="
padding: 10px 14px;
border: 1px solid var(--border);
border-radius: 8px;
cursor: pointer;
transition: background 0.15s;
display: flex;
justify-content: space-between;
align-items: center;
">
<div>
<div style="font-weight:500;">${escapeHtml(s.name)}</div>
<div style="font-size:0.8rem;color:var(--text-muted);">${escapeHtml(s.api_url)}</div>
</div>
<span class="badge" style="background:var(--success-color);color:#fff;font-size:0.7rem;padding:2px 8px;border-radius:10px;">选择</span>
</div>
`).join('');
listEl.querySelectorAll('.cpa-service-item').forEach(item => {
item.addEventListener('mouseenter', () => item.style.background = 'var(--surface-hover)');
item.addEventListener('mouseleave', () => item.style.background = '');
item.addEventListener('click', () => {
cleanup();
resolve({ cpa_service_id: parseInt(item.dataset.id) });
});
});
}
function cleanup() {
modal.classList.remove('active');
closeBtn.removeEventListener('click', onCancel);
cancelBtn.removeEventListener('click', onCancel);
globalBtn.removeEventListener('click', onGlobal);
}
function onCancel() { cleanup(); resolve(null); }
function onGlobal() { cleanup(); resolve({ cpa_service_id: null }); }
closeBtn.addEventListener('click', onCancel);
cancelBtn.addEventListener('click', onCancel);
globalBtn.addEventListener('click', onGlobal);
});
}
// 上传单个账号到CPA // 上传单个账号到CPA
async function uploadToCpa(id) { async function uploadToCpa(id) {
const choice = await selectCpaService();
if (choice === null) return; // 用户取消
try { try {
toast.info('正在上传到CPA...'); toast.info('正在上传到CPA...');
const result = await api.post(`/accounts/${id}/upload-cpa`); const payload = {};
if (choice.cpa_service_id != null) payload.cpa_service_id = choice.cpa_service_id;
const result = await api.post(`/accounts/${id}/upload-cpa`, payload);
if (result.success) { if (result.success) {
toast.success('上传成功'); toast.success('上传成功');
@@ -762,6 +837,9 @@ async function handleBatchUploadCpa() {
const count = getEffectiveCount(); const count = getEffectiveCount();
if (count === 0) return; if (count === 0) return;
const choice = await selectCpaService();
if (choice === null) return; // 用户取消
const confirmed = await confirm(`确定要将选中的 ${count} 个账号上传到CPA吗`); const confirmed = await confirm(`确定要将选中的 ${count} 个账号上传到CPA吗`);
if (!confirmed) return; if (!confirmed) return;
@@ -769,15 +847,13 @@ async function handleBatchUploadCpa() {
elements.batchUploadCpaBtn.textContent = '上传中...'; elements.batchUploadCpaBtn.textContent = '上传中...';
try { try {
const result = await api.post('/accounts/batch-upload-cpa', buildBatchPayload()); const payload = buildBatchPayload();
if (choice.cpa_service_id != null) payload.cpa_service_id = choice.cpa_service_id;
const result = await api.post('/accounts/batch-upload-cpa', payload);
let message = `成功: ${result.success_count}`; let message = `成功: ${result.success_count}`;
if (result.failed_count > 0) { if (result.failed_count > 0) message += `, 失败: ${result.failed_count}`;
message += `, 失败: ${result.failed_count}`; if (result.skipped_count > 0) message += `, 跳过: ${result.skipped_count}`;
}
if (result.skipped_count > 0) {
message += `, 跳过: ${result.skipped_count}`;
}
toast.success(message); toast.success(message);
loadAccounts(); loadAccounts();

View File

@@ -83,7 +83,9 @@ const elements = {
concurrencyHint: document.getElementById('concurrency-hint'), concurrencyHint: document.getElementById('concurrency-hint'),
intervalGroup: document.getElementById('interval-group'), intervalGroup: document.getElementById('interval-group'),
// 注册后自动操作 // 注册后自动操作
autoUploadCpa: document.getElementById('auto-upload-cpa') autoUploadCpa: document.getElementById('auto-upload-cpa'),
cpaServiceSelectGroup: document.getElementById('cpa-service-select-group'),
cpaServiceSelect: document.getElementById('cpa-service-select')
}; };
// 初始化 // 初始化
@@ -97,7 +99,7 @@ document.addEventListener('DOMContentLoaded', () => {
checkCpaEnabled(); checkCpaEnabled();
}); });
// 检查 CPA 是否启用,未启用则禁用复选框 // 检查 CPA 是否启用,未启用则禁用复选框;同时加载 CPA 服务列表
async function checkCpaEnabled() { async function checkCpaEnabled() {
if (!elements.autoUploadCpa) return; if (!elements.autoUploadCpa) return;
try { try {
@@ -111,6 +113,32 @@ async function checkCpaEnabled() {
} catch (e) { } catch (e) {
elements.autoUploadCpa.disabled = true; elements.autoUploadCpa.disabled = true;
} }
// 加载 CPA 服务列表
await loadCpaServiceOptions();
// 复选框联动显示/隐藏服务选择器
if (elements.autoUploadCpa) {
elements.autoUploadCpa.addEventListener('change', () => {
if (elements.cpaServiceSelectGroup) {
elements.cpaServiceSelectGroup.style.display =
elements.autoUploadCpa.checked ? 'block' : 'none';
}
});
}
}
async function loadCpaServiceOptions() {
if (!elements.cpaServiceSelect) return;
try {
const services = await api.get('/cpa-services?enabled=true');
// 保留「使用全局配置」选项
const defaultOpt = '<option value="">使用全局配置</option>';
const opts = services.map(s =>
`<option value="${s.id}">${s.name.replace(/</g,'&lt;')}</option>`
).join('');
elements.cpaServiceSelect.innerHTML = defaultOpt + opts;
} catch (e) {
// 加载失败静默处理,保持默认选项
}
} }
// 事件监听 // 事件监听
@@ -357,6 +385,10 @@ async function handleStartRegistration(e) {
email_service_type: emailServiceType, email_service_type: emailServiceType,
auto_upload_cpa: elements.autoUploadCpa ? elements.autoUploadCpa.checked : false auto_upload_cpa: elements.autoUploadCpa ? elements.autoUploadCpa.checked : false
}; };
// 带上指定 CPA 服务
if (requestData.auto_upload_cpa && elements.cpaServiceSelect && elements.cpaServiceSelect.value) {
requestData.cpa_service_id = parseInt(elements.cpaServiceSelect.value);
}
// 如果选择了数据库中的服务,传递 service_id // 如果选择了数据库中的服务,传递 service_id
if (serviceId && serviceId !== 'default') { if (serviceId && serviceId !== 'default') {

View File

@@ -44,6 +44,15 @@ const elements = {
// CPA 设置 // CPA 设置
cpaForm: document.getElementById('cpa-form'), cpaForm: document.getElementById('cpa-form'),
testCpaBtn: document.getElementById('test-cpa-btn'), testCpaBtn: document.getElementById('test-cpa-btn'),
// CPA 服务管理
addCpaServiceBtn: document.getElementById('add-cpa-service-btn'),
cpaServicesTable: document.getElementById('cpa-services-table'),
cpaServiceEditModal: document.getElementById('cpa-service-edit-modal'),
closeCpaServiceModal: document.getElementById('close-cpa-service-modal'),
cancelCpaServiceBtn: document.getElementById('cancel-cpa-service-btn'),
cpaServiceForm: document.getElementById('cpa-service-form'),
cpaServiceModalTitle: document.getElementById('cpa-service-modal-title'),
testCpaServiceBtn: document.getElementById('test-cpa-service-btn'),
// Team Manager 设置 // Team Manager 设置
tmForm: document.getElementById('tm-form'), tmForm: document.getElementById('tm-form'),
testTmBtn: document.getElementById('test-tm-btn'), testTmBtn: document.getElementById('test-tm-btn'),
@@ -65,6 +74,7 @@ document.addEventListener('DOMContentLoaded', () => {
loadEmailServices(); loadEmailServices();
loadDatabaseInfo(); loadDatabaseInfo();
loadProxies(); loadProxies();
loadCpaServices();
initEventListeners(); initEventListeners();
}); });
@@ -247,6 +257,28 @@ function initEventListeners() {
if (elements.testTmBtn) { if (elements.testTmBtn) {
elements.testTmBtn.addEventListener('click', handleTestTm); elements.testTmBtn.addEventListener('click', handleTestTm);
} }
// CPA 服务管理
if (elements.addCpaServiceBtn) {
elements.addCpaServiceBtn.addEventListener('click', () => openCpaServiceModal());
}
if (elements.closeCpaServiceModal) {
elements.closeCpaServiceModal.addEventListener('click', closeCpaServiceModal);
}
if (elements.cancelCpaServiceBtn) {
elements.cancelCpaServiceBtn.addEventListener('click', closeCpaServiceModal);
}
if (elements.cpaServiceEditModal) {
elements.cpaServiceEditModal.addEventListener('click', (e) => {
if (e.target === elements.cpaServiceEditModal) closeCpaServiceModal();
});
}
if (elements.cpaServiceForm) {
elements.cpaServiceForm.addEventListener('submit', handleSaveCpaService);
}
if (elements.testCpaServiceBtn) {
elements.testCpaServiceBtn.addEventListener('click', handleTestCpaService);
}
} }
// 加载设置 // 加载设置
@@ -1182,3 +1214,172 @@ async function handleTestTm() {
elements.testTmBtn.textContent = '🔌 测试连接'; elements.testTmBtn.textContent = '🔌 测试连接';
} }
} }
// ============== CPA 服务管理 ==============
async function loadCpaServices() {
if (!elements.cpaServicesTable) return;
try {
const services = await api.get('/cpa-services');
renderCpaServicesTable(services);
} catch (e) {
elements.cpaServicesTable.innerHTML = `<tr><td colspan="5" style="text-align:center;color:var(--danger-color);">${e.message}</td></tr>`;
}
}
function renderCpaServicesTable(services) {
if (!services || services.length === 0) {
elements.cpaServicesTable.innerHTML = '<tr><td colspan="5" style="text-align:center;color:var(--text-muted);padding:20px;">暂无 CPA 服务,点击「添加服务」新增</td></tr>';
return;
}
elements.cpaServicesTable.innerHTML = services.map(s => `
<tr>
<td>${escapeHtml(s.name)}</td>
<td style="font-size:0.85rem;color:var(--text-muted);">${escapeHtml(s.api_url)}</td>
<td>
<span class="badge" style="background:${s.enabled ? 'var(--success-color)' : 'var(--border)'};color:${s.enabled ? '#fff' : 'var(--text-muted)'};font-size:0.75rem;padding:2px 8px;border-radius:10px;">
${s.enabled ? '启用' : '禁用'}
</span>
</td>
<td style="text-align:center;">${s.priority}</td>
<td>
<button class="btn btn-secondary btn-sm" onclick="editCpaService(${s.id})">编辑</button>
<button class="btn btn-secondary btn-sm" onclick="testCpaServiceById(${s.id})">测试</button>
<button class="btn btn-danger btn-sm" onclick="deleteCpaService(${s.id}, '${escapeHtml(s.name)}')">删除</button>
</td>
</tr>
`).join('');
}
function openCpaServiceModal(service = null) {
document.getElementById('cpa-service-id').value = service ? service.id : '';
document.getElementById('cpa-service-name').value = service ? service.name : '';
document.getElementById('cpa-service-url').value = service ? service.api_url : '';
document.getElementById('cpa-service-token').value = '';
document.getElementById('cpa-service-priority').value = service ? service.priority : 0;
document.getElementById('cpa-service-enabled').checked = service ? service.enabled : true;
elements.cpaServiceModalTitle.textContent = service ? '编辑 CPA 服务' : '添加 CPA 服务';
elements.cpaServiceEditModal.classList.add('active');
}
function closeCpaServiceModal() {
elements.cpaServiceEditModal.classList.remove('active');
}
async function editCpaService(id) {
try {
const service = await api.get(`/cpa-services/${id}`);
openCpaServiceModal(service);
} catch (e) {
toast.error('获取服务信息失败: ' + e.message);
}
}
async function handleSaveCpaService(e) {
e.preventDefault();
const id = document.getElementById('cpa-service-id').value;
const name = document.getElementById('cpa-service-name').value.trim();
const apiUrl = document.getElementById('cpa-service-url').value.trim();
const apiToken = document.getElementById('cpa-service-token').value.trim();
const priority = parseInt(document.getElementById('cpa-service-priority').value) || 0;
const enabled = document.getElementById('cpa-service-enabled').checked;
if (!name || !apiUrl) {
toast.error('名称和 API URL 不能为空');
return;
}
if (!id && !apiToken) {
toast.error('新增服务时 API Token 不能为空');
return;
}
try {
const payload = { name, api_url: apiUrl, priority, enabled };
if (apiToken) payload.api_token = apiToken;
if (id) {
await api.patch(`/cpa-services/${id}`, payload);
toast.success('服务已更新');
} else {
payload.api_token = apiToken;
await api.post('/cpa-services', payload);
toast.success('服务已添加');
}
closeCpaServiceModal();
loadCpaServices();
} catch (e) {
toast.error('保存失败: ' + e.message);
}
}
async function deleteCpaService(id, name) {
const confirmed = await confirm(`确定要删除 CPA 服务「${name}」吗?`);
if (!confirmed) return;
try {
await api.delete(`/cpa-services/${id}`);
toast.success('已删除');
loadCpaServices();
} catch (e) {
toast.error('删除失败: ' + e.message);
}
}
async function testCpaServiceById(id) {
try {
const result = await api.post(`/cpa-services/${id}/test`);
if (result.success) {
toast.success(result.message);
} else {
toast.error(result.message);
}
} catch (e) {
toast.error('测试失败: ' + e.message);
}
}
async function handleTestCpaService() {
const apiUrl = document.getElementById('cpa-service-url').value.trim();
const apiToken = document.getElementById('cpa-service-token').value.trim();
const id = document.getElementById('cpa-service-id').value;
if (!apiUrl) {
toast.error('请先填写 API URL');
return;
}
// 新增时必须有 token编辑时 token 可为空(用已保存的)
if (!id && !apiToken) {
toast.error('请先填写 API Token');
return;
}
elements.testCpaServiceBtn.disabled = true;
elements.testCpaServiceBtn.textContent = '测试中...';
try {
let result;
if (id && !apiToken) {
// 编辑时未填 token直接测试已保存的服务
result = await api.post(`/cpa-services/${id}/test`);
} else {
result = await api.post('/cpa-services/test-connection', { api_url: apiUrl, api_token: apiToken });
}
if (result.success) {
toast.success(result.message);
} else {
toast.error(result.message);
}
} catch (e) {
toast.error('测试失败: ' + e.message);
} finally {
elements.testCpaServiceBtn.disabled = false;
elements.testCpaServiceBtn.textContent = '🔌 测试连接';
}
}
function escapeHtml(text) {
if (!text) return '';
const d = document.createElement('div');
d.textContent = text;
return d.innerHTML;
}

View File

@@ -213,6 +213,26 @@
</div> </div>
</div> </div>
<!-- CPA 服务选择模态框 -->
<div class="modal" id="cpa-service-modal">
<div class="modal-content" style="max-width: 480px;">
<div class="modal-header">
<h3>☁️ 选择 CPA 服务</h3>
<button class="modal-close" id="close-cpa-modal">&times;</button>
</div>
<div class="modal-body">
<p style="color: var(--text-muted); margin-bottom: 12px; font-size: 0.9rem;">选择要上传到的 CPA 服务,或使用全局配置。</p>
<div id="cpa-service-list" style="display: flex; flex-direction: column; gap: 8px; max-height: 300px; overflow-y: auto;">
<div style="text-align: center; color: var(--text-muted);">加载中...</div>
</div>
</div>
<div class="modal-footer" style="padding: 12px 20px; border-top: 1px solid var(--border); display: flex; gap: 8px; justify-content: flex-end;">
<button class="btn btn-secondary" id="cpa-use-global-btn">使用全局配置</button>
<button class="btn btn-secondary" id="cancel-cpa-modal-btn">取消</button>
</div>
</div>
</div>
<script src="/static/js/utils.js"></script> <script src="/static/js/utils.js"></script>
<script src="/static/js/accounts.js"></script> <script src="/static/js/accounts.js"></script>
</body> </body>

View File

@@ -228,6 +228,12 @@
<input type="checkbox" id="auto-upload-cpa"> <input type="checkbox" id="auto-upload-cpa">
<span>上传到 CPA</span> <span>上传到 CPA</span>
</label> </label>
<div id="cpa-service-select-group" style="display:none; margin-top: 8px; padding-left: 4px;">
<label style="font-size:0.85rem; color:var(--text-muted); margin-bottom:4px; display:block;">选择 CPA 服务</label>
<select id="cpa-service-select" style="width:100%; padding:6px 10px; border:1px solid var(--border); border-radius:6px; background:var(--surface); color:var(--text-primary); font-size:0.9rem;">
<option value="">使用全局配置</option>
</select>
</div>
</div> </div>
<div class="form-actions" style="flex-direction: column;"> <div class="form-actions" style="flex-direction: column;">

View File

@@ -261,7 +261,7 @@
<div class="card"> <div class="card">
<div class="card-header"> <div class="card-header">
<h3>CPA 上传配置</h3> <h3>CPA 上传配置</h3>
<span class="hint">配置 Codex Protocol API 上传功能</span> <span class="hint">配置 CliProxyApi 上传功能</span>
</div> </div>
<div class="card-body"> <div class="card-body">
<form id="cpa-form"> <form id="cpa-form">
@@ -292,6 +292,77 @@
</form> </form>
</div> </div>
</div> </div>
<!-- CPA 服务管理 -->
<div class="card" style="margin-top: 16px;">
<div class="card-header">
<h3>CPA 服务管理</h3>
<button class="btn btn-primary btn-sm" id="add-cpa-service-btn">+ 添加服务</button>
</div>
<div class="card-body" style="padding: 0;">
<div class="table-container">
<table class="data-table">
<thead>
<tr>
<th>名称</th>
<th>API URL</th>
<th style="width:80px;">状态</th>
<th style="width:60px;">优先级</th>
<th style="width:160px;">操作</th>
</tr>
</thead>
<tbody id="cpa-services-table">
<tr><td colspan="5" style="text-align:center;color:var(--text-muted);padding:20px;">加载中...</td></tr>
</tbody>
</table>
</div>
</div>
</div>
</div>
<!-- CPA 服务编辑模态框 -->
<div class="modal" id="cpa-service-edit-modal">
<div class="modal-content" style="max-width: 500px;">
<div class="modal-header">
<h3 id="cpa-service-modal-title">添加 CPA 服务</h3>
<button class="modal-close" id="close-cpa-service-modal">&times;</button>
</div>
<div class="modal-body">
<form id="cpa-service-form">
<input type="hidden" id="cpa-service-id">
<div class="form-group">
<label for="cpa-service-name">名称 *</label>
<input type="text" id="cpa-service-name" placeholder="例如: 主服务" required>
</div>
<div class="form-group">
<label for="cpa-service-url">API URL *</label>
<input type="text" id="cpa-service-url" placeholder="https://cpa.example.com" required>
</div>
<div class="form-group">
<label for="cpa-service-token">API Token</label>
<input type="password" id="cpa-service-token" placeholder="编辑时留空则保持原值" autocomplete="new-password">
</div>
<div class="form-row">
<div class="form-group">
<label for="cpa-service-priority">优先级</label>
<input type="number" id="cpa-service-priority" value="0" min="0">
<p class="hint">数字越小优先级越高</p>
</div>
<div class="form-group">
<label>&nbsp;</label>
<label style="margin-top:10px;display:flex;align-items:center;gap:8px;">
<input type="checkbox" id="cpa-service-enabled" checked> 启用
</label>
</div>
</div>
<div class="form-actions">
<button type="submit" class="btn btn-primary">💾 保存</button>
<button type="button" class="btn btn-secondary" id="test-cpa-service-btn">🔌 测试连接</button>
<button type="button" class="btn btn-secondary" id="cancel-cpa-service-btn">取消</button>
</div>
</form>
</div>
</div>
</div> </div>
<!-- Team Manager 设置 --> <!-- Team Manager 设置 -->