58 lines
1.2 KiB
Python
58 lines
1.2 KiB
Python
"""
|
|
Legacy Schemas - 用于向后兼容
|
|
|
|
这些是旧架构的 Pydantic 模型，仅用于 legacy API routes。
|
|
新代码请使用 schemas.py 中的模型。
|
|
"""
|
|
|
|
from pydantic import BaseModel
|
|
from typing import Optional, List
|
|
from datetime import datetime
|
|
from enum import Enum
|
|
|
|
class AnalysisStatus(str, Enum):
|
|
PENDING = "pending"
|
|
IN_PROGRESS = "in_progress"
|
|
COMPLETED = "completed"
|
|
FAILED = "failed"
|
|
|
|
class StockSearchRequest(BaseModel):
|
|
query: str
|
|
|
|
class StockSearchResponse(BaseModel):
|
|
market: str
|
|
symbol: str
|
|
company_name: str
|
|
|
|
class AnalysisRequest(BaseModel):
|
|
market: str
|
|
symbol: str
|
|
company_name: str
|
|
model: Optional[str] = None
|
|
data_source: Optional[str] = None
|
|
|
|
class ReportSectionSchema(BaseModel):
|
|
section_name: str
|
|
content: str
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
class ReportResponse(BaseModel):
|
|
id: int
|
|
market: str
|
|
symbol: str
|
|
company_name: str
|
|
status: AnalysisStatus
|
|
ai_model: Optional[str] = None
|
|
created_at: datetime
|
|
sections: List[ReportSectionSchema] = []
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
class ConfigUpdateRequest(BaseModel):
|
|
key: str
|
|
value: str
|