liumangmang
6044b00685
- 上游 Key 命名改为 {prefix}-{upstream.id}-{safe_group_name}-{group_id}
- 唯一约束 (upstream_id, group_id, managed_prefix) 加 managed_prefix 列
- 上游检测成功时同步 Key 状态，远端已删/分组已删自动清理
- 重复分组导入跳过，目标网站已存在同名分组返回 exists
- 账号导入平台自动识别（auto/manual 模式）
- 全选可导入 Key 按钮 + 目标分组自动匹配
- 导入幂等：已导入过的 Key 校验远端账号，不存在则重建
- 新增同步接口 POST /sync-imported-upstream-keys
- account_exists() 通过拉取账号列表判断，避免 404 误判
- credentials.base_url 注入来源上游地址，避免 401
- 前端导入弹窗自动同步+刷新按钮+并发/优先级设置
- 新增 12 个测试覆盖同步、幂等、远端删除、校验失败路径
127 lines
3.4 KiB
Python
127 lines
3.4 KiB
Python
from datetime import datetime
|
|
from typing import Optional, Any
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class AuthConfigBearer(BaseModel):
|
|
token: str = ""
|
|
|
|
|
|
class AuthConfigApiKey(BaseModel):
|
|
key: str = ""
|
|
header: str = "Authorization"
|
|
|
|
|
|
class AuthConfigLoginPassword(BaseModel):
|
|
email: str = ""
|
|
password: str = ""
|
|
login_path: str = "/auth/login"
|
|
|
|
|
|
class UpstreamCreate(BaseModel):
|
|
name: str
|
|
base_url: str
|
|
api_prefix: str = "/api/v1"
|
|
auth_type: str = "login_password" # none | bearer | api_key | login_password
|
|
auth_config: dict[str, Any] = {}
|
|
rate_endpoint: str = "/groups/rates"
|
|
groups_endpoint: str = "/groups/available"
|
|
enabled: bool = True
|
|
check_interval_seconds: int = 600
|
|
timeout_seconds: int = 30
|
|
balance_endpoint: str = ""
|
|
balance_response_path: str = ""
|
|
balance_divisor: float = 1.0
|
|
|
|
|
|
class UpstreamUpdate(BaseModel):
|
|
name: Optional[str] = None
|
|
base_url: Optional[str] = None
|
|
api_prefix: Optional[str] = None
|
|
auth_type: Optional[str] = None
|
|
auth_config: Optional[dict[str, Any]] = None
|
|
rate_endpoint: Optional[str] = None
|
|
groups_endpoint: Optional[str] = None
|
|
enabled: Optional[bool] = None
|
|
check_interval_seconds: Optional[int] = None
|
|
timeout_seconds: Optional[int] = None
|
|
balance_endpoint: Optional[str] = None
|
|
balance_response_path: Optional[str] = None
|
|
balance_divisor: Optional[float] = None
|
|
|
|
|
|
class UpstreamResponse(BaseModel):
|
|
id: int
|
|
name: str
|
|
base_url: str
|
|
api_prefix: str
|
|
auth_type: str
|
|
auth_config_masked: dict[str, Any] # secrets replaced with ***
|
|
rate_endpoint: str
|
|
groups_endpoint: str
|
|
enabled: bool
|
|
check_interval_seconds: int
|
|
timeout_seconds: int
|
|
last_status: str
|
|
last_checked_at: Optional[datetime]
|
|
last_error: Optional[str]
|
|
balance: Optional[float] = None
|
|
balance_updated_at: Optional[datetime] = None
|
|
balance_endpoint: str = ""
|
|
balance_response_path: str = ""
|
|
balance_divisor: float = 1.0
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class SnapshotResponse(BaseModel):
|
|
id: int
|
|
upstream_id: int
|
|
snapshot: dict[str, Any]
|
|
captured_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class TestResult(BaseModel):
|
|
success: bool
|
|
message: str
|
|
detail: Optional[str] = None
|
|
|
|
|
|
class GenerateKeysByGroupsRequest(BaseModel):
|
|
group_ids: list[str] = Field(default_factory=list)
|
|
name_prefix: str = "SmartUp"
|
|
quota: float = Field(default=0, ge=0)
|
|
expires_in_days: Optional[int] = Field(default=None, ge=1)
|
|
rate_limit_5h: float = Field(default=0, ge=0)
|
|
rate_limit_1d: float = Field(default=0, ge=0)
|
|
rate_limit_7d: float = Field(default=0, ge=0)
|
|
endpoint: str = "/keys"
|
|
|
|
|
|
class GeneratedUpstreamKeyResponse(BaseModel):
|
|
id: Optional[int] = None
|
|
upstream_id: int
|
|
group_id: str
|
|
group_name: str = ""
|
|
key_id: Optional[str] = None
|
|
key_name: str
|
|
key_value: Optional[str] = None
|
|
masked_key: str = ""
|
|
status: str
|
|
error: Optional[str] = None
|
|
imported_website_id: Optional[int] = None
|
|
imported_account_id: Optional[str] = None
|
|
imported_at: Optional[datetime] = None
|
|
created_at: Optional[datetime] = None
|
|
updated_at: Optional[datetime] = None
|
|
|
|
|
|
class GenerateKeysByGroupsResponse(BaseModel):
|
|
success: bool
|
|
message: str
|
|
items: list[GeneratedUpstreamKeyResponse]
|