66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
|
"""API 请求/响应模型（pydantic v2）。"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
from datetime import date, datetime
|
|||
|
|
from typing import List, Optional
|
|||
|
|
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------------- 事件 ----------------
|
|||
|
|
class EventIn(BaseModel):
|
|||
|
|
event_date: date
|
|||
|
|
title: str = Field(min_length=1, max_length=255)
|
|||
|
|
category: str = "其他"
|
|||
|
|
impact_direction: str = "neutral" # bullish/bearish/neutral
|
|||
|
|
severity: int = Field(default=3, ge=1, le=5)
|
|||
|
|
related_indices: Optional[str] = None # CSV，如 "000001.SH,399006.SZ"
|
|||
|
|
cycle_tag: str = "none" # top/bottom/none
|
|||
|
|
description: Optional[str] = None
|
|||
|
|
source_url: Optional[str] = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
class EventOut(EventIn):
|
|||
|
|
id: int
|
|||
|
|
source: str = "manual"
|
|||
|
|
created_at: Optional[datetime] = None
|
|||
|
|
updated_at: Optional[datetime] = None
|
|||
|
|
|
|||
|
|
model_config = {"from_attributes": True}
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------------- ETF ----------------
|
|||
|
|
class EtfFlowIn(BaseModel):
|
|||
|
|
trade_date: date
|
|||
|
|
etf_code: str = ""
|
|||
|
|
etf_name: Optional[str] = None
|
|||
|
|
category: Optional[str] = None
|
|||
|
|
related_index: Optional[str] = None
|
|||
|
|
net_inflow: Optional[float] = None # 亿元
|
|||
|
|
shares_change: Optional[float] = None
|
|||
|
|
note: Optional[str] = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
class EtfFlowOut(EtfFlowIn):
|
|||
|
|
id: int
|
|||
|
|
source: str = "manual"
|
|||
|
|
created_at: Optional[datetime] = None
|
|||
|
|
|
|||
|
|
model_config = {"from_attributes": True}
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------------- 顶底人工标注 ----------------
|
|||
|
|
class AnnotationIn(BaseModel):
|
|||
|
|
index_code: str
|
|||
|
|
anno_date: date
|
|||
|
|
kind: str = "watch" # top/bottom/watch
|
|||
|
|
note: Optional[str] = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ---------------- 导入结果 ----------------
|
|||
|
|
class ImportResult(BaseModel):
|
|||
|
|
inserted: int
|
|||
|
|
updated: int
|
|||
|
|
total: int
|
|||
|
|
errors: List[str] = []
|