Spaces:
Sleeping
Sleeping
File size: 6,653 Bytes
7482820 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 | """
Sub2API 服务管理 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.upload.sub2api_upload import test_sub2api_connection, batch_upload_to_sub2api
router = APIRouter()
# ============== Pydantic Models ==============
class Sub2ApiServiceCreate(BaseModel):
name: str
api_url: str
api_key: str
enabled: bool = True
priority: int = 0
class Sub2ApiServiceUpdate(BaseModel):
name: Optional[str] = None
api_url: Optional[str] = None
api_key: Optional[str] = None
enabled: Optional[bool] = None
priority: Optional[int] = None
class Sub2ApiServiceResponse(BaseModel):
id: int
name: str
api_url: str
has_key: bool
enabled: bool
priority: int
created_at: Optional[str] = None
updated_at: Optional[str] = None
class Config:
from_attributes = True
class Sub2ApiTestRequest(BaseModel):
api_url: Optional[str] = None
api_key: Optional[str] = None
class Sub2ApiUploadRequest(BaseModel):
account_ids: List[int]
service_id: Optional[int] = None
concurrency: int = 3
priority: int = 50
def _to_response(svc) -> Sub2ApiServiceResponse:
return Sub2ApiServiceResponse(
id=svc.id,
name=svc.name,
api_url=svc.api_url,
has_key=bool(svc.api_key),
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[Sub2ApiServiceResponse])
async def list_sub2api_services(enabled: Optional[bool] = None):
"""获取 Sub2API 服务列表"""
with get_db() as db:
services = crud.get_sub2api_services(db, enabled=enabled)
return [_to_response(s) for s in services]
@router.post("", response_model=Sub2ApiServiceResponse)
async def create_sub2api_service(request: Sub2ApiServiceCreate):
"""新增 Sub2API 服务"""
with get_db() as db:
svc = crud.create_sub2api_service(
db,
name=request.name,
api_url=request.api_url,
api_key=request.api_key,
enabled=request.enabled,
priority=request.priority,
)
return _to_response(svc)
@router.get("/{service_id}", response_model=Sub2ApiServiceResponse)
async def get_sub2api_service(service_id: int):
"""获取单个 Sub2API 服务详情"""
with get_db() as db:
svc = crud.get_sub2api_service_by_id(db, service_id)
if not svc:
raise HTTPException(status_code=404, detail="Sub2API 服务不存在")
return _to_response(svc)
@router.get("/{service_id}/full")
async def get_sub2api_service_full(service_id: int):
"""获取 Sub2API 服务完整配置（含 API Key）"""
with get_db() as db:
svc = crud.get_sub2api_service_by_id(db, service_id)
if not svc:
raise HTTPException(status_code=404, detail="Sub2API 服务不存在")
return {
"id": svc.id,
"name": svc.name,
"api_url": svc.api_url,
"api_key": svc.api_key,
"enabled": svc.enabled,
"priority": svc.priority,
}
@router.patch("/{service_id}", response_model=Sub2ApiServiceResponse)
async def update_sub2api_service(service_id: int, request: Sub2ApiServiceUpdate):
"""更新 Sub2API 服务配置"""
with get_db() as db:
svc = crud.get_sub2api_service_by_id(db, service_id)
if not svc:
raise HTTPException(status_code=404, detail="Sub2API 服务不存在")
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_key 留空则保持原值
if request.api_key:
update_data["api_key"] = request.api_key
if request.enabled is not None:
update_data["enabled"] = request.enabled
if request.priority is not None:
update_data["priority"] = request.priority
svc = crud.update_sub2api_service(db, service_id, **update_data)
return _to_response(svc)
@router.delete("/{service_id}")
async def delete_sub2api_service(service_id: int):
"""删除 Sub2API 服务"""
with get_db() as db:
svc = crud.get_sub2api_service_by_id(db, service_id)
if not svc:
raise HTTPException(status_code=404, detail="Sub2API 服务不存在")
crud.delete_sub2api_service(db, service_id)
return {"success": True, "message": f"Sub2API 服务 {svc.name} 已删除"}
@router.post("/{service_id}/test")
async def test_sub2api_service(service_id: int):
"""测试 Sub2API 服务连接"""
with get_db() as db:
svc = crud.get_sub2api_service_by_id(db, service_id)
if not svc:
raise HTTPException(status_code=404, detail="Sub2API 服务不存在")
success, message = test_sub2api_connection(svc.api_url, svc.api_key)
return {"success": success, "message": message}
@router.post("/test-connection")
async def test_sub2api_connection_direct(request: Sub2ApiTestRequest):
"""直接测试 Sub2API 连接（用于添加前验证）"""
if not request.api_url or not request.api_key:
raise HTTPException(status_code=400, detail="api_url 和 api_key 不能为空")
success, message = test_sub2api_connection(request.api_url, request.api_key)
return {"success": success, "message": message}
@router.post("/upload")
async def upload_accounts_to_sub2api(request: Sub2ApiUploadRequest):
"""批量上传账号到 Sub2API 平台"""
if not request.account_ids:
raise HTTPException(status_code=400, detail="账号 ID 列表不能为空")
with get_db() as db:
if request.service_id:
svc = crud.get_sub2api_service_by_id(db, request.service_id)
else:
svcs = crud.get_sub2api_services(db, enabled=True)
svc = svcs[0] if svcs else None
if not svc:
raise HTTPException(status_code=400, detail="未找到可用的 Sub2API 服务")
api_url = svc.api_url
api_key = svc.api_key
results = batch_upload_to_sub2api(
request.account_ids,
api_url,
api_key,
concurrency=request.concurrency,
priority=request.priority,
)
return results
|