35c0b7de16
Preserve local user table for superDream-specific features while syncing user lifecycle, API key CRUD and usage queries through sub2api. Admin token handles reads and user lifecycle; per-user tokens (Fernet-encrypted in DB) handle key writes that admin endpoints do not expose. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
96 lines
2.6 KiB
Python
96 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
# ── Auth ──────────────────────────────────────────────────────────────
|
|
|
|
class RegisterRequest(BaseModel):
|
|
email: str
|
|
password: str
|
|
|
|
|
|
class LoginRequest(BaseModel):
|
|
email: str
|
|
password: str
|
|
|
|
|
|
class TokenResponse(BaseModel):
|
|
access_token: str
|
|
refresh_token: Optional[str] = None
|
|
token_type: str = "bearer"
|
|
|
|
|
|
class RefreshRequest(BaseModel):
|
|
refresh_token: str
|
|
|
|
|
|
class ForgotPasswordRequest(BaseModel):
|
|
email: str
|
|
|
|
|
|
class ResetPasswordRequest(BaseModel):
|
|
token: str
|
|
new_password: str
|
|
|
|
|
|
class UserResponse(BaseModel):
|
|
"""Local user view. Balance mirrored from sub2api when available."""
|
|
|
|
id: str
|
|
email: str
|
|
status: str
|
|
created_at: datetime
|
|
sub2api_user_id: Optional[int] = None
|
|
balance: float = 0.0
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# ── API Key (shapes match sub2api dto.APIKey 1:1) ─────────────────────
|
|
|
|
class CreateKeyRequest(BaseModel):
|
|
name: str
|
|
group_id: Optional[int] = None
|
|
custom_key: Optional[str] = None
|
|
ip_whitelist: List[str] = []
|
|
ip_blacklist: List[str] = []
|
|
quota: Optional[float] = None
|
|
expires_in_days: Optional[int] = None
|
|
rate_limit_5h: Optional[float] = None
|
|
rate_limit_1d: Optional[float] = None
|
|
rate_limit_7d: Optional[float] = None
|
|
|
|
|
|
class UpdateKeyRequest(BaseModel):
|
|
name: Optional[str] = None
|
|
group_id: Optional[int] = None
|
|
status: Optional[str] = None
|
|
ip_whitelist: Optional[List[str]] = None
|
|
ip_blacklist: Optional[List[str]] = None
|
|
quota: Optional[float] = None
|
|
expires_at: Optional[str] = None
|
|
reset_quota: Optional[bool] = None
|
|
rate_limit_5h: Optional[float] = None
|
|
rate_limit_1d: Optional[float] = None
|
|
rate_limit_7d: Optional[float] = None
|
|
reset_rate_limit_usage: Optional[bool] = None
|
|
|
|
|
|
# ── Usage ─────────────────────────────────────────────────────────────
|
|
|
|
class DashboardAPIKeysUsageRequest(BaseModel):
|
|
api_key_ids: List[int]
|
|
|
|
|
|
# ── Common ────────────────────────────────────────────────────────────
|
|
|
|
class MessageResponse(BaseModel):
|
|
message: str
|
|
|
|
|
|
JSONDict = Dict[str, Any] |