|
|
|
|
|
"""
|
|
|
|
|
|
K线数据Schema
|
|
|
|
|
|
"""
|
|
|
|
|
|
from datetime import date, datetime
|
|
|
|
|
|
from typing import List, Optional, Dict, Any
|
|
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class KlineRequest(BaseModel):
|
|
|
|
|
|
"""K线数据请求"""
|
|
|
|
|
|
codes: str = Field(..., description="股票代码，多个用逗号分隔")
|
|
|
|
|
|
start_date: str = Field(..., description="开始日期(YYYYMMDD)")
|
|
|
|
|
|
end_date: str = Field(..., description="结束日期(YYYYMMDD)")
|
|
|
|
|
|
period: str = Field(default="daily", description="周期: daily, min1, min5, min15, min30, min60")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class KlineData(BaseModel):
|
|
|
|
|
|
"""单条K线数据"""
|
|
|
|
|
|
trade_date: Optional[str] = None
|
|
|
|
|
|
trade_datetime: Optional[str] = None
|
|
|
|
|
|
open: float
|
|
|
|
|
|
high: float
|
|
|
|
|
|
low: float
|
|
|
|
|
|
close: float
|
|
|
|
|
|
volume: int
|
|
|
|
|
|
amount: float
|
|
|
|
|
|
settle: Optional[float] = None
|
|
|
|
|
|
open_interest: Optional[int] = None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class KlineResponse(BaseModel):
|
|
|
|
|
|
"""K线数据响应"""
|
|
|
|
|
|
code: str
|
|
|
|
|
|
data: List[KlineData]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class KlineChartData(BaseModel):
|
|
|
|
|
|
"""K线图数据（ECharts格式）"""
|
|
|
|
|
|
categoryData: List[str]
|
|
|
|
|
|
values: List[List[float]] # [open, close, low, high, volume]
|
|
|
|
|
|
volumes: List[List[float]] # [index, volume, sign]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class BatchKlineRequest(BaseModel):
|
|
|
|
|
|
"""批量K线请求"""
|
|
|
|
|
|
codes: List[str]
|
|
|
|
|
|
start_date: str
|
|
|
|
|
|
end_date: str
|
|
|
|
|
|
period: str = "daily"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class SnapshotData(BaseModel):
|
|
|
|
|
|
"""快照数据"""
|
|
|
|
|
|
trade_time: str
|
|
|
|
|
|
pre_close: Optional[float] = None
|
|
|
|
|
|
last: Optional[float] = None
|
|
|
|
|
|
open: Optional[float] = None
|
|
|
|
|
|
high: Optional[float] = None
|
|
|
|
|
|
low: Optional[float] = None
|
|
|
|
|
|
close: Optional[float] = None
|
|
|
|
|
|
volume: Optional[int] = None
|
|
|
|
|
|
amount: Optional[float] = None
|
|
|
|
|
|
ask_price1: Optional[float] = None
|
|
|
|
|
|
ask_volume1: Optional[int] = None
|
|
|
|
|
|
bid_price1: Optional[float] = None
|
|
|
|
|
|
bid_volume1: Optional[int] = None
|
