352 lines
18 KiB
Python
352 lines
18 KiB
Python
"""数据中台 ORM 模型（SQLAlchemy 2.0）。"""
|
||
from __future__ import annotations
|
||
|
||
import datetime as dt
|
||
|
||
from sqlalchemy import (BigInteger, Date, DateTime, Float, Integer, String,
|
||
Text, UniqueConstraint, func)
|
||
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
|
||
|
||
|
||
class Base(DeclarativeBase):
|
||
pass
|
||
|
||
|
||
class Security(Base):
|
||
"""证券基础信息。"""
|
||
__tablename__ = "securities"
|
||
code: Mapped[str] = mapped_column(String(12), primary_key=True)
|
||
name: Mapped[str] = mapped_column(String(40))
|
||
market: Mapped[str] = mapped_column(String(8), default="A")
|
||
updated_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
|
||
|
||
class DailyQuote(Base):
|
||
"""个股日线（前复权）。"""
|
||
__tablename__ = "quotes_daily"
|
||
__table_args__ = (UniqueConstraint("code", "date", name="uq_quote_code_date"),)
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
code: Mapped[str] = mapped_column(String(12), index=True)
|
||
date: Mapped[dt.date] = mapped_column(Date, index=True)
|
||
open: Mapped[float] = mapped_column(Float)
|
||
high: Mapped[float] = mapped_column(Float)
|
||
low: Mapped[float] = mapped_column(Float)
|
||
close: Mapped[float] = mapped_column(Float)
|
||
volume: Mapped[int] = mapped_column(BigInteger, default=0)
|
||
amount: Mapped[float] = mapped_column(Float, default=0.0)
|
||
|
||
|
||
class IndexDaily(Base):
|
||
"""指数日线。"""
|
||
__tablename__ = "index_daily"
|
||
__table_args__ = (UniqueConstraint("code", "date", name="uq_index_code_date"),)
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
code: Mapped[str] = mapped_column(String(12), index=True)
|
||
name: Mapped[str] = mapped_column(String(40), default="")
|
||
date: Mapped[dt.date] = mapped_column(Date, index=True)
|
||
open: Mapped[float] = mapped_column(Float)
|
||
high: Mapped[float] = mapped_column(Float)
|
||
low: Mapped[float] = mapped_column(Float)
|
||
close: Mapped[float] = mapped_column(Float)
|
||
volume: Mapped[int] = mapped_column(BigInteger, default=0)
|
||
|
||
|
||
class SectorDaily(Base):
|
||
"""板块每日快照。"""
|
||
__tablename__ = "sector_daily"
|
||
__table_args__ = (UniqueConstraint("date", "name", name="uq_sector_date_name"),)
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
date: Mapped[dt.date] = mapped_column(Date, index=True)
|
||
name: Mapped[str] = mapped_column(String(40))
|
||
pct: Mapped[float] = mapped_column(Float, default=0.0)
|
||
amount: Mapped[float] = mapped_column(Float, default=0.0)
|
||
count: Mapped[int] = mapped_column(Integer, default=0)
|
||
leader: Mapped[str] = mapped_column(String(40), default="")
|
||
|
||
|
||
class FundFlowDaily(Base):
|
||
"""行业资金流每日快照。"""
|
||
__tablename__ = "fund_flow_daily"
|
||
__table_args__ = (UniqueConstraint("date", "name", name="uq_fund_date_name"),)
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
date: Mapped[dt.date] = mapped_column(Date, index=True)
|
||
name: Mapped[str] = mapped_column(String(40))
|
||
net: Mapped[float] = mapped_column(Float, default=0.0)
|
||
pct: Mapped[float] = mapped_column(Float, default=0.0)
|
||
|
||
|
||
class SentimentDaily(Base):
|
||
"""市场情绪每日快照。"""
|
||
__tablename__ = "sentiment_daily"
|
||
date: Mapped[dt.date] = mapped_column(Date, primary_key=True)
|
||
up: Mapped[int] = mapped_column(Integer, default=0)
|
||
down: Mapped[int] = mapped_column(Integer, default=0)
|
||
flat: Mapped[int] = mapped_column(Integer, default=0)
|
||
limit_up: Mapped[int] = mapped_column(Integer, default=0)
|
||
limit_down: Mapped[int] = mapped_column(Integer, default=0)
|
||
|
||
|
||
class DragonTiger(Base):
|
||
"""龙虎榜明细。"""
|
||
__tablename__ = "dragon_tiger"
|
||
__table_args__ = (UniqueConstraint("date", "code", "reason", name="uq_lhb"),)
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
date: Mapped[dt.date] = mapped_column(Date, index=True)
|
||
code: Mapped[str] = mapped_column(String(12))
|
||
name: Mapped[str] = mapped_column(String(40), default="")
|
||
pct: Mapped[float] = mapped_column(Float, default=0.0)
|
||
net: Mapped[float] = mapped_column(Float, default=0.0)
|
||
reason: Mapped[str] = mapped_column(String(120), default="")
|
||
|
||
|
||
class StockMetric(Base):
|
||
"""个股最新因子快照（供全市场选股快速查询）。"""
|
||
__tablename__ = "stock_metrics"
|
||
code: Mapped[str] = mapped_column(String(12), primary_key=True)
|
||
name: Mapped[str] = mapped_column(String(40), default="")
|
||
date: Mapped[dt.date] = mapped_column(Date, index=True)
|
||
close: Mapped[float] = mapped_column(Float, default=0.0)
|
||
pct: Mapped[float] = mapped_column(Float, default=0.0, index=True)
|
||
ma5: Mapped[float] = mapped_column(Float, default=0.0)
|
||
ma10: Mapped[float] = mapped_column(Float, default=0.0)
|
||
ma20: Mapped[float] = mapped_column(Float, default=0.0)
|
||
ma60: Mapped[float] = mapped_column(Float, default=0.0)
|
||
vol_ratio: Mapped[float] = mapped_column(Float, default=0.0)
|
||
ret5: Mapped[float] = mapped_column(Float, default=0.0, index=True)
|
||
ret20: Mapped[float] = mapped_column(Float, default=0.0)
|
||
ret60: Mapped[float] = mapped_column(Float, default=0.0)
|
||
pos60: Mapped[float] = mapped_column(Float, default=0.0) # 0~1，60日价格分位
|
||
rsi14: Mapped[float] = mapped_column(Float, default=0.0)
|
||
macd_gold: Mapped[bool] = mapped_column(default=False)
|
||
ma_bull: Mapped[bool] = mapped_column(default=False)
|
||
up_streak: Mapped[int] = mapped_column(Integer, default=0)
|
||
amount: Mapped[float] = mapped_column(Float, default=0.0)
|
||
|
||
|
||
class Trade(Base):
|
||
"""交易记录（用于持仓盈亏与归因）。"""
|
||
__tablename__ = "trades"
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
date: Mapped[dt.date] = mapped_column(Date, index=True)
|
||
code: Mapped[str] = mapped_column(String(12), index=True)
|
||
name: Mapped[str] = mapped_column(String(40), default="")
|
||
side: Mapped[str] = mapped_column(String(4)) # buy / sell
|
||
price: Mapped[float] = mapped_column(Float)
|
||
qty: Mapped[int] = mapped_column(Integer)
|
||
fee: Mapped[float] = mapped_column(Float, default=0.0)
|
||
reason: Mapped[str] = mapped_column(String(60), default="")
|
||
emotion: Mapped[str] = mapped_column(String(20), default="")
|
||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
|
||
|
||
class AlertRule(Base):
|
||
"""预警规则。"""
|
||
__tablename__ = "alert_rules"
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
code: Mapped[str] = mapped_column(String(12), index=True)
|
||
name: Mapped[str] = mapped_column(String(40), default="")
|
||
kind: Mapped[str] = mapped_column(String(20)) # price_above/price_below/pct_above/pct_below
|
||
threshold: Mapped[float] = mapped_column(Float)
|
||
channel: Mapped[str] = mapped_column(String(20), default="站内")
|
||
note: Mapped[str] = mapped_column(String(80), default="")
|
||
status: Mapped[str] = mapped_column(String(12), default="active") # active/triggered
|
||
last_value: Mapped[float] = mapped_column(Float, default=0.0)
|
||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
triggered_at: Mapped[dt.datetime | None] = mapped_column(DateTime, nullable=True)
|
||
|
||
|
||
class AlertEvent(Base):
|
||
"""预警触发事件（站内通知）。"""
|
||
__tablename__ = "alert_events"
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
rule_id: Mapped[int] = mapped_column(Integer, index=True)
|
||
code: Mapped[str] = mapped_column(String(12))
|
||
name: Mapped[str] = mapped_column(String(40), default="")
|
||
message: Mapped[str] = mapped_column(String(160))
|
||
value: Mapped[float] = mapped_column(Float, default=0.0)
|
||
read: Mapped[bool] = mapped_column(default=False)
|
||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
|
||
|
||
class DailyReport(Base):
|
||
"""AI 自动复盘日报（收盘后生成，可推送）。"""
|
||
__tablename__ = "daily_reports"
|
||
date: Mapped[dt.date] = mapped_column(Date, primary_key=True)
|
||
source: Mapped[str] = mapped_column(String(8), default="rule") # llm / rule
|
||
title: Mapped[str] = mapped_column(String(80), default="")
|
||
content: Mapped[str] = mapped_column(Text, default="") # markdown 正文
|
||
pushed: Mapped[bool] = mapped_column(default=False)
|
||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
|
||
|
||
class SignalStat(Base):
|
||
"""信号历史胜率（基于全市场历史日线回测的统计，支撑 AI 证据链的『历史命中率』）。"""
|
||
__tablename__ = "signal_stats"
|
||
__table_args__ = (UniqueConstraint("signal", "horizon", name="uq_signal_horizon"),)
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
signal: Mapped[str] = mapped_column(String(24), index=True)
|
||
horizon: Mapped[int] = mapped_column(Integer, default=5) # 向后 N 个交易日
|
||
samples: Mapped[int] = mapped_column(Integer, default=0)
|
||
win_rate: Mapped[float] = mapped_column(Float, default=0.0) # 上涨占比 %
|
||
avg_ret: Mapped[float] = mapped_column(Float, default=0.0) # 平均收益 %
|
||
updated_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
|
||
|
||
class Prediction(Base):
|
||
"""AI 诊断/预测留痕，N 日后核验真实涨跌，形成可回溯的『实测准确率』。"""
|
||
__tablename__ = "predictions"
|
||
__table_args__ = (UniqueConstraint("code", "date", "kind", name="uq_pred"),)
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
date: Mapped[dt.date] = mapped_column(Date, index=True)
|
||
code: Mapped[str] = mapped_column(String(12), index=True)
|
||
name: Mapped[str] = mapped_column(String(40), default="")
|
||
kind: Mapped[str] = mapped_column(String(16), default="diagnose")
|
||
score: Mapped[float] = mapped_column(Float, default=0.0)
|
||
confidence: Mapped[float] = mapped_column(Float, default=0.0)
|
||
direction: Mapped[str] = mapped_column(String(6), default="flat") # up/down/flat
|
||
horizon: Mapped[int] = mapped_column(Integer, default=5)
|
||
base_close: Mapped[float] = mapped_column(Float, default=0.0)
|
||
actual_ret: Mapped[float] = mapped_column(Float, default=0.0)
|
||
status: Mapped[str] = mapped_column(String(8), default="open") # open/closed
|
||
hit: Mapped[bool | None] = mapped_column(nullable=True)
|
||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
|
||
|
||
class JobRun(Base):
|
||
"""定时/手动任务执行日志。"""
|
||
__tablename__ = "job_runs"
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
job: Mapped[str] = mapped_column(String(40))
|
||
status: Mapped[str] = mapped_column(String(16)) # running/success/error
|
||
started_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
finished_at: Mapped[dt.datetime | None] = mapped_column(DateTime, nullable=True)
|
||
message: Mapped[str] = mapped_column(Text, default="")
|
||
|
||
|
||
class SelectorStrategy(Base):
|
||
"""选股策略保存。"""
|
||
__tablename__ = "selector_strategies"
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
name: Mapped[str] = mapped_column(String(80))
|
||
description: Mapped[str] = mapped_column(String(200), default="")
|
||
strategy_json: Mapped[str] = mapped_column(Text) # JSON格式的策略定义
|
||
is_preset: Mapped[bool] = mapped_column(default=False) # 是否预设策略
|
||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
updated_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
|
||
|
||
class SelectorAlert(Base):
|
||
"""选股条件预警。"""
|
||
__tablename__ = "selector_alerts"
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
strategy_id: Mapped[int] = mapped_column(Integer, index=True)
|
||
strategy_name: Mapped[str] = mapped_column(String(80))
|
||
status: Mapped[str] = mapped_column(String(12), default="active") # active/paused
|
||
last_checked: Mapped[dt.datetime | None] = mapped_column(DateTime, nullable=True)
|
||
last_count: Mapped[int] = mapped_column(Integer, default=0)
|
||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
|
||
|
||
class SocialPost(Base):
|
||
"""社区帖子。"""
|
||
__tablename__ = "social_posts"
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
source: Mapped[str] = mapped_column(String(20), index=True) # eastmoney/xueqiu/guba
|
||
post_id: Mapped[str] = mapped_column(String(100), unique=True)
|
||
code: Mapped[str] = mapped_column(String(12), index=True, default="")
|
||
title: Mapped[str] = mapped_column(String(200))
|
||
content: Mapped[str] = mapped_column(Text, default="")
|
||
author: Mapped[str] = mapped_column(String(80), default="")
|
||
comment_count: Mapped[int] = mapped_column(Integer, default=0)
|
||
view_count: Mapped[int] = mapped_column(Integer, default=0)
|
||
sentiment: Mapped[str] = mapped_column(String(20), default="neutral") # bullish/bearish/neutral
|
||
keywords: Mapped[str] = mapped_column(String(200), default="") # 逗号分隔
|
||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now(), index=True)
|
||
|
||
|
||
class SentimentIndex(Base):
|
||
"""社区情绪指数（每日）。"""
|
||
__tablename__ = "sentiment_index"
|
||
date: Mapped[dt.date] = mapped_column(Date, primary_key=True)
|
||
bullish_count: Mapped[int] = mapped_column(Integer, default=0)
|
||
bearish_count: Mapped[int] = mapped_column(Integer, default=0)
|
||
neutral_count: Mapped[int] = mapped_column(Integer, default=0)
|
||
bullish_ratio: Mapped[float] = mapped_column(Float, default=0.0) # 0-100
|
||
total_posts: Mapped[int] = mapped_column(Integer, default=0)
|
||
top_keywords: Mapped[str] = mapped_column(String(500), default="") # JSON格式
|
||
updated_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
|
||
|
||
class CorporateEvent(Base):
|
||
"""公司事件（财报、增减持、限售解禁等）。"""
|
||
__tablename__ = "corporate_events"
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
code: Mapped[str] = mapped_column(String(12), index=True)
|
||
name: Mapped[str] = mapped_column(String(40), default="")
|
||
event_type: Mapped[str] = mapped_column(String(20), index=True) # earnings/insider/unlock/dividend
|
||
event_date: Mapped[dt.date] = mapped_column(Date, index=True)
|
||
title: Mapped[str] = mapped_column(String(200))
|
||
description: Mapped[str] = mapped_column(Text, default="")
|
||
amount: Mapped[float] = mapped_column(Float, default=0.0) # 金额（亿元）
|
||
impact: Mapped[str] = mapped_column(String(20), default="neutral") # positive/negative/neutral
|
||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
|
||
|
||
class PolicyEvent(Base):
|
||
"""行业政策事件。"""
|
||
__tablename__ = "policy_events"
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
sector: Mapped[str] = mapped_column(String(40), index=True) # 受影响板块
|
||
event_date: Mapped[dt.date] = mapped_column(Date, index=True)
|
||
title: Mapped[str] = mapped_column(String(200))
|
||
content: Mapped[str] = mapped_column(Text, default="")
|
||
policy_type: Mapped[str] = mapped_column(String(40)) # subsidy/restriction/support/regulation
|
||
impact: Mapped[str] = mapped_column(String(20), default="neutral")
|
||
affected_stocks: Mapped[str] = mapped_column(String(500), default="") # 逗号分隔的股票代码
|
||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
|
||
|
||
class FinancialReport(Base):
|
||
"""财务报表数据。"""
|
||
__tablename__ = "financial_reports"
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
code: Mapped[str] = mapped_column(String(12), index=True)
|
||
name: Mapped[str] = mapped_column(String(40), default="")
|
||
report_date: Mapped[dt.date] = mapped_column(Date, index=True) # 报告期
|
||
publish_date: Mapped[dt.date] = mapped_column(Date, index=True) # 发布日期
|
||
report_type: Mapped[str] = mapped_column(String(20)) # Q1/Q2/Q3/annual
|
||
|
||
# 核心指标
|
||
revenue: Mapped[float] = mapped_column(Float, default=0.0) # 营收（亿元）
|
||
net_profit: Mapped[float] = mapped_column(Float, default=0.0) # 净利润（亿元）
|
||
roe: Mapped[float] = mapped_column(Float, default=0.0) # 净资产收益率(%)
|
||
gross_margin: Mapped[float] = mapped_column(Float, default=0.0) # 毛利率(%)
|
||
revenue_growth: Mapped[float] = mapped_column(Float, default=0.0) # 营收同比增长(%)
|
||
profit_growth: Mapped[float] = mapped_column(Float, default=0.0) # 净利润同比增长(%)
|
||
|
||
# 风险指标
|
||
inventory: Mapped[float] = mapped_column(Float, default=0.0) # 存货（亿元）
|
||
receivable: Mapped[float] = mapped_column(Float, default=0.0) # 应收账款（亿元）
|
||
debt_ratio: Mapped[float] = mapped_column(Float, default=0.0) # 资产负债率(%)
|
||
|
||
# AI摘要
|
||
ai_summary: Mapped[str] = mapped_column(String(500), default="")
|
||
|
||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now())
|
||
|
||
|
||
class IntradayEvent(Base):
|
||
"""盘中异动事件记录。"""
|
||
__tablename__ = "intraday_events"
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
code: Mapped[str] = mapped_column(String(12), index=True)
|
||
name: Mapped[str] = mapped_column(String(40), default="")
|
||
event_type: Mapped[str] = mapped_column(String(20), index=True) # surge/volume_break/limit_open/consecutive/big_order
|
||
price: Mapped[float] = mapped_column(Float, default=0.0)
|
||
pct: Mapped[float] = mapped_column(Float, default=0.0)
|
||
volume_ratio: Mapped[float] = mapped_column(Float, default=0.0)
|
||
amount: Mapped[float] = mapped_column(Float, default=0.0) # 对于big_order是单笔金额
|
||
description: Mapped[str] = mapped_column(String(200), default="")
|
||
detected_at: Mapped[dt.datetime] = mapped_column(DateTime, server_default=func.now(), index=True)
|
||
notified: Mapped[bool] = mapped_column(default=False)
|