Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
 leefer
|
4bab921d14 | ||
|
leefer
|
cf2aad28ec | ||
|
leefer
|
814e75730a |
+4
-483
@@ -1,486 +1,7 @@
|
||||
from __future__ import annotations
|
||||
"""Compatibility alias for the canonical curated strategy library."""
|
||||
|
||||
from typing import Any
|
||||
import sys
|
||||
|
||||
from backend.features.screener import strategies as _implementation
|
||||
|
||||
def _meta(
|
||||
category: str,
|
||||
quality: str,
|
||||
frequency: str,
|
||||
risk: str,
|
||||
data_group: str,
|
||||
history_days: int,
|
||||
backtest_days: int,
|
||||
take_profit: float,
|
||||
stop_loss: float,
|
||||
**extra: Any,
|
||||
) -> dict[str, Any]:
|
||||
return {
|
||||
"library": "curated",
|
||||
"category": category,
|
||||
"quality": quality,
|
||||
"frequency": frequency,
|
||||
"risk": risk,
|
||||
"data_group": data_group,
|
||||
"history_days": history_days,
|
||||
"backtest_days": backtest_days,
|
||||
"take_profit": take_profit,
|
||||
"stop_loss": stop_loss,
|
||||
**extra,
|
||||
}
|
||||
|
||||
|
||||
ADVANCED_CURATED_STRATEGIES = [
|
||||
{
|
||||
"name": "中期动量·强者恒强",
|
||||
"description": "用60日至5日前的中期动量识别持续强势，同时剔除当日无法正常成交的涨停标的。",
|
||||
"regimes": ["repair", "fermentation", "climax", "divergence"],
|
||||
"formula": {
|
||||
"meta": _meta("动量反转", "A-", "每周", "中", "历史行情", 80, 10, 8, -5),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "close", "op": "between", "value": [3, 100]},
|
||||
{"field": "momentum_60_5_rank", "op": ">=", "value": 0.90},
|
||||
{"field": "is_limit_up_today", "op": "==", "value": 0},
|
||||
],
|
||||
"score": [
|
||||
{"field": "momentum_60_5", "weight": 0.55, "direction": "desc"},
|
||||
{"field": "relative_strength", "weight": 0.25, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.20, "direction": "desc"},
|
||||
],
|
||||
"limit": 25,
|
||||
"min_score": 0.50,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "强者回调",
|
||||
"description": "在中期强势股池中寻找回踩20日线、短期超卖且近20日无跌停的牛回头候选。",
|
||||
"regimes": ["repair", "fermentation", "divergence"],
|
||||
"formula": {
|
||||
"meta": _meta("动量反转", "A-", "每日", "中", "历史行情", 80, 10, 8, -5),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "momentum_60_5_rank", "op": ">=", "value": 0.70},
|
||||
{"field": "return_5d_rank", "op": "<=", "value": 0.20},
|
||||
{"field": "above_ma20", "op": "==", "value": 1},
|
||||
{"field": "rsi_6", "op": "<=", "value": 30},
|
||||
{"field": "no_limit_down_20d", "op": "==", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "momentum_60_5", "weight": 0.42, "direction": "desc"},
|
||||
{"field": "return_5d", "weight": 0.33, "direction": "asc"},
|
||||
{"field": "amount_billion", "weight": 0.25, "direction": "desc"},
|
||||
],
|
||||
"limit": 20,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "超跌反转",
|
||||
"description": "筛选短期极端回撤、充分换手但尚未形成长期单边下跌的修复候选。",
|
||||
"regimes": ["ice", "repair"],
|
||||
"formula": {
|
||||
"meta": _meta("动量反转", "B+", "每日", "高", "行情与财务", 80, 5, 8, -5),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "return_5d_rank", "op": "<=", "value": 0.05},
|
||||
{"field": "turnover_5d", "op": ">=", "value": 30},
|
||||
{"field": "return_60d", "op": ">=", "value": -40},
|
||||
{"field": "financial_risk", "op": "==", "value": 0},
|
||||
{"field": "is_limit_down_today", "op": "==", "value": 0},
|
||||
],
|
||||
"score": [
|
||||
{"field": "return_5d", "weight": 0.45, "direction": "asc"},
|
||||
{"field": "turnover_5d", "weight": 0.30, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.25, "direction": "desc"},
|
||||
],
|
||||
"limit": 10,
|
||||
"min_score": 0.50,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "相对强度新高",
|
||||
"description": "以个股相对沪深300的强度线识别弱市领涨和结构性抱团标的。",
|
||||
"regimes": ["ice", "repair", "fermentation", "divergence"],
|
||||
"formula": {
|
||||
"meta": _meta("动量反转", "A", "每周", "中", "行情与指数", 130, 20, 12, -7, requires_benchmark=True),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 250},
|
||||
"filters": [
|
||||
{"field": "amount_billion", "op": ">=", "value": 1},
|
||||
{"field": "rs_high_120", "op": "==", "value": 1},
|
||||
{"field": "excess_return_60d", "op": ">=", "value": 10},
|
||||
{"field": "ma60_slope", "op": ">", "value": 0},
|
||||
],
|
||||
"score": [
|
||||
{"field": "excess_return_60d", "weight": 0.50, "direction": "desc"},
|
||||
{"field": "ma60_slope", "weight": 0.25, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.25, "direction": "desc"},
|
||||
],
|
||||
"limit": 20,
|
||||
"min_score": 0.52,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "均线多头排列",
|
||||
"description": "使用5、10、20、60日均线多头结构、20日线斜率和250日位置确认趋势。",
|
||||
"regimes": ["repair", "fermentation", "climax", "divergence"],
|
||||
"formula": {
|
||||
"meta": _meta("趋势追踪", "A-", "每周", "中低", "历史行情", 260, 20, 12, -7),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 365},
|
||||
"filters": [
|
||||
{"field": "ma_bull_alignment", "op": "==", "value": 1},
|
||||
{"field": "ma20_slope_5d", "op": ">", "value": 0},
|
||||
{"field": "drawdown_from_high_250", "op": "<=", "value": 20},
|
||||
],
|
||||
"score": [
|
||||
{"field": "ma20_slope_5d", "weight": 0.38, "direction": "desc"},
|
||||
{"field": "drawdown_from_high_250", "weight": 0.32, "direction": "asc"},
|
||||
{"field": "relative_strength", "weight": 0.30, "direction": "desc"},
|
||||
],
|
||||
"limit": 30,
|
||||
"min_score": 0.50,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "唐奇安通道突破",
|
||||
"description": "收盘突破前20日高点，并以突破幅度、量能和突破前振幅过滤假突破。",
|
||||
"regimes": ["repair", "fermentation", "divergence"],
|
||||
"formula": {
|
||||
"meta": _meta("趋势追踪", "A-", "每日", "中", "历史行情", 80, 20, 12, -7),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "donchian_breakout_pct", "op": ">=", "value": 2},
|
||||
{"field": "volume_ratio_5d", "op": ">=", "value": 1.8},
|
||||
{"field": "range_20d", "op": "<=", "value": 35},
|
||||
],
|
||||
"score": [
|
||||
{"field": "volume_ratio_5d", "weight": 0.40, "direction": "desc"},
|
||||
{"field": "donchian_breakout_pct", "weight": 0.35, "direction": "desc"},
|
||||
{"field": "range_20d", "weight": 0.25, "direction": "asc"},
|
||||
],
|
||||
"limit": 15,
|
||||
"min_score": 0.52,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "周线趋势·日线买点",
|
||||
"description": "周线MACD位于多头区间，日线金叉或回踩20日线收阳时确认多周期共振。",
|
||||
"regimes": ["repair", "fermentation", "divergence"],
|
||||
"formula": {
|
||||
"meta": _meta("趋势追踪", "A", "每周", "中低", "多周期行情", 180, 20, 12, -7),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 365},
|
||||
"filters": [
|
||||
{"field": "weekly_trend_signal", "op": "==", "value": 1},
|
||||
{"field": "daily_buy_trigger", "op": "==", "value": 1},
|
||||
{"field": "weekly_amount_trend", "op": "==", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "ma20_slope_5d", "weight": 0.35, "direction": "desc"},
|
||||
{"field": "relative_strength", "weight": 0.35, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.30, "direction": "desc"},
|
||||
],
|
||||
"limit": 20,
|
||||
"min_score": 0.52,
|
||||
},
|
||||
},
|
||||
]
|
||||
|
||||
ADVANCED_CURATED_STRATEGIES.extend(
|
||||
[
|
||||
{
|
||||
"name": "空间板",
|
||||
"description": "识别当日新晋市场最高板，并要求所属方向具备足够的涨停支撑。",
|
||||
"regimes": ["repair", "fermentation"],
|
||||
"formula": {
|
||||
"meta": _meta("连板接力", "B+", "每日", "很高", "涨停结构", 80, 3, 8, -6),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 120},
|
||||
"filters": [
|
||||
{"field": "is_market_height", "op": "==", "value": 1},
|
||||
{"field": "new_space_board", "op": "==", "value": 1},
|
||||
{"field": "sector_limit_count", "op": ">=", "value": 3},
|
||||
],
|
||||
"score": [
|
||||
{"field": "limit_streak", "weight": 0.50, "direction": "desc"},
|
||||
{"field": "sector_limit_count", "weight": 0.30, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.20, "direction": "desc"},
|
||||
],
|
||||
"limit": 5,
|
||||
"min_score": 0.45,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "龙头首阴",
|
||||
"description": "筛选三板以上强势股断板后的首次缩量阴线，并结合板块强度观察承接质量。",
|
||||
"regimes": ["fermentation", "climax"],
|
||||
"formula": {
|
||||
"meta": _meta("低吸反核", "B", "每日", "很高", "涨停结构", 80, 5, 8, -6),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 120},
|
||||
"filters": [
|
||||
{"field": "max_continuous_board_10d", "op": ">=", "value": 3},
|
||||
{"field": "dragon_first_yin", "op": "==", "value": 1},
|
||||
{"field": "yin_day_pct", "op": ">=", "value": -7},
|
||||
{"field": "vol_vs_previous", "op": "<=", "value": 0.8},
|
||||
],
|
||||
"score": [
|
||||
{"field": "max_continuous_board_10d", "weight": 0.45, "direction": "desc"},
|
||||
{"field": "vol_vs_previous", "weight": 0.30, "direction": "asc"},
|
||||
{"field": "sector_strength", "weight": 0.25, "direction": "desc"},
|
||||
],
|
||||
"limit": 5,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "断板反包",
|
||||
"description": "连板断板后1至3日内，以涨停收复断板高点和量能确认N字反包。",
|
||||
"regimes": ["repair", "fermentation"],
|
||||
"formula": {
|
||||
"meta": _meta("低吸反核", "B+", "每日", "高", "涨停结构", 80, 3, 8, -6),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 120},
|
||||
"filters": [
|
||||
{"field": "broken_reversal", "op": "==", "value": 1},
|
||||
{"field": "days_since_broken", "op": "between", "value": [1, 3]},
|
||||
{"field": "close_above_broken_high", "op": "==", "value": 1},
|
||||
{"field": "vol_vs_broken_day", "op": ">=", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "days_since_broken", "weight": 0.35, "direction": "asc"},
|
||||
{"field": "vol_vs_broken_day", "weight": 0.35, "direction": "desc"},
|
||||
{"field": "sector_strength", "weight": 0.30, "direction": "desc"},
|
||||
],
|
||||
"limit": 5,
|
||||
"min_score": 0.46,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "核按钮反核",
|
||||
"description": "近5日强势股盘中深水急杀后收回，并以长下影和非放量结构确认承接。",
|
||||
"regimes": ["repair", "fermentation"],
|
||||
"formula": {
|
||||
"meta": _meta("低吸反核", "B+", "每日", "很高", "历史行情", 80, 5, 8, -6),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 120},
|
||||
"filters": [
|
||||
{"field": "recent_limit_up_5d", "op": ">=", "value": 1},
|
||||
{"field": "intraday_min_pct", "op": "<=", "value": -7},
|
||||
{"field": "pct_chg", "op": ">=", "value": -3},
|
||||
{"field": "lower_shadow_ratio", "op": ">=", "value": 2},
|
||||
{"field": "vol_vs_previous", "op": "<=", "value": 1.1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "lower_shadow_ratio", "weight": 0.42, "direction": "desc"},
|
||||
{"field": "intraday_min_pct", "weight": 0.30, "direction": "asc"},
|
||||
{"field": "sector_strength", "weight": 0.28, "direction": "desc"},
|
||||
],
|
||||
"limit": 5,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
]
|
||||
)
|
||||
|
||||
ADVANCED_CURATED_STRATEGIES.extend(
|
||||
[
|
||||
{
|
||||
"name": "景气-趋势-拥挤三维行业打分",
|
||||
"description": "以行业财务景气、价格趋势和交易拥挤度合成行业得分，再选取行业内动量与成交承载靠前的公司。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"行业轮动", "A-", "双周", "中", "行业、财务与交易拥挤", 80, 20, 12, -7,
|
||||
requires_fundamental=True,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "sector_composite_score", "op": ">=", "value": 0.58},
|
||||
{"field": "sector_crowding_rank", "op": "<=", "value": 0.90},
|
||||
{"field": "sector_stock_momentum_rank", "op": ">=", "value": 0.50},
|
||||
{"field": "amount_billion", "op": ">=", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "sector_composite_score", "weight": 0.55, "direction": "desc"},
|
||||
{"field": "sector_stock_momentum_rank", "weight": 0.25, "direction": "desc"},
|
||||
{"field": "sector_crowding_rank", "weight": 0.20, "direction": "asc"},
|
||||
],
|
||||
"limit": 12,
|
||||
"min_score": 0.50,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "大小盘/成长价值风格切换(元策略)",
|
||||
"description": "比较大小盘与成长价值组合近20日相对表现，动态选择当前占优风格中的匹配标的。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"元策略", "A-", "每周", "中低", "行情、估值与财务", 80, 20, 12, -7,
|
||||
requires_fundamental=True, requires_valuation=True,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 250},
|
||||
"filters": [
|
||||
{"field": "style_fit_score", "op": ">=", "value": 0.65},
|
||||
{"field": "amount_billion", "op": ">=", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "style_fit_score", "weight": 0.70, "direction": "desc"},
|
||||
{"field": "relative_strength", "weight": 0.30, "direction": "desc"},
|
||||
],
|
||||
"limit": 20,
|
||||
"min_score": 0.52,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "业绩超预期漂移(SUE/PEAD)",
|
||||
"description": "以业绩预告和业绩快报的同报告期差异识别超预期事件，并限定在公告后的首个交易窗口。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"业绩事件", "A-", "事件驱动", "中", "业绩预告与快报", 80, 20, 12, -7,
|
||||
requires_earnings_events=True,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "earnings_surprise_pct", "op": ">=", "value": 10},
|
||||
{"field": "revenue_yoy", "op": ">", "value": 0},
|
||||
{"field": "earnings_event_quality", "op": "==", "value": 1},
|
||||
{"field": "earnings_days_since_announce", "op": "between", "value": [1, 5]},
|
||||
],
|
||||
"score": [
|
||||
{"field": "earnings_surprise_pct", "weight": 0.60, "direction": "desc"},
|
||||
{"field": "relative_strength", "weight": 0.25, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.15, "direction": "desc"},
|
||||
],
|
||||
"limit": 15,
|
||||
"min_score": 0.50,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "多因子综合打分(IC动态加权)",
|
||||
"description": "将价值、成长、质量、动量和交易情绪标准化，并按近期横截面有效性动态合成综合分。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"多因子", "A-", "每周", "中", "行情、估值与财务", 260, 20, 12, -7,
|
||||
requires_fundamental=True, requires_valuation=True,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 250},
|
||||
"filters": [
|
||||
{"field": "multi_factor_composite", "op": ">=", "value": 0.65},
|
||||
{"field": "financial_risk", "op": "==", "value": 0},
|
||||
{"field": "amount_billion", "op": ">=", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "multi_factor_composite", "weight": 0.75, "direction": "desc"},
|
||||
{"field": "relative_strength", "weight": 0.15, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.10, "direction": "desc"},
|
||||
],
|
||||
"limit": 30,
|
||||
"min_score": 0.55,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "热度突增潜伏(另类数据)",
|
||||
"description": "从同花顺和东方财富人气榜中寻找排名快速跃升、但价格尚未明显兑现的观察候选。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"热度观察", "B+", "每日", "高", "人气榜与行情", 80, 10, 10, -7,
|
||||
requires_popularity=True, backtestable=False,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 120},
|
||||
"filters": [
|
||||
{"field": "popularity_score", "op": ">=", "value": 15},
|
||||
{"field": "return_10d", "op": "<=", "value": 5},
|
||||
{"field": "recent_limit_up_5d", "op": "==", "value": 0},
|
||||
{"field": "amount_billion", "op": ">=", "value": 0.5},
|
||||
],
|
||||
"score": [
|
||||
{"field": "popularity_score", "weight": 0.50, "direction": "desc"},
|
||||
{"field": "popularity_rank_change", "weight": 0.25, "direction": "desc"},
|
||||
{"field": "popularity_dual_source", "weight": 0.10, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.15, "direction": "desc"},
|
||||
],
|
||||
"limit": 10,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "机构榜溢价",
|
||||
"description": "筛选龙虎榜机构专用席位低位净买入的公司，并以席位数量和成交承载确认信号。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"资金席位", "B+", "每日", "中高", "龙虎榜机构席位", 80, 10, 10, -7,
|
||||
requires_institutions=True,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "institution_net_buy_million", "op": ">=", "value": 30},
|
||||
{"field": "institution_seat_count", "op": ">=", "value": 1},
|
||||
{"field": "return_60d", "op": "<=", "value": 30},
|
||||
{"field": "previous_limit_streak", "op": "<=", "value": 2},
|
||||
],
|
||||
"score": [
|
||||
{"field": "institution_net_buy_million", "weight": 0.55, "direction": "desc"},
|
||||
{"field": "institution_seat_count", "weight": 0.15, "direction": "desc"},
|
||||
{"field": "relative_position_60", "weight": 0.20, "direction": "asc"},
|
||||
{"field": "amount_billion", "weight": 0.10, "direction": "desc"},
|
||||
],
|
||||
"limit": 10,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
]
|
||||
)
|
||||
|
||||
ADVANCED_CURATED_STRATEGIES.extend(
|
||||
[
|
||||
{
|
||||
"name": "行业动量轮动",
|
||||
"description": "选择20日涨幅居前的行业，并在行业内部保留趋势与成交承载更强的前排公司。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta("行业轮动", "A-", "双周", "中", "行业与历史行情", 80, 20, 12, -7),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "sector_momentum_rank", "op": ">=", "value": 0.90},
|
||||
{"field": "sector_stock_momentum_rank", "op": ">=", "value": 0.80},
|
||||
{"field": "amount_billion", "op": ">=", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "sector_return_20d", "weight": 0.38, "direction": "desc"},
|
||||
{"field": "return_20d", "weight": 0.32, "direction": "desc"},
|
||||
{"field": "total_mv_billion", "weight": 0.18, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.12, "direction": "desc"},
|
||||
],
|
||||
"limit": 12,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "主力资金行业流入",
|
||||
"description": "寻找近5日主力资金持续净流入、行业涨幅尚未充分兑现的板块前排。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"行业轮动", "B+", "每周", "中高", "行业与资金流", 80, 10, 10, -7,
|
||||
requires_moneyflow_history=True,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "sector_flow_rank", "op": ">=", "value": 0.85},
|
||||
{"field": "sector_net_flow_5d_million", "op": ">", "value": 0},
|
||||
{"field": "sector_return_5d", "op": "<=", "value": 8},
|
||||
{"field": "flow_to_circ_mv_5d", "op": ">", "value": 0},
|
||||
{"field": "amount_billion", "op": ">=", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "flow_to_circ_mv_5d", "weight": 0.42, "direction": "desc"},
|
||||
{"field": "sector_net_flow_5d_million", "weight": 0.30, "direction": "desc"},
|
||||
{"field": "sector_return_5d", "weight": 0.16, "direction": "asc"},
|
||||
{"field": "amount_billion", "weight": 0.12, "direction": "desc"},
|
||||
],
|
||||
"limit": 15,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
]
|
||||
)
|
||||
sys.modules[__name__] = _implementation
|
||||
|
||||
+22
-876
@@ -40,53 +40,27 @@ from heaven_engine import (
|
||||
hexagram_from_lines,
|
||||
)
|
||||
from backend.data.providers.ifind_client import IfindError
|
||||
from llm_strategy import LLMCompilerError, compile_strategy_with_llm, test_llm_connection
|
||||
from llm_strategy import LLMCompilerError, test_llm_connection
|
||||
from mentor_agent import MentorAgentError, stream_with_mentor
|
||||
from market_insights import MarketInsightsService
|
||||
from screener import (
|
||||
FACTOR_FIELDS,
|
||||
FACTOR_GROUPS,
|
||||
REGIMES,
|
||||
FactorDataService,
|
||||
compile_local_strategy,
|
||||
)
|
||||
from backend.features.accounts.http import AccountHttpMixin
|
||||
from backend.features.accounts.security import SecretVault
|
||||
from backend.features.accounts.service import AccountService
|
||||
from backend.features.auction import AuctionServiceMixin
|
||||
from backend.features.dragon_tiger import DragonTigerServiceMixin
|
||||
from backend.features.pools import PoolServiceMixin
|
||||
from backend.features.sentiment import SentimentServiceMixin
|
||||
from backend.features.sentiment.engine import (
|
||||
build_sentiment_history,
|
||||
latest_contiguous_history,
|
||||
from backend.features.popularity import PopularityServiceMixin
|
||||
from backend.features.rotation import RotationServiceMixin
|
||||
from backend.features.screener.service import (
|
||||
SCREENER_LIBRARY_VERSION,
|
||||
ScreenerServiceMixin,
|
||||
automatic_screener_jobs,
|
||||
)
|
||||
from backend.features.sentiment import SentimentServiceMixin
|
||||
from backend.features.system import SystemHttpMixin
|
||||
from backend.features.themes import ThemeServiceMixin
|
||||
from backend.data.providers.tushare_client import TushareClient, TushareError, _sector_coverage_issue
|
||||
|
||||
|
||||
SCREENER_LIBRARY_VERSION = 8
|
||||
|
||||
|
||||
def automatic_screener_jobs(
|
||||
strategies: list[dict[str, Any]], regime_id: str
|
||||
) -> list[dict[str, Any]]:
|
||||
"""Build the close-of-day jobs; only stage screening is regime-gated."""
|
||||
smart_strategy = next(
|
||||
(
|
||||
item for item in strategies
|
||||
if item.get("formula", {}).get("meta", {}).get("library") != "curated"
|
||||
and regime_id in (item.get("regimes") or [])
|
||||
),
|
||||
None,
|
||||
)
|
||||
curated = [
|
||||
item for item in strategies
|
||||
if item.get("formula", {}).get("meta", {}).get("library") == "curated"
|
||||
]
|
||||
jobs = ([{"mode": "smart", "strategy": smart_strategy}] if smart_strategy else [])
|
||||
jobs.extend({"mode": "curated", "strategy": item} for item in curated)
|
||||
return jobs
|
||||
|
||||
|
||||
LEGACY_SECRET_KEYS = {
|
||||
"TUSHARE_TOKEN",
|
||||
"IFIND_REFRESH_TOKEN",
|
||||
@@ -140,7 +114,17 @@ MENTOR_ETF_UNIVERSE = (
|
||||
)
|
||||
|
||||
|
||||
class DashboardService(MarketServiceMixin, SentimentServiceMixin, PoolServiceMixin):
|
||||
class DashboardService(
|
||||
MarketServiceMixin,
|
||||
SentimentServiceMixin,
|
||||
PoolServiceMixin,
|
||||
RotationServiceMixin,
|
||||
AuctionServiceMixin,
|
||||
ThemeServiceMixin,
|
||||
PopularityServiceMixin,
|
||||
DragonTigerServiceMixin,
|
||||
ScreenerServiceMixin,
|
||||
):
|
||||
def __init__(self) -> None:
|
||||
runtime = load_runtime_settings()
|
||||
self.vault = SecretVault(runtime.encryption_key)
|
||||
@@ -734,158 +718,6 @@ class DashboardService(MarketServiceMixin, SentimentServiceMixin, PoolServiceMix
|
||||
|
||||
|
||||
|
||||
def rotation_history(self, trade_date: str, limit: int = 9) -> dict[str, Any]:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
# 板块轮动固定展示最近 9 个交易日，按由近到远排列。
|
||||
limit = 9
|
||||
snapshots = self.database.list_snapshot_payloads(normalized_date, 240)
|
||||
by_trade_date: dict[str, dict[str, Any]] = {}
|
||||
for snapshot in snapshots:
|
||||
meta = snapshot.get("meta") or {}
|
||||
actual_date = str(meta.get("trade_date") or snapshot.get("_snapshot_date") or "")
|
||||
compact_date = actual_date.replace("-", "")
|
||||
if len(compact_date) == 8:
|
||||
by_trade_date[compact_date] = snapshot
|
||||
|
||||
sentiment_dates = {
|
||||
str(row.get("trade_date") or "").replace("-", "")
|
||||
for row in latest_contiguous_history(build_sentiment_history(snapshots))
|
||||
}
|
||||
ordered_dates = sorted(
|
||||
date_key for date_key in by_trade_date
|
||||
if not sentiment_dates or date_key in sentiment_dates
|
||||
)[-limit:][::-1]
|
||||
rows = []
|
||||
for date_key in ordered_dates:
|
||||
snapshot = by_trade_date[date_key]
|
||||
sector_context = {
|
||||
str(item.get("name") or ""): item
|
||||
for item in snapshot.get("sectors") or []
|
||||
}
|
||||
sectors = []
|
||||
for item in (snapshot.get("sector_rotation") or [])[:12]:
|
||||
name = str(item.get("name") or "").strip()
|
||||
context = sector_context.get(name, {})
|
||||
sectors.append(
|
||||
{
|
||||
"name": name,
|
||||
"rank": int(item.get("rank") or len(sectors) + 1),
|
||||
"trend": item.get("trend") or "持平",
|
||||
"count": int(item.get("count") or 0),
|
||||
"strength": float(item.get("strength") or context.get("strength") or 0),
|
||||
"change": float(context.get("change") or 0),
|
||||
"leader": item.get("leader") or context.get("leader") or "--",
|
||||
}
|
||||
)
|
||||
rows.append(
|
||||
{
|
||||
"trade_date": f"{date_key[:4]}-{date_key[4:6]}-{date_key[6:]}",
|
||||
"sectors": sectors,
|
||||
}
|
||||
)
|
||||
return {
|
||||
"trade_date": rows[0]["trade_date"] if rows else normalized_date,
|
||||
"available_days": len(ordered_dates),
|
||||
"requested_days": limit,
|
||||
"rows": rows,
|
||||
}
|
||||
|
||||
def rotation_sector_members(self, trade_date: str, sector_name: str) -> dict[str, Any]:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
sector_name = validate_text(sector_name, "板块名称", 60, required=True)
|
||||
dashboard = self.get_dashboard(normalized_date)
|
||||
actual_date = normalize_date(
|
||||
str((dashboard.get("meta") or {}).get("trade_date") or normalized_date)
|
||||
)
|
||||
cache_key = f"{actual_date}:{sector_name}"
|
||||
cached = self.database.get_data_snapshot("rotation_sector_members_v1", cache_key)
|
||||
if cached:
|
||||
cached["meta"] = {**(cached.get("meta") or {}), "cached": True}
|
||||
return cached
|
||||
if not self.configured:
|
||||
raise ValueError("板块成分数据暂不可用。")
|
||||
|
||||
representative = next(
|
||||
(
|
||||
item for item in dashboard.get("limits") or []
|
||||
if str(item.get("sector") or "").strip() == sector_name
|
||||
),
|
||||
None,
|
||||
)
|
||||
if not representative:
|
||||
raise ValueError("未找到该板块的代表股票，暂时无法核验成分股。")
|
||||
raw_code = str(representative.get("ts_code") or representative.get("code") or "")
|
||||
if "." in raw_code:
|
||||
ts_code = raw_code
|
||||
elif raw_code.startswith(("4", "8", "92")):
|
||||
ts_code = f"{raw_code}.BJ"
|
||||
elif raw_code.startswith(("6", "68", "90")):
|
||||
ts_code = f"{raw_code}.SH"
|
||||
else:
|
||||
ts_code = f"{raw_code}.SZ"
|
||||
client = self._tushare_client()
|
||||
try:
|
||||
industry = client.sw_stock_industry(ts_code, actual_date)
|
||||
sector_code = str(industry.get("l2_code") or "")
|
||||
members = client.sw_sector_members(sector_code, actual_date)
|
||||
except TushareError as exc:
|
||||
raise ValueError(f"该板块成分股暂不可用：{exc}") from exc
|
||||
|
||||
daily_rows = self.database.daily_bars_for_date(actual_date)
|
||||
if len(daily_rows) < 1000:
|
||||
try:
|
||||
daily_rows = client.query(
|
||||
"daily",
|
||||
{"trade_date": actual_date},
|
||||
"ts_code,trade_date,open,high,low,close,pct_chg,vol,amount",
|
||||
)
|
||||
if daily_rows:
|
||||
self.database.upsert_daily_bars(daily_rows)
|
||||
except TushareError:
|
||||
daily_rows = self.database.daily_bars_for_date(actual_date)
|
||||
daily_map = {str(item.get("ts_code") or ""): item for item in daily_rows}
|
||||
rows = []
|
||||
for member in members:
|
||||
member_code = str(member.get("ts_code") or "")
|
||||
quote = daily_map.get(member_code) or {}
|
||||
rows.append(
|
||||
{
|
||||
"code": member_code.split(".")[0],
|
||||
"ts_code": member_code,
|
||||
"name": str(member.get("name") or "--"),
|
||||
"change": quote.get("pct_chg"),
|
||||
"open": quote.get("open"),
|
||||
"close": quote.get("close"),
|
||||
"amount_billion": (
|
||||
round(float(quote.get("amount") or 0) / 100000, 2)
|
||||
if quote else None
|
||||
),
|
||||
"quoted": bool(quote),
|
||||
}
|
||||
)
|
||||
rows.sort(
|
||||
key=lambda item: (
|
||||
bool(item.get("quoted")),
|
||||
float(item.get("change") or -999),
|
||||
float(item.get("amount_billion") or 0),
|
||||
),
|
||||
reverse=True,
|
||||
)
|
||||
result = {
|
||||
"meta": {
|
||||
"trade_date": self._display_compact_date(actual_date),
|
||||
"sector_name": str(industry.get("l2_name") or sector_name),
|
||||
"sector_code": sector_code,
|
||||
"member_count": len(rows),
|
||||
"quoted_count": sum(bool(item.get("quoted")) for item in rows),
|
||||
"cached": False,
|
||||
},
|
||||
"rows": rows,
|
||||
}
|
||||
self.database.save_data_snapshot(
|
||||
"rotation_sector_members_v1", cache_key, "tushare", result
|
||||
)
|
||||
return result
|
||||
|
||||
def status(self) -> dict[str, Any]:
|
||||
llm_access = self.llm_access_status()
|
||||
@@ -903,28 +735,9 @@ class DashboardService(MarketServiceMixin, SentimentServiceMixin, PoolServiceMix
|
||||
}
|
||||
|
||||
|
||||
def _market_insights(self) -> MarketInsightsService:
|
||||
if not self.configured:
|
||||
raise ValueError("行情数据尚未配置。")
|
||||
return MarketInsightsService(
|
||||
self.database,
|
||||
self._tushare_client(),
|
||||
ifind=self.ifind,
|
||||
)
|
||||
|
||||
def auction_center(self, trade_date: str, force: bool = False) -> dict[str, Any]:
|
||||
return self._market_insights().auction_center(
|
||||
normalize_date(trade_date), force, self.current_user_id
|
||||
)
|
||||
|
||||
def theme_library(self, trade_date: str, force: bool = False) -> dict[str, Any]:
|
||||
return self._market_insights().theme_library(normalize_date(trade_date), force)
|
||||
|
||||
def theme_detail(self, code: str, trade_date: str) -> dict[str, Any]:
|
||||
return self._market_insights().theme_detail(code, normalize_date(trade_date))
|
||||
|
||||
def popularity(self, trade_date: str, force: bool = False) -> dict[str, Any]:
|
||||
return self._market_insights().popularity(normalize_date(trade_date), force)
|
||||
|
||||
@staticmethod
|
||||
def _ifind_field(row: dict[str, Any], tokens: tuple[str, ...]) -> Any:
|
||||
@@ -950,141 +763,6 @@ class DashboardService(MarketServiceMixin, SentimentServiceMixin, PoolServiceMix
|
||||
return match.group(1)
|
||||
return ""
|
||||
|
||||
@staticmethod
|
||||
def _strategy_missing_data(
|
||||
strategy: dict[str, Any], factor_dates: list[str], factor_health: dict[str, Any]
|
||||
) -> list[str]:
|
||||
formula = strategy.get("formula") or {}
|
||||
meta = formula.get("meta") or {}
|
||||
used_fields = {
|
||||
str(item.get("field") or "")
|
||||
for item in list(formula.get("filters") or []) + list(formula.get("score") or [])
|
||||
}
|
||||
valuation_fields = {"pe_ttm", "pb", "ps_ttm", "dividend_yield_ttm", "total_mv_billion"}
|
||||
fundamental_fields = {"roe", "roa", "roic", "gross_margin", "netprofit_yoy", "revenue_yoy", "ocf_to_opincome"}
|
||||
auction_fields = {"auction_change", "auction_amount_million", "auction_turnover_rate", "auction_volume_ratio"}
|
||||
missing = []
|
||||
required_history = max(21, min(260, int(meta.get("history_days") or 21)))
|
||||
if len(factor_dates) < required_history:
|
||||
missing.append(f"历史行情（需{required_history}日）")
|
||||
if used_fields & valuation_fields and not factor_health["valuation"]:
|
||||
missing.append("估值数据")
|
||||
if used_fields & fundamental_fields and not factor_health["fundamental"]:
|
||||
missing.append("财务质量")
|
||||
if meta.get("requires_valuation") and not factor_health["valuation"]:
|
||||
missing.append("估值数据")
|
||||
if meta.get("requires_fundamental") and not factor_health["fundamental"]:
|
||||
missing.append("财务质量")
|
||||
if "dividend_years" in used_fields and not factor_health["dividend_history"]:
|
||||
missing.append("历年分红")
|
||||
if used_fields & auction_fields and not factor_health["auction"]:
|
||||
missing.append("竞价数据")
|
||||
if meta.get("requires_benchmark") and not factor_health.get("benchmark"):
|
||||
missing.append("沪深300基准")
|
||||
if meta.get("requires_moneyflow_history") and not factor_health.get("moneyflow_history"):
|
||||
missing.append("近5日资金流")
|
||||
if meta.get("requires_earnings_events") and not factor_health.get("earnings_events"):
|
||||
missing.append("业绩预告与快报")
|
||||
if meta.get("requires_popularity") and not factor_health.get("popularity"):
|
||||
missing.append("当日人气榜")
|
||||
if meta.get("requires_institutions") and not factor_health.get("institutions"):
|
||||
missing.append("龙虎榜机构席位")
|
||||
return list(dict.fromkeys(missing))
|
||||
|
||||
def screener_setup(self, trade_date: str) -> dict[str, Any]:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
regime = self.screener.detect_regime(normalized_date)
|
||||
factor_dates = self.database.factor_dates(normalized_date, 300)
|
||||
auction_dates = self.database.auction_factor_dates(normalized_date, 100)
|
||||
factor_health = self.screener.factor_health(normalized_date)
|
||||
strategies = self.database.list_screener_strategies(self.current_user_id)
|
||||
for strategy in strategies:
|
||||
missing = self._strategy_missing_data(strategy, factor_dates, factor_health)
|
||||
strategy["data_ready"] = not missing
|
||||
strategy["missing_data"] = missing
|
||||
automatic_results = self.database.screener_runs_for_date(0, normalized_date)
|
||||
personal_results = self.database.screener_runs_for_date(
|
||||
self.current_user_id, normalized_date
|
||||
)
|
||||
recent_results = [
|
||||
*[item for item in automatic_results if item.get("meta", {}).get("mode") in {"smart", "curated"}],
|
||||
*[item for item in personal_results if item.get("meta", {}).get("mode") == "quant"],
|
||||
]
|
||||
latest_results: dict[str, dict[str, Any]] = {}
|
||||
for result in reversed(recent_results):
|
||||
mode = str(result.get("meta", {}).get("mode") or "smart")
|
||||
latest_results[mode] = result
|
||||
automatic_status = self.database.get_data_snapshot(
|
||||
"screener_auto_v1", normalized_date
|
||||
) or {}
|
||||
return {
|
||||
"trade_date": normalized_date,
|
||||
"regime": regime,
|
||||
"regimes": [{"id": key, "label": value} for key, value in REGIMES.items()],
|
||||
"strategies": strategies,
|
||||
"factor_fields": [{"id": key, "label": value} for key, value in FACTOR_FIELDS.items()],
|
||||
"factor_groups": [
|
||||
{
|
||||
"name": name,
|
||||
"fields": [{"id": field, "label": FACTOR_FIELDS[field]} for field in fields],
|
||||
}
|
||||
for name, fields in FACTOR_GROUPS.items()
|
||||
],
|
||||
"operators": [">", ">=", "<", "<=", "==", "between"],
|
||||
"factor_data": {
|
||||
"date_count": len(factor_dates),
|
||||
"start_date": factor_dates[0] if factor_dates else "",
|
||||
"end_date": factor_dates[-1] if factor_dates else "",
|
||||
"ready": len(factor_dates) >= 21,
|
||||
"auction_date_count": len(auction_dates),
|
||||
"auction_ready": bool(auction_dates and auction_dates[-1] == factor_dates[-1]) if factor_dates else False,
|
||||
"health": factor_health,
|
||||
},
|
||||
"llm": {
|
||||
"configured": self.llm_configured,
|
||||
"model": self.llm_primary_model if self.llm_configured else "",
|
||||
"fallback_configured": self.llm_fallback_configured,
|
||||
"fallback_model": self.llm_fallback_model if self.llm_fallback_configured else "",
|
||||
},
|
||||
"latest_results": latest_results,
|
||||
"recent_results": recent_results,
|
||||
"automatic_status": automatic_status,
|
||||
# Kept during the client transition for compatibility with older frontends.
|
||||
"latest_result": latest_results.get("smart"),
|
||||
}
|
||||
|
||||
def screener_tracking(self, limit: int = 12) -> dict[str, Any]:
|
||||
return self.strategy_tracking.list_tracking(self.current_user_id, limit)
|
||||
|
||||
def add_screener_tracking(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||
try:
|
||||
run_id = int(payload.get("run_id") or 0)
|
||||
except (TypeError, ValueError) as exc:
|
||||
raise ValueError("选股批次无效。") from exc
|
||||
code = str(payload.get("code") or "").strip()
|
||||
if run_id <= 0 or not re.fullmatch(r"\d{6}", code):
|
||||
raise ValueError("选股批次或股票代码无效。")
|
||||
return self.strategy_tracking.add_candidate(self.current_user_id, run_id, code)
|
||||
|
||||
def remove_screener_tracking(self, track_id: int) -> dict[str, Any]:
|
||||
return self.strategy_tracking.remove_candidate(self.current_user_id, track_id)
|
||||
|
||||
def refresh_screener_tracking(self, trade_date: str) -> dict[str, Any]:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
notice = ""
|
||||
if self.configured:
|
||||
try:
|
||||
FactorDataService(self.database, self._tushare_client()).sync(
|
||||
normalized_date, 15
|
||||
)
|
||||
except TushareError:
|
||||
notice = "最新日线暂未补齐，已按现有数据更新跟踪。"
|
||||
else:
|
||||
notice = "公共行情尚未配置，已按现有数据更新跟踪。"
|
||||
return {
|
||||
"tracking": self.screener_tracking(),
|
||||
"notice": notice,
|
||||
}
|
||||
|
||||
def alert_center(self, status: str = "all", as_of: str = "") -> dict[str, Any]:
|
||||
tracking = self.strategy_tracking.list_tracking(self.current_user_id, 12)
|
||||
@@ -1288,220 +966,6 @@ class DashboardService(MarketServiceMixin, SentimentServiceMixin, PoolServiceMix
|
||||
},
|
||||
}
|
||||
|
||||
def sync_screener_data(self, trade_date: str, lookback: int = 45) -> dict[str, Any]:
|
||||
if not self.configured:
|
||||
raise ValueError("请先配置 Tushare Token。")
|
||||
normalized_date = normalize_date(trade_date)
|
||||
lookback = max(25, min(260, int(lookback)))
|
||||
with self.sync_lock:
|
||||
return FactorDataService(self.database, self._tushare_client()).sync(
|
||||
normalized_date, lookback
|
||||
)
|
||||
|
||||
def _schedule_automatic_screeners(
|
||||
self, trade_date: str, snapshot: dict[str, Any] | None = None
|
||||
) -> bool:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
now = datetime.now().astimezone()
|
||||
if (
|
||||
normalized_date != now.strftime("%Y%m%d")
|
||||
or now.weekday() >= 5
|
||||
or now.time().replace(tzinfo=None) < datetime.strptime("15:10", "%H:%M").time()
|
||||
or self.auto_screener_lock.locked()
|
||||
):
|
||||
return False
|
||||
snapshot = snapshot or self.database.get_snapshot(normalized_date) or {}
|
||||
actual_date = str((snapshot.get("meta") or {}).get("trade_date") or "").replace("-", "")
|
||||
if actual_date != normalized_date:
|
||||
return False
|
||||
marker = self.database.get_data_snapshot("screener_auto_v1", normalized_date) or {}
|
||||
if (
|
||||
marker.get("status") == "complete"
|
||||
and int(marker.get("library_version") or 0) == SCREENER_LIBRARY_VERSION
|
||||
):
|
||||
return False
|
||||
last_attempt = self._auto_screener_last_attempt.get(normalized_date)
|
||||
if last_attempt and (now - last_attempt).total_seconds() < 600:
|
||||
return False
|
||||
self._auto_screener_last_attempt[normalized_date] = now
|
||||
return self.jobs.submit(
|
||||
"screener.automatic",
|
||||
f"{normalized_date}:v{SCREENER_LIBRARY_VERSION}",
|
||||
lambda: self.run_automatic_screeners(normalized_date),
|
||||
{"trade_date": normalized_date, "trigger": "post-close"},
|
||||
)
|
||||
|
||||
def run_automatic_screeners(self, trade_date: str) -> dict[str, Any]:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
with self.auto_screener_lock:
|
||||
started_at = datetime.now().astimezone().isoformat(timespec="seconds")
|
||||
status: dict[str, Any] = {
|
||||
"trade_date": normalized_date,
|
||||
"library_version": SCREENER_LIBRARY_VERSION,
|
||||
"status": "running",
|
||||
"started_at": started_at,
|
||||
"completed": [],
|
||||
"skipped": [],
|
||||
"failed": [],
|
||||
}
|
||||
self.database.save_data_snapshot(
|
||||
"screener_auto_v1", normalized_date, "system", status
|
||||
)
|
||||
try:
|
||||
factor_sync = FactorDataService(
|
||||
self.database, self._tushare_client()
|
||||
).sync(normalized_date, 260)
|
||||
factor_dates = self.database.factor_dates(normalized_date, 300)
|
||||
if not factor_dates or factor_dates[-1] != normalized_date:
|
||||
raise ValueError("当日收盘行情尚未入库")
|
||||
factor_health = self.screener.factor_health(normalized_date)
|
||||
regime = self.screener.detect_regime(normalized_date)
|
||||
regime_id = str(regime.get("id") or "repair")
|
||||
strategies = self.database.list_screener_strategies(None)
|
||||
jobs = automatic_screener_jobs(strategies, regime_id)
|
||||
existing = {
|
||||
(
|
||||
str(item.get("meta", {}).get("mode") or "smart"),
|
||||
str(item.get("meta", {}).get("strategy_name") or ""),
|
||||
)
|
||||
for item in self.database.screener_runs_for_date(0, normalized_date)
|
||||
if int(item.get("meta", {}).get("library_version") or 0)
|
||||
== SCREENER_LIBRARY_VERSION
|
||||
}
|
||||
required_history = max(
|
||||
[
|
||||
int((job["strategy"].get("formula", {}).get("meta", {}) or {}).get("history_days") or 80)
|
||||
for job in jobs if job.get("strategy")
|
||||
] or [80]
|
||||
)
|
||||
factors, actual_date = self.screener.build_factors(
|
||||
normalized_date, history_days=required_history
|
||||
)
|
||||
if actual_date != normalized_date:
|
||||
raise ValueError("当日因子尚未完成收盘定格")
|
||||
for job in jobs:
|
||||
strategy = job["strategy"]
|
||||
mode = str(job["mode"])
|
||||
name = str(strategy.get("name") or "未命名策略")
|
||||
if (mode, name) in existing:
|
||||
status["completed"].append({"mode": mode, "name": name, "cached": True})
|
||||
continue
|
||||
missing = self._strategy_missing_data(
|
||||
strategy, factor_dates, factor_health
|
||||
)
|
||||
if missing:
|
||||
status["skipped"].append(
|
||||
{"mode": mode, "name": name, "reason": "、".join(missing)}
|
||||
)
|
||||
continue
|
||||
try:
|
||||
formula = copy.deepcopy(strategy.get("formula") or {})
|
||||
formula.setdefault("meta", {})["library_version"] = (
|
||||
SCREENER_LIBRARY_VERSION
|
||||
)
|
||||
result = self.screener.screen(
|
||||
0,
|
||||
normalized_date,
|
||||
formula,
|
||||
regime_id,
|
||||
name,
|
||||
False,
|
||||
None,
|
||||
mode,
|
||||
factors,
|
||||
actual_date,
|
||||
)
|
||||
status["completed"].append(
|
||||
{
|
||||
"mode": mode,
|
||||
"name": name,
|
||||
"candidate_count": len(result.get("candidates") or []),
|
||||
}
|
||||
)
|
||||
except Exception as exc:
|
||||
status["failed"].append(
|
||||
{"mode": mode, "name": name, "reason": str(exc)}
|
||||
)
|
||||
status.update(
|
||||
{
|
||||
"status": "complete" if not status["failed"] else "partial",
|
||||
"finished_at": datetime.now().astimezone().isoformat(timespec="seconds"),
|
||||
"factor_sync": factor_sync,
|
||||
"regime": regime,
|
||||
}
|
||||
)
|
||||
except Exception as exc:
|
||||
status.update(
|
||||
{
|
||||
"status": "failed",
|
||||
"finished_at": datetime.now().astimezone().isoformat(timespec="seconds"),
|
||||
"error": str(exc),
|
||||
}
|
||||
)
|
||||
self.database.save_data_snapshot(
|
||||
"screener_auto_v1", normalized_date, "system", status
|
||||
)
|
||||
return status
|
||||
|
||||
def compile_screener_strategy(self, prompt: str, regime: str) -> dict[str, Any]:
|
||||
prompt = prompt.strip()
|
||||
if not prompt or len(prompt) > 3000:
|
||||
raise ValueError("策略描述应为 1 至 3000 个字符。")
|
||||
if regime not in REGIMES:
|
||||
raise ValueError("市场阶段不支持。")
|
||||
notice = ""
|
||||
source = self.llm_source
|
||||
if source == "platform":
|
||||
try:
|
||||
gateway_result = self.llm_gateway.call(
|
||||
"screener",
|
||||
"strategy-compiler-v1",
|
||||
lambda profile: compile_strategy_with_llm(
|
||||
prompt,
|
||||
regime,
|
||||
profile.api_key,
|
||||
profile.base_url,
|
||||
profile.model,
|
||||
),
|
||||
(LLMCompilerError,),
|
||||
)
|
||||
compiled = gateway_result.value
|
||||
if gateway_result.role == "fallback":
|
||||
compiled["compiler"] = "llm_fallback"
|
||||
notice = "智能策略生成服务已自动切换。"
|
||||
except LLMGatewayError as exc:
|
||||
if exc.code != "unavailable":
|
||||
raise
|
||||
compiled = compile_local_strategy(prompt, regime)
|
||||
notice = "智能策略生成暂不可用，已使用本地模板。"
|
||||
else:
|
||||
compiled = compile_local_strategy(prompt, regime)
|
||||
notice = "智能策略生成暂不可用，已使用本地模板。"
|
||||
compiled["formula"] = self.screener.validate_formula(compiled["formula"])
|
||||
compiled["notice"] = notice
|
||||
return compiled
|
||||
|
||||
def save_screener_strategy(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||
name = validate_text(payload.get("name"), "策略名称", 60, required=True)
|
||||
description = validate_text(payload.get("description"), "策略说明", 1000)
|
||||
regimes = payload.get("regimes") or []
|
||||
if not isinstance(regimes, list) or not regimes or any(item not in REGIMES for item in regimes):
|
||||
raise ValueError("策略适用阶段不正确。")
|
||||
formula = self.screener.validate_formula(payload.get("formula") or {})
|
||||
strategy_id = self.database.save_screener_strategy(
|
||||
self.current_user_id, name, description, regimes, formula
|
||||
)
|
||||
return {
|
||||
"id": strategy_id,
|
||||
"strategies": self.database.list_screener_strategies(self.current_user_id),
|
||||
}
|
||||
|
||||
def delete_screener_strategy(self, strategy_id: int) -> dict[str, Any]:
|
||||
deleted = self.database.delete_screener_strategy(self.current_user_id, strategy_id)
|
||||
return {
|
||||
"deleted": deleted,
|
||||
"strategies": self.database.list_screener_strategies(self.current_user_id),
|
||||
}
|
||||
|
||||
def mentor_setup(self, trade_date: str) -> dict[str, Any]:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
@@ -3187,324 +2651,6 @@ class DashboardService(MarketServiceMixin, SentimentServiceMixin, PoolServiceMix
|
||||
)
|
||||
return result
|
||||
|
||||
def run_screener(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||
trade_date = normalize_date(str(payload.get("trade_date") or date.today().isoformat()))
|
||||
regime = str(payload.get("regime") or "")
|
||||
if regime not in REGIMES:
|
||||
raise ValueError("市场阶段不支持。")
|
||||
strategy_name = validate_text(payload.get("strategy_name"), "策略名称", 60, required=True)
|
||||
formula = payload.get("formula") or {}
|
||||
requested_mode = str(payload.get("mode") or "").strip()
|
||||
if requested_mode and requested_mode not in {"smart", "curated", "quant"}:
|
||||
raise ValueError("选股模式不受支持。")
|
||||
if requested_mode:
|
||||
mode = requested_mode
|
||||
else:
|
||||
meta = formula.get("meta") if isinstance(formula, dict) else {}
|
||||
library = str((meta or {}).get("library") or "")
|
||||
category = str((meta or {}).get("category") or "")
|
||||
if library == "curated":
|
||||
mode = "curated"
|
||||
elif library == "quant" or (library == "custom" and category == "量化公式"):
|
||||
mode = "quant"
|
||||
else:
|
||||
mode = "smart"
|
||||
realtime_snapshot = None
|
||||
dashboard = self.get_dashboard(trade_date)
|
||||
if self.configured and dashboard.get("meta", {}).get("realtime"):
|
||||
try:
|
||||
realtime_snapshot = self._tushare_client().realtime_factor_snapshot(trade_date)
|
||||
except TushareError as exc:
|
||||
raise ValueError(f"实时选股行情不可用，已停止筛选：{exc}") from exc
|
||||
result = self.screener.screen(
|
||||
self.current_user_id, trade_date, formula, regime, strategy_name,
|
||||
bool(payload.get("run_backtest", True)),
|
||||
realtime_snapshot,
|
||||
mode,
|
||||
)
|
||||
return result
|
||||
|
||||
def get_hot_money_profiles(self, force: bool = False) -> dict[str, Any]:
|
||||
cache_kind = "hot_money_profiles_v1"
|
||||
cache_key = "directory"
|
||||
cached = self.database.get_data_snapshot(cache_kind, cache_key)
|
||||
if cached and not force:
|
||||
cached["meta"] = {**cached.get("meta", {}), "cached": True}
|
||||
return cached
|
||||
if self.configured:
|
||||
try:
|
||||
payload = self._tushare_client().hot_money_profiles()
|
||||
except TushareError:
|
||||
if cached:
|
||||
cached["meta"] = {
|
||||
**cached.get("meta", {}),
|
||||
"cached": True,
|
||||
"stale": True,
|
||||
"notice": "名录暂未完成更新，当前展示最近一次收录结果。",
|
||||
}
|
||||
return cached
|
||||
return {
|
||||
"meta": {
|
||||
"source": "unavailable",
|
||||
"status": "unavailable",
|
||||
"schema_version": 1,
|
||||
"cached": False,
|
||||
"updated_at": datetime.now().astimezone().isoformat(timespec="seconds"),
|
||||
"notice": "游资名录暂不可用，请稍后重试。",
|
||||
},
|
||||
"summary": {
|
||||
"profile_count": 0,
|
||||
"described_count": 0,
|
||||
"organization_count": 0,
|
||||
},
|
||||
"profiles": [],
|
||||
}
|
||||
payload["meta"]["cached"] = False
|
||||
if payload.get("meta", {}).get("status") == "success":
|
||||
self.database.save_data_snapshot(cache_kind, cache_key, "tushare", payload)
|
||||
return payload
|
||||
if cached:
|
||||
cached["meta"] = {**cached.get("meta", {}), "cached": True}
|
||||
return cached
|
||||
return {
|
||||
"meta": {
|
||||
"source": "unavailable",
|
||||
"status": "unavailable",
|
||||
"schema_version": 1,
|
||||
"cached": False,
|
||||
"updated_at": datetime.now().astimezone().isoformat(timespec="seconds"),
|
||||
"notice": "游资名录暂不可用，请联系管理员检查行情配置。",
|
||||
},
|
||||
"summary": {
|
||||
"profile_count": 0,
|
||||
"described_count": 0,
|
||||
"organization_count": 0,
|
||||
},
|
||||
"profiles": [],
|
||||
}
|
||||
|
||||
def get_dragon_tiger(self, trade_date: str, force: bool = False) -> dict[str, Any]:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
cache_kind = "hot_money_detail_v3"
|
||||
if not force:
|
||||
cached = self.database.get_data_snapshot(cache_kind, normalized_date)
|
||||
if (
|
||||
cached
|
||||
and cached.get("meta", {}).get("source") == "tushare"
|
||||
and cached.get("meta", {}).get("status") == "success"
|
||||
and int(cached.get("meta", {}).get("schema_version") or 0) == 3
|
||||
):
|
||||
cached["meta"] = {**cached.get("meta", {}), "cached": True}
|
||||
return cached
|
||||
if self.configured:
|
||||
try:
|
||||
payload = self._tushare_client().dragon_tiger(normalized_date)
|
||||
except TushareError as exc:
|
||||
return {
|
||||
"meta": {
|
||||
"requested_date": f"{normalized_date[:4]}-{normalized_date[4:6]}-{normalized_date[6:8]}",
|
||||
"trade_date": f"{normalized_date[:4]}-{normalized_date[4:6]}-{normalized_date[6:8]}",
|
||||
"source": "tushare_error",
|
||||
"status": "error",
|
||||
"schema_version": 3,
|
||||
"cached": False,
|
||||
"updated_at": datetime.now().astimezone().isoformat(timespec="seconds"),
|
||||
"notice": "龙虎榜数据暂不可用，请稍后重试。",
|
||||
},
|
||||
"summary": {
|
||||
"trader_count": 0,
|
||||
"identity_count": 0,
|
||||
"operation_count": 0,
|
||||
"active_stock_count": 0,
|
||||
"seat_net_buy_million": 0,
|
||||
"unclassified_count": 0,
|
||||
"directory_count": 0,
|
||||
},
|
||||
"traders": [],
|
||||
"unclassified_seats": [],
|
||||
"rows": [],
|
||||
}
|
||||
payload["meta"]["cached"] = False
|
||||
if payload.get("meta", {}).get("status") == "success":
|
||||
self.database.save_data_snapshot(cache_kind, normalized_date, "tushare", payload)
|
||||
return payload
|
||||
|
||||
return {
|
||||
"meta": {
|
||||
"requested_date": f"{normalized_date[:4]}-{normalized_date[4:6]}-{normalized_date[6:8]}",
|
||||
"trade_date": f"{normalized_date[:4]}-{normalized_date[4:6]}-{normalized_date[6:8]}",
|
||||
"source": "unavailable",
|
||||
"status": "unavailable",
|
||||
"schema_version": 3,
|
||||
"cached": False,
|
||||
"notice": "龙虎榜数据暂不可用，请联系管理员检查行情配置。",
|
||||
},
|
||||
"summary": {
|
||||
"trader_count": 0,
|
||||
"identity_count": 0,
|
||||
"operation_count": 0,
|
||||
"active_stock_count": 0,
|
||||
"seat_net_buy_million": 0,
|
||||
"unclassified_count": 0,
|
||||
"directory_count": 0,
|
||||
},
|
||||
"traders": [],
|
||||
"unclassified_seats": [],
|
||||
"rows": [],
|
||||
}
|
||||
|
||||
|
||||
|
||||
def _apply_seat_aliases(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||
aliases = self.database.list_seat_aliases()
|
||||
result = dict(payload)
|
||||
rows = payload.get("rows") or []
|
||||
for row in rows:
|
||||
for institution in row.get("institutions") or []:
|
||||
institution["alias"] = aliases.get(institution.get("seat_name", ""), "")
|
||||
traders: dict[tuple[str, str], dict[str, Any]] = {}
|
||||
unclassified: dict[str, dict[str, Any]] = {}
|
||||
seen_operations: set[tuple[Any, ...]] = set()
|
||||
builtin_aliases = {
|
||||
"国泰海通证券股份有限公司南京太平南路证券营业部": "作手新一",
|
||||
}
|
||||
|
||||
for row in rows:
|
||||
for institution in row.get("institutions") or []:
|
||||
seat_name = str(institution.get("seat_name") or "未知席位").strip()
|
||||
saved_alias = str(institution.get("alias") or "").strip()
|
||||
builtin_alias = builtin_aliases.get(seat_name, "")
|
||||
if saved_alias or builtin_alias:
|
||||
identity_name = saved_alias or builtin_alias
|
||||
identity_type = "trader"
|
||||
recognized = True
|
||||
identity_source = "manual" if saved_alias else "builtin"
|
||||
elif "机构专用" in seat_name:
|
||||
identity_name = "机构专用"
|
||||
identity_type = "institution"
|
||||
recognized = True
|
||||
identity_source = "system"
|
||||
elif "沪股通专用" in seat_name or "深股通专用" in seat_name:
|
||||
identity_name = "北向资金"
|
||||
identity_type = "channel"
|
||||
recognized = True
|
||||
identity_source = "system"
|
||||
else:
|
||||
identity_name = seat_name
|
||||
identity_type = "unclassified"
|
||||
recognized = False
|
||||
identity_source = "raw"
|
||||
|
||||
buy = round(float(institution.get("buy_million") or 0), 2)
|
||||
sell = round(float(institution.get("sell_million") or 0), 2)
|
||||
net_buy = round(float(institution.get("net_buy_million") or 0), 2)
|
||||
operation_key = (row.get("code"), seat_name, buy, sell, net_buy)
|
||||
if operation_key in seen_operations:
|
||||
continue
|
||||
seen_operations.add(operation_key)
|
||||
|
||||
group_key = (identity_type, identity_name)
|
||||
group = traders.setdefault(
|
||||
group_key,
|
||||
{
|
||||
"name": identity_name,
|
||||
"identity_type": identity_type,
|
||||
"identity_source": identity_source,
|
||||
"recognized": recognized,
|
||||
"buy_million": 0.0,
|
||||
"sell_million": 0.0,
|
||||
"net_buy_million": 0.0,
|
||||
"seat_names": set(),
|
||||
"stock_codes": set(),
|
||||
"operations": [],
|
||||
},
|
||||
)
|
||||
group["buy_million"] += buy
|
||||
group["sell_million"] += sell
|
||||
group["net_buy_million"] += net_buy
|
||||
group["seat_names"].add(seat_name)
|
||||
group["stock_codes"].add(str(row.get("code") or ""))
|
||||
group["operations"].append(
|
||||
{
|
||||
"code": row.get("code") or "",
|
||||
"name": row.get("name") or "--",
|
||||
"change": row.get("change") or 0,
|
||||
"direction": "买入" if net_buy > 0 else "卖出" if net_buy < 0 else "持平",
|
||||
"buy_million": buy,
|
||||
"sell_million": sell,
|
||||
"net_buy_million": net_buy,
|
||||
"reason": row.get("reason") or "--",
|
||||
"seat_name": seat_name,
|
||||
"seat_alias": identity_name if recognized else "",
|
||||
}
|
||||
)
|
||||
|
||||
if not recognized:
|
||||
pending = unclassified.setdefault(
|
||||
seat_name,
|
||||
{
|
||||
"seat_name": seat_name,
|
||||
"stock_codes": set(),
|
||||
"operation_count": 0,
|
||||
"buy_million": 0.0,
|
||||
"sell_million": 0.0,
|
||||
"net_buy_million": 0.0,
|
||||
},
|
||||
)
|
||||
pending["stock_codes"].add(str(row.get("code") or ""))
|
||||
pending["operation_count"] += 1
|
||||
pending["buy_million"] += buy
|
||||
pending["sell_million"] += sell
|
||||
pending["net_buy_million"] += net_buy
|
||||
|
||||
type_order = {"trader": 0, "institution": 1, "channel": 2, "unclassified": 3}
|
||||
aggregated = list(traders.values())
|
||||
aggregated.sort(
|
||||
key=lambda item: (
|
||||
type_order.get(item["identity_type"], 9),
|
||||
-abs(item["net_buy_million"]),
|
||||
item["name"],
|
||||
)
|
||||
)
|
||||
for index, group in enumerate(aggregated, start=1):
|
||||
group["id"] = f"identity-{index}"
|
||||
group["buy_million"] = round(group["buy_million"], 2)
|
||||
group["sell_million"] = round(group["sell_million"], 2)
|
||||
group["net_buy_million"] = round(group["net_buy_million"], 2)
|
||||
group["seat_count"] = len(group.pop("seat_names"))
|
||||
group["stock_count"] = len(group.pop("stock_codes"))
|
||||
group["operation_count"] = len(group["operations"])
|
||||
group["operations"].sort(
|
||||
key=lambda item: abs(float(item.get("net_buy_million") or 0)), reverse=True
|
||||
)
|
||||
|
||||
pending_seats = list(unclassified.values())
|
||||
for pending in pending_seats:
|
||||
pending["stock_count"] = len(pending.pop("stock_codes"))
|
||||
pending["buy_million"] = round(pending["buy_million"], 2)
|
||||
pending["sell_million"] = round(pending["sell_million"], 2)
|
||||
pending["net_buy_million"] = round(pending["net_buy_million"], 2)
|
||||
pending_seats.sort(key=lambda item: abs(item["net_buy_million"]), reverse=True)
|
||||
|
||||
operation_count = sum(item["operation_count"] for item in aggregated)
|
||||
active_stocks = {
|
||||
operation["code"] for item in aggregated for operation in item["operations"]
|
||||
}
|
||||
seat_net_buy = round(sum(item["net_buy_million"] for item in aggregated), 2)
|
||||
result["rows"] = rows
|
||||
result["traders"] = aggregated
|
||||
result["unclassified_seats"] = pending_seats
|
||||
result["summary"] = {
|
||||
**(payload.get("summary") or {}),
|
||||
"trader_count": sum(item["identity_type"] == "trader" for item in aggregated),
|
||||
"identity_count": len(aggregated),
|
||||
"operation_count": operation_count,
|
||||
"active_stock_count": len(active_stocks),
|
||||
"seat_net_buy_million": seat_net_buy,
|
||||
"unclassified_count": len(pending_seats),
|
||||
}
|
||||
return result
|
||||
|
||||
|
||||
SERVICE = DashboardService()
|
||||
|
||||
|
||||
@@ -8,7 +8,7 @@ from backend.data import DataGateway, build_data_gateway
|
||||
from backend.database.repositories import RepositoryBundle, build_repository_bundle
|
||||
from backend.features.alerts import AlertService
|
||||
from backend.features.review import TradeJournalService
|
||||
from backend.features.screener import StrategyTrackingService
|
||||
from backend.features.screener.tracking import StrategyTrackingService
|
||||
from backend.jobs import InProcessJobRunner, JobRegistry, SQLiteJobRunRepository
|
||||
from database import ReviewDatabase
|
||||
from mentor_agent import MentorSkillRegistry
|
||||
|
||||
@@ -0,0 +1,4 @@
|
||||
from .repository import AuctionRepositoryMixin
|
||||
from .service import AuctionServiceMixin
|
||||
|
||||
__all__ = ["AuctionRepositoryMixin", "AuctionServiceMixin"]
|
||||
@@ -0,0 +1,63 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any
|
||||
|
||||
|
||||
class AuctionRepositoryMixin:
|
||||
def upsert_auction_factors(self, rows: list[dict[str, Any]]) -> int:
|
||||
values = []
|
||||
for row in rows:
|
||||
trade_date = str(row.get("trade_date") or "")
|
||||
ts_code = str(row.get("ts_code") or "")
|
||||
price = float(row.get("price") or 0)
|
||||
pre_close = float(row.get("pre_close") or 0)
|
||||
if not trade_date or not ts_code or price <= 0 or pre_close <= 0:
|
||||
continue
|
||||
values.append(
|
||||
(
|
||||
trade_date,
|
||||
ts_code,
|
||||
price,
|
||||
pre_close,
|
||||
(price / pre_close - 1) * 100,
|
||||
float(row.get("vol") or 0),
|
||||
float(row.get("amount") or 0),
|
||||
float(row.get("turnover_rate") or 0),
|
||||
float(row.get("volume_ratio") or 0),
|
||||
)
|
||||
)
|
||||
with self.connect() as connection:
|
||||
connection.executemany(
|
||||
"""
|
||||
INSERT INTO auction_factors
|
||||
(trade_date, ts_code, price, pre_close, change, vol, amount,
|
||||
turnover_rate, volume_ratio)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(trade_date, ts_code) DO UPDATE SET
|
||||
price=excluded.price, pre_close=excluded.pre_close,
|
||||
change=excluded.change, vol=excluded.vol, amount=excluded.amount,
|
||||
turnover_rate=excluded.turnover_rate,
|
||||
volume_ratio=excluded.volume_ratio
|
||||
""",
|
||||
values,
|
||||
)
|
||||
return len(values)
|
||||
|
||||
def auction_factor_dates(self, end_date: str = "", limit: int = 80) -> list[str]:
|
||||
where = "WHERE trade_date <= ?" if end_date else ""
|
||||
parameters: tuple[Any, ...] = (end_date, limit) if end_date else (limit,)
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute(
|
||||
f"SELECT DISTINCT trade_date FROM auction_factors {where} "
|
||||
"ORDER BY trade_date DESC LIMIT ?",
|
||||
parameters,
|
||||
).fetchall()
|
||||
return [row["trade_date"] for row in reversed(rows)]
|
||||
|
||||
def auction_factors_for_date(self, trade_date: str) -> list[dict[str, Any]]:
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute(
|
||||
"SELECT * FROM auction_factors WHERE trade_date = ? ORDER BY ts_code",
|
||||
(trade_date,),
|
||||
).fetchall()
|
||||
return [dict(row) for row in rows]
|
||||
@@ -0,0 +1,13 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any
|
||||
|
||||
from backend.bootstrap.config import normalize_date
|
||||
from backend.features.market.insights import MarketInsightsService
|
||||
|
||||
|
||||
class AuctionServiceMixin:
|
||||
def auction_center(self, trade_date: str, force: bool = False) -> dict[str, Any]:
|
||||
return self._market_insights().auction_center(
|
||||
normalize_date(trade_date), force, self.current_user_id
|
||||
)
|
||||
@@ -0,0 +1,4 @@
|
||||
from .repository import DragonTigerRepositoryMixin
|
||||
from .service import DragonTigerServiceMixin
|
||||
|
||||
__all__ = ["DragonTigerRepositoryMixin", "DragonTigerServiceMixin"]
|
||||
@@ -0,0 +1,61 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from datetime import datetime
|
||||
from typing import Any
|
||||
|
||||
|
||||
class DragonTigerRepositoryMixin:
|
||||
def list_seat_aliases(self) -> dict[str, str]:
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute("SELECT seat_name, alias FROM seat_aliases").fetchall()
|
||||
return {row["seat_name"]: row["alias"] for row in rows}
|
||||
|
||||
def save_seat_alias(self, seat_name: str, alias: str) -> None:
|
||||
now = datetime.now().astimezone().isoformat(timespec="seconds")
|
||||
with self.connect() as connection:
|
||||
connection.execute(
|
||||
"""
|
||||
INSERT INTO seat_aliases (seat_name, alias, updated_at)
|
||||
VALUES (?, ?, ?)
|
||||
ON CONFLICT(seat_name) DO UPDATE SET
|
||||
alias = excluded.alias,
|
||||
updated_at = excluded.updated_at
|
||||
""",
|
||||
(seat_name, alias, now),
|
||||
)
|
||||
|
||||
def upsert_lhb_institutions(self, rows: list[dict[str, Any]]) -> int:
|
||||
grouped: dict[tuple[str, str], dict[str, float | int]] = {}
|
||||
for row in rows:
|
||||
trade_date = str(row.get("trade_date") or "")
|
||||
ts_code = str(row.get("ts_code") or "")
|
||||
seat_name = str(row.get("exalter") or row.get("seat_name") or "")
|
||||
if not trade_date or not ts_code or "机构专用" not in seat_name:
|
||||
continue
|
||||
group = grouped.setdefault(
|
||||
(trade_date, ts_code),
|
||||
{"net": 0.0, "buy": 0.0, "sell": 0.0, "seats": 0},
|
||||
)
|
||||
group["net"] = float(group["net"]) + float(row.get("net_buy") or row.get("net_amount") or 0)
|
||||
group["buy"] = float(group["buy"]) + float(row.get("buy") or row.get("buy_amount") or 0)
|
||||
group["sell"] = float(group["sell"]) + float(row.get("sell") or row.get("sell_amount") or 0)
|
||||
group["seats"] = int(group["seats"]) + 1
|
||||
values = [
|
||||
(trade_date, ts_code, item["net"], item["buy"], item["sell"], item["seats"])
|
||||
for (trade_date, ts_code), item in grouped.items()
|
||||
]
|
||||
with self.connect() as connection:
|
||||
connection.executemany(
|
||||
"""
|
||||
INSERT INTO lhb_institution_daily
|
||||
(trade_date, ts_code, net_buy_amount, buy_amount, sell_amount, seat_count)
|
||||
VALUES (?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(trade_date, ts_code) DO UPDATE SET
|
||||
net_buy_amount=excluded.net_buy_amount,
|
||||
buy_amount=excluded.buy_amount,
|
||||
sell_amount=excluded.sell_amount,
|
||||
seat_count=excluded.seat_count
|
||||
""",
|
||||
values,
|
||||
)
|
||||
return len(values)
|
||||
@@ -0,0 +1,288 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from datetime import datetime
|
||||
from typing import Any
|
||||
|
||||
from backend.bootstrap.config import normalize_date
|
||||
from backend.data.providers.tushare_client import TushareError
|
||||
|
||||
|
||||
class DragonTigerServiceMixin:
|
||||
def get_hot_money_profiles(self, force: bool = False) -> dict[str, Any]:
|
||||
cache_kind = "hot_money_profiles_v1"
|
||||
cache_key = "directory"
|
||||
cached = self.database.get_data_snapshot(cache_kind, cache_key)
|
||||
if cached and not force:
|
||||
cached["meta"] = {**cached.get("meta", {}), "cached": True}
|
||||
return cached
|
||||
if self.configured:
|
||||
try:
|
||||
payload = self._tushare_client().hot_money_profiles()
|
||||
except TushareError:
|
||||
if cached:
|
||||
cached["meta"] = {
|
||||
**cached.get("meta", {}),
|
||||
"cached": True,
|
||||
"stale": True,
|
||||
"notice": "名录暂未完成更新，当前展示最近一次收录结果。",
|
||||
}
|
||||
return cached
|
||||
return {
|
||||
"meta": {
|
||||
"source": "unavailable",
|
||||
"status": "unavailable",
|
||||
"schema_version": 1,
|
||||
"cached": False,
|
||||
"updated_at": datetime.now().astimezone().isoformat(timespec="seconds"),
|
||||
"notice": "游资名录暂不可用，请稍后重试。",
|
||||
},
|
||||
"summary": {
|
||||
"profile_count": 0,
|
||||
"described_count": 0,
|
||||
"organization_count": 0,
|
||||
},
|
||||
"profiles": [],
|
||||
}
|
||||
payload["meta"]["cached"] = False
|
||||
if payload.get("meta", {}).get("status") == "success":
|
||||
self.database.save_data_snapshot(cache_kind, cache_key, "tushare", payload)
|
||||
return payload
|
||||
if cached:
|
||||
cached["meta"] = {**cached.get("meta", {}), "cached": True}
|
||||
return cached
|
||||
return {
|
||||
"meta": {
|
||||
"source": "unavailable",
|
||||
"status": "unavailable",
|
||||
"schema_version": 1,
|
||||
"cached": False,
|
||||
"updated_at": datetime.now().astimezone().isoformat(timespec="seconds"),
|
||||
"notice": "游资名录暂不可用，请联系管理员检查行情配置。",
|
||||
},
|
||||
"summary": {
|
||||
"profile_count": 0,
|
||||
"described_count": 0,
|
||||
"organization_count": 0,
|
||||
},
|
||||
"profiles": [],
|
||||
}
|
||||
|
||||
def get_dragon_tiger(self, trade_date: str, force: bool = False) -> dict[str, Any]:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
cache_kind = "hot_money_detail_v3"
|
||||
if not force:
|
||||
cached = self.database.get_data_snapshot(cache_kind, normalized_date)
|
||||
if (
|
||||
cached
|
||||
and cached.get("meta", {}).get("source") == "tushare"
|
||||
and cached.get("meta", {}).get("status") == "success"
|
||||
and int(cached.get("meta", {}).get("schema_version") or 0) == 3
|
||||
):
|
||||
cached["meta"] = {**cached.get("meta", {}), "cached": True}
|
||||
return cached
|
||||
if self.configured:
|
||||
try:
|
||||
payload = self._tushare_client().dragon_tiger(normalized_date)
|
||||
except TushareError as exc:
|
||||
return {
|
||||
"meta": {
|
||||
"requested_date": f"{normalized_date[:4]}-{normalized_date[4:6]}-{normalized_date[6:8]}",
|
||||
"trade_date": f"{normalized_date[:4]}-{normalized_date[4:6]}-{normalized_date[6:8]}",
|
||||
"source": "tushare_error",
|
||||
"status": "error",
|
||||
"schema_version": 3,
|
||||
"cached": False,
|
||||
"updated_at": datetime.now().astimezone().isoformat(timespec="seconds"),
|
||||
"notice": "龙虎榜数据暂不可用，请稍后重试。",
|
||||
},
|
||||
"summary": {
|
||||
"trader_count": 0,
|
||||
"identity_count": 0,
|
||||
"operation_count": 0,
|
||||
"active_stock_count": 0,
|
||||
"seat_net_buy_million": 0,
|
||||
"unclassified_count": 0,
|
||||
"directory_count": 0,
|
||||
},
|
||||
"traders": [],
|
||||
"unclassified_seats": [],
|
||||
"rows": [],
|
||||
}
|
||||
payload["meta"]["cached"] = False
|
||||
if payload.get("meta", {}).get("status") == "success":
|
||||
self.database.save_data_snapshot(cache_kind, normalized_date, "tushare", payload)
|
||||
return payload
|
||||
|
||||
return {
|
||||
"meta": {
|
||||
"requested_date": f"{normalized_date[:4]}-{normalized_date[4:6]}-{normalized_date[6:8]}",
|
||||
"trade_date": f"{normalized_date[:4]}-{normalized_date[4:6]}-{normalized_date[6:8]}",
|
||||
"source": "unavailable",
|
||||
"status": "unavailable",
|
||||
"schema_version": 3,
|
||||
"cached": False,
|
||||
"notice": "龙虎榜数据暂不可用，请联系管理员检查行情配置。",
|
||||
},
|
||||
"summary": {
|
||||
"trader_count": 0,
|
||||
"identity_count": 0,
|
||||
"operation_count": 0,
|
||||
"active_stock_count": 0,
|
||||
"seat_net_buy_million": 0,
|
||||
"unclassified_count": 0,
|
||||
"directory_count": 0,
|
||||
},
|
||||
"traders": [],
|
||||
"unclassified_seats": [],
|
||||
"rows": [],
|
||||
}
|
||||
|
||||
def _apply_seat_aliases(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||
aliases = self.database.list_seat_aliases()
|
||||
result = dict(payload)
|
||||
rows = payload.get("rows") or []
|
||||
for row in rows:
|
||||
for institution in row.get("institutions") or []:
|
||||
institution["alias"] = aliases.get(institution.get("seat_name", ""), "")
|
||||
traders: dict[tuple[str, str], dict[str, Any]] = {}
|
||||
unclassified: dict[str, dict[str, Any]] = {}
|
||||
seen_operations: set[tuple[Any, ...]] = set()
|
||||
builtin_aliases = {
|
||||
"国泰海通证券股份有限公司南京太平南路证券营业部": "作手新一",
|
||||
}
|
||||
|
||||
for row in rows:
|
||||
for institution in row.get("institutions") or []:
|
||||
seat_name = str(institution.get("seat_name") or "未知席位").strip()
|
||||
saved_alias = str(institution.get("alias") or "").strip()
|
||||
builtin_alias = builtin_aliases.get(seat_name, "")
|
||||
if saved_alias or builtin_alias:
|
||||
identity_name = saved_alias or builtin_alias
|
||||
identity_type = "trader"
|
||||
recognized = True
|
||||
identity_source = "manual" if saved_alias else "builtin"
|
||||
elif "机构专用" in seat_name:
|
||||
identity_name = "机构专用"
|
||||
identity_type = "institution"
|
||||
recognized = True
|
||||
identity_source = "system"
|
||||
elif "沪股通专用" in seat_name or "深股通专用" in seat_name:
|
||||
identity_name = "北向资金"
|
||||
identity_type = "channel"
|
||||
recognized = True
|
||||
identity_source = "system"
|
||||
else:
|
||||
identity_name = seat_name
|
||||
identity_type = "unclassified"
|
||||
recognized = False
|
||||
identity_source = "raw"
|
||||
|
||||
buy = round(float(institution.get("buy_million") or 0), 2)
|
||||
sell = round(float(institution.get("sell_million") or 0), 2)
|
||||
net_buy = round(float(institution.get("net_buy_million") or 0), 2)
|
||||
operation_key = (row.get("code"), seat_name, buy, sell, net_buy)
|
||||
if operation_key in seen_operations:
|
||||
continue
|
||||
seen_operations.add(operation_key)
|
||||
|
||||
group_key = (identity_type, identity_name)
|
||||
group = traders.setdefault(
|
||||
group_key,
|
||||
{
|
||||
"name": identity_name,
|
||||
"identity_type": identity_type,
|
||||
"identity_source": identity_source,
|
||||
"recognized": recognized,
|
||||
"buy_million": 0.0,
|
||||
"sell_million": 0.0,
|
||||
"net_buy_million": 0.0,
|
||||
"seat_names": set(),
|
||||
"stock_codes": set(),
|
||||
"operations": [],
|
||||
},
|
||||
)
|
||||
group["buy_million"] += buy
|
||||
group["sell_million"] += sell
|
||||
group["net_buy_million"] += net_buy
|
||||
group["seat_names"].add(seat_name)
|
||||
group["stock_codes"].add(str(row.get("code") or ""))
|
||||
group["operations"].append(
|
||||
{
|
||||
"code": row.get("code") or "",
|
||||
"name": row.get("name") or "--",
|
||||
"change": row.get("change") or 0,
|
||||
"direction": "买入" if net_buy > 0 else "卖出" if net_buy < 0 else "持平",
|
||||
"buy_million": buy,
|
||||
"sell_million": sell,
|
||||
"net_buy_million": net_buy,
|
||||
"reason": row.get("reason") or "--",
|
||||
"seat_name": seat_name,
|
||||
"seat_alias": identity_name if recognized else "",
|
||||
}
|
||||
)
|
||||
|
||||
if not recognized:
|
||||
pending = unclassified.setdefault(
|
||||
seat_name,
|
||||
{
|
||||
"seat_name": seat_name,
|
||||
"stock_codes": set(),
|
||||
"operation_count": 0,
|
||||
"buy_million": 0.0,
|
||||
"sell_million": 0.0,
|
||||
"net_buy_million": 0.0,
|
||||
},
|
||||
)
|
||||
pending["stock_codes"].add(str(row.get("code") or ""))
|
||||
pending["operation_count"] += 1
|
||||
pending["buy_million"] += buy
|
||||
pending["sell_million"] += sell
|
||||
pending["net_buy_million"] += net_buy
|
||||
|
||||
type_order = {"trader": 0, "institution": 1, "channel": 2, "unclassified": 3}
|
||||
aggregated = list(traders.values())
|
||||
aggregated.sort(
|
||||
key=lambda item: (
|
||||
type_order.get(item["identity_type"], 9),
|
||||
-abs(item["net_buy_million"]),
|
||||
item["name"],
|
||||
)
|
||||
)
|
||||
for index, group in enumerate(aggregated, start=1):
|
||||
group["id"] = f"identity-{index}"
|
||||
group["buy_million"] = round(group["buy_million"], 2)
|
||||
group["sell_million"] = round(group["sell_million"], 2)
|
||||
group["net_buy_million"] = round(group["net_buy_million"], 2)
|
||||
group["seat_count"] = len(group.pop("seat_names"))
|
||||
group["stock_count"] = len(group.pop("stock_codes"))
|
||||
group["operation_count"] = len(group["operations"])
|
||||
group["operations"].sort(
|
||||
key=lambda item: abs(float(item.get("net_buy_million") or 0)), reverse=True
|
||||
)
|
||||
|
||||
pending_seats = list(unclassified.values())
|
||||
for pending in pending_seats:
|
||||
pending["stock_count"] = len(pending.pop("stock_codes"))
|
||||
pending["buy_million"] = round(pending["buy_million"], 2)
|
||||
pending["sell_million"] = round(pending["sell_million"], 2)
|
||||
pending["net_buy_million"] = round(pending["net_buy_million"], 2)
|
||||
pending_seats.sort(key=lambda item: abs(item["net_buy_million"]), reverse=True)
|
||||
|
||||
operation_count = sum(item["operation_count"] for item in aggregated)
|
||||
active_stocks = {
|
||||
operation["code"] for item in aggregated for operation in item["operations"]
|
||||
}
|
||||
seat_net_buy = round(sum(item["net_buy_million"] for item in aggregated), 2)
|
||||
result["rows"] = rows
|
||||
result["traders"] = aggregated
|
||||
result["unclassified_seats"] = pending_seats
|
||||
result["summary"] = {
|
||||
**(payload.get("summary") or {}),
|
||||
"trader_count": sum(item["identity_type"] == "trader" for item in aggregated),
|
||||
"identity_count": len(aggregated),
|
||||
"operation_count": operation_count,
|
||||
"active_stock_count": len(active_stocks),
|
||||
"seat_net_buy_million": seat_net_buy,
|
||||
"unclassified_count": len(pending_seats),
|
||||
}
|
||||
return result
|
||||
File diff suppressed because it is too large
Load Diff
@@ -6,6 +6,74 @@ from typing import Any
|
||||
|
||||
|
||||
class MarketRepositoryMixin:
|
||||
def upsert_stock_master(self, rows: list[dict[str, Any]]) -> int:
|
||||
now = datetime.now().astimezone().isoformat(timespec="seconds")
|
||||
values = [
|
||||
(
|
||||
row.get("ts_code", ""),
|
||||
str(row.get("ts_code", "")).split(".")[0],
|
||||
row.get("name") or "--",
|
||||
row.get("industry") or "",
|
||||
row.get("market") or "",
|
||||
str(row.get("list_date") or ""),
|
||||
now,
|
||||
)
|
||||
for row in rows if row.get("ts_code")
|
||||
]
|
||||
with self.connect() as connection:
|
||||
connection.executemany(
|
||||
"""
|
||||
INSERT INTO stock_master
|
||||
(ts_code, code, name, industry, market, list_date, updated_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(ts_code) DO UPDATE SET
|
||||
code=excluded.code, name=excluded.name, industry=excluded.industry,
|
||||
market=excluded.market, list_date=excluded.list_date, updated_at=excluded.updated_at
|
||||
""",
|
||||
values,
|
||||
)
|
||||
return len(values)
|
||||
|
||||
def list_stock_master(self) -> list[dict[str, Any]]:
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute(
|
||||
"SELECT ts_code, code, name, industry, market, list_date FROM stock_master"
|
||||
).fetchall()
|
||||
return [dict(row) for row in rows]
|
||||
|
||||
def upsert_daily_bars(self, rows: list[dict[str, Any]]) -> int:
|
||||
values = [
|
||||
(
|
||||
str(row.get("trade_date") or ""), row.get("ts_code", ""),
|
||||
float(row.get("open") or 0), float(row.get("high") or 0),
|
||||
float(row.get("low") or 0), float(row.get("close") or 0),
|
||||
float(row.get("pct_chg") or 0), float(row.get("vol") or 0),
|
||||
float(row.get("amount") or 0),
|
||||
)
|
||||
for row in rows if row.get("trade_date") and row.get("ts_code")
|
||||
]
|
||||
with self.connect() as connection:
|
||||
connection.executemany(
|
||||
"""
|
||||
INSERT INTO daily_bars
|
||||
(trade_date, ts_code, open, high, low, close, pct_chg, vol, amount)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(trade_date, ts_code) DO UPDATE SET
|
||||
open=excluded.open, high=excluded.high, low=excluded.low,
|
||||
close=excluded.close, pct_chg=excluded.pct_chg,
|
||||
vol=excluded.vol, amount=excluded.amount
|
||||
""",
|
||||
values,
|
||||
)
|
||||
return len(values)
|
||||
|
||||
def daily_bars_for_date(self, trade_date: str) -> list[dict[str, Any]]:
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute(
|
||||
"SELECT * FROM daily_bars WHERE trade_date = ? ORDER BY ts_code",
|
||||
(trade_date,),
|
||||
).fetchall()
|
||||
return [dict(row) for row in rows]
|
||||
def get_snapshot(self, trade_date: str) -> dict[str, Any] | None:
|
||||
with self.connect() as connection:
|
||||
row = connection.execute(
|
||||
|
||||
@@ -14,6 +14,7 @@ from backend.bootstrap.config import (
|
||||
from backend.data.providers.ifind_client import IfindError
|
||||
from backend.data.providers.tushare_client import TushareClient, TushareError
|
||||
from backend.features.market.charts import ChartDataError
|
||||
from backend.features.market.insights import MarketInsightsService
|
||||
from backend.features.sentiment.engine import SENTIMENT_ENGINE_VERSION
|
||||
|
||||
|
||||
@@ -36,6 +37,14 @@ THS_SEARCH_TYPES = {
|
||||
|
||||
|
||||
class MarketServiceMixin:
|
||||
def _market_insights(self) -> MarketInsightsService:
|
||||
if not self.configured:
|
||||
raise ValueError("行情数据尚未配置。")
|
||||
return MarketInsightsService(
|
||||
self.database,
|
||||
self._tushare_client(),
|
||||
ifind=self.ifind,
|
||||
)
|
||||
def _tushare_client(self) -> TushareClient:
|
||||
gateway = getattr(self, "data_gateway", None)
|
||||
if gateway is not None:
|
||||
|
||||
@@ -0,0 +1,4 @@
|
||||
from .repository import PopularityRepositoryMixin
|
||||
from .service import PopularityServiceMixin
|
||||
|
||||
__all__ = ["PopularityRepositoryMixin", "PopularityServiceMixin"]
|
||||
@@ -0,0 +1,37 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any
|
||||
|
||||
|
||||
class PopularityRepositoryMixin:
|
||||
def upsert_popularity_factors(self, rows: list[dict[str, Any]]) -> int:
|
||||
values = [
|
||||
(
|
||||
str(row.get("trade_date") or ""),
|
||||
str(row.get("ts_code") or ""),
|
||||
int(row["ths_rank"]) if row.get("ths_rank") not in (None, "") else None,
|
||||
int(row["dc_rank"]) if row.get("dc_rank") not in (None, "") else None,
|
||||
float(row.get("combined_score") or 0),
|
||||
int(row["rank_change"]) if row.get("rank_change") not in (None, "") else None,
|
||||
int(bool(row.get("dual_source"))),
|
||||
)
|
||||
for row in rows
|
||||
if row.get("trade_date") and row.get("ts_code")
|
||||
]
|
||||
with self.connect() as connection:
|
||||
connection.executemany(
|
||||
"""
|
||||
INSERT INTO popularity_factors
|
||||
(trade_date, ts_code, ths_rank, dc_rank, combined_score,
|
||||
rank_change, dual_source)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(trade_date, ts_code) DO UPDATE SET
|
||||
ths_rank=excluded.ths_rank,
|
||||
dc_rank=excluded.dc_rank,
|
||||
combined_score=excluded.combined_score,
|
||||
rank_change=excluded.rank_change,
|
||||
dual_source=excluded.dual_source
|
||||
""",
|
||||
values,
|
||||
)
|
||||
return len(values)
|
||||
@@ -0,0 +1,11 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any
|
||||
|
||||
from backend.bootstrap.config import normalize_date
|
||||
from backend.features.market.insights import MarketInsightsService
|
||||
|
||||
|
||||
class PopularityServiceMixin:
|
||||
def popularity(self, trade_date: str, force: bool = False) -> dict[str, Any]:
|
||||
return self._market_insights().popularity(normalize_date(trade_date), force)
|
||||
@@ -0,0 +1,5 @@
|
||||
"""Sector rotation history and constituent detail feature."""
|
||||
|
||||
from .service import RotationServiceMixin
|
||||
|
||||
__all__ = ["RotationServiceMixin"]
|
||||
@@ -0,0 +1,165 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any
|
||||
|
||||
from backend.bootstrap.config import normalize_date, validate_text
|
||||
from backend.data.providers.tushare_client import TushareError
|
||||
from backend.features.sentiment.engine import (
|
||||
build_sentiment_history,
|
||||
latest_contiguous_history,
|
||||
)
|
||||
|
||||
|
||||
class RotationServiceMixin:
|
||||
def rotation_history(self, trade_date: str, limit: int = 9) -> dict[str, Any]:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
# 板块轮动固定展示最近 9 个交易日，按由近到远排列。
|
||||
limit = 9
|
||||
snapshots = self.database.list_snapshot_payloads(normalized_date, 240)
|
||||
by_trade_date: dict[str, dict[str, Any]] = {}
|
||||
for snapshot in snapshots:
|
||||
meta = snapshot.get("meta") or {}
|
||||
actual_date = str(meta.get("trade_date") or snapshot.get("_snapshot_date") or "")
|
||||
compact_date = actual_date.replace("-", "")
|
||||
if len(compact_date) == 8:
|
||||
by_trade_date[compact_date] = snapshot
|
||||
|
||||
sentiment_dates = {
|
||||
str(row.get("trade_date") or "").replace("-", "")
|
||||
for row in latest_contiguous_history(build_sentiment_history(snapshots))
|
||||
}
|
||||
ordered_dates = sorted(
|
||||
date_key for date_key in by_trade_date
|
||||
if not sentiment_dates or date_key in sentiment_dates
|
||||
)[-limit:][::-1]
|
||||
rows = []
|
||||
for date_key in ordered_dates:
|
||||
snapshot = by_trade_date[date_key]
|
||||
sector_context = {
|
||||
str(item.get("name") or ""): item
|
||||
for item in snapshot.get("sectors") or []
|
||||
}
|
||||
sectors = []
|
||||
for item in (snapshot.get("sector_rotation") or [])[:12]:
|
||||
name = str(item.get("name") or "").strip()
|
||||
context = sector_context.get(name, {})
|
||||
sectors.append(
|
||||
{
|
||||
"name": name,
|
||||
"rank": int(item.get("rank") or len(sectors) + 1),
|
||||
"trend": item.get("trend") or "持平",
|
||||
"count": int(item.get("count") or 0),
|
||||
"strength": float(item.get("strength") or context.get("strength") or 0),
|
||||
"change": float(context.get("change") or 0),
|
||||
"leader": item.get("leader") or context.get("leader") or "--",
|
||||
}
|
||||
)
|
||||
rows.append(
|
||||
{
|
||||
"trade_date": f"{date_key[:4]}-{date_key[4:6]}-{date_key[6:]}",
|
||||
"sectors": sectors,
|
||||
}
|
||||
)
|
||||
return {
|
||||
"trade_date": rows[0]["trade_date"] if rows else normalized_date,
|
||||
"available_days": len(ordered_dates),
|
||||
"requested_days": limit,
|
||||
"rows": rows,
|
||||
}
|
||||
|
||||
def rotation_sector_members(self, trade_date: str, sector_name: str) -> dict[str, Any]:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
sector_name = validate_text(sector_name, "板块名称", 60, required=True)
|
||||
dashboard = self.get_dashboard(normalized_date)
|
||||
actual_date = normalize_date(
|
||||
str((dashboard.get("meta") or {}).get("trade_date") or normalized_date)
|
||||
)
|
||||
cache_key = f"{actual_date}:{sector_name}"
|
||||
cached = self.database.get_data_snapshot("rotation_sector_members_v1", cache_key)
|
||||
if cached:
|
||||
cached["meta"] = {**(cached.get("meta") or {}), "cached": True}
|
||||
return cached
|
||||
if not self.configured:
|
||||
raise ValueError("板块成分数据暂不可用。")
|
||||
|
||||
representative = next(
|
||||
(
|
||||
item for item in dashboard.get("limits") or []
|
||||
if str(item.get("sector") or "").strip() == sector_name
|
||||
),
|
||||
None,
|
||||
)
|
||||
if not representative:
|
||||
raise ValueError("未找到该板块的代表股票，暂时无法核验成分股。")
|
||||
raw_code = str(representative.get("ts_code") or representative.get("code") or "")
|
||||
if "." in raw_code:
|
||||
ts_code = raw_code
|
||||
elif raw_code.startswith(("4", "8", "92")):
|
||||
ts_code = f"{raw_code}.BJ"
|
||||
elif raw_code.startswith(("6", "68", "90")):
|
||||
ts_code = f"{raw_code}.SH"
|
||||
else:
|
||||
ts_code = f"{raw_code}.SZ"
|
||||
client = self._tushare_client()
|
||||
try:
|
||||
industry = client.sw_stock_industry(ts_code, actual_date)
|
||||
sector_code = str(industry.get("l2_code") or "")
|
||||
members = client.sw_sector_members(sector_code, actual_date)
|
||||
except TushareError as exc:
|
||||
raise ValueError(f"该板块成分股暂不可用：{exc}") from exc
|
||||
|
||||
daily_rows = self.database.daily_bars_for_date(actual_date)
|
||||
if len(daily_rows) < 1000:
|
||||
try:
|
||||
daily_rows = client.query(
|
||||
"daily",
|
||||
{"trade_date": actual_date},
|
||||
"ts_code,trade_date,open,high,low,close,pct_chg,vol,amount",
|
||||
)
|
||||
if daily_rows:
|
||||
self.database.upsert_daily_bars(daily_rows)
|
||||
except TushareError:
|
||||
daily_rows = self.database.daily_bars_for_date(actual_date)
|
||||
daily_map = {str(item.get("ts_code") or ""): item for item in daily_rows}
|
||||
rows = []
|
||||
for member in members:
|
||||
member_code = str(member.get("ts_code") or "")
|
||||
quote = daily_map.get(member_code) or {}
|
||||
rows.append(
|
||||
{
|
||||
"code": member_code.split(".")[0],
|
||||
"ts_code": member_code,
|
||||
"name": str(member.get("name") or "--"),
|
||||
"change": quote.get("pct_chg"),
|
||||
"open": quote.get("open"),
|
||||
"close": quote.get("close"),
|
||||
"amount_billion": (
|
||||
round(float(quote.get("amount") or 0) / 100000, 2)
|
||||
if quote else None
|
||||
),
|
||||
"quoted": bool(quote),
|
||||
}
|
||||
)
|
||||
rows.sort(
|
||||
key=lambda item: (
|
||||
bool(item.get("quoted")),
|
||||
float(item.get("change") or -999),
|
||||
float(item.get("amount_billion") or 0),
|
||||
),
|
||||
reverse=True,
|
||||
)
|
||||
result = {
|
||||
"meta": {
|
||||
"trade_date": self._display_compact_date(actual_date),
|
||||
"sector_name": str(industry.get("l2_name") or sector_name),
|
||||
"sector_code": sector_code,
|
||||
"member_count": len(rows),
|
||||
"quoted_count": sum(bool(item.get("quoted")) for item in rows),
|
||||
"cached": False,
|
||||
},
|
||||
"rows": rows,
|
||||
}
|
||||
self.database.save_data_snapshot(
|
||||
"rotation_sector_members_v1", cache_key, "tushare", result
|
||||
)
|
||||
return result
|
||||
@@ -1,3 +1 @@
|
||||
from .tracking import StrategyTrackingService
|
||||
|
||||
__all__ = ["StrategyTrackingService"]
|
||||
"""Stock screening, custom selection, and strategy tracking feature."""
|
||||
|
||||
@@ -0,0 +1,146 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import time
|
||||
import urllib.error
|
||||
import urllib.request
|
||||
from typing import Any
|
||||
|
||||
from screener import FACTOR_FIELDS, REGIMES
|
||||
|
||||
|
||||
class LLMCompilerError(RuntimeError):
|
||||
pass
|
||||
|
||||
|
||||
def test_llm_connection(
|
||||
api_key: str,
|
||||
base_url: str,
|
||||
model: str,
|
||||
timeout: int = 30,
|
||||
) -> dict[str, Any]:
|
||||
if not api_key or not model:
|
||||
raise LLMCompilerError("API Key 或模型未配置。")
|
||||
endpoint = f"{base_url.rstrip('/')}/chat/completions"
|
||||
payload = json.dumps(
|
||||
{
|
||||
"model": model,
|
||||
"messages": [{"role": "user", "content": "只回复 OK"}],
|
||||
"stream": False,
|
||||
},
|
||||
ensure_ascii=False,
|
||||
).encode("utf-8")
|
||||
request = urllib.request.Request(
|
||||
endpoint,
|
||||
data=payload,
|
||||
headers={
|
||||
"Content-Type": "application/json",
|
||||
"Authorization": f"Bearer {api_key}",
|
||||
"User-Agent": "XiaobaiReviewWeb/0.5",
|
||||
},
|
||||
method="POST",
|
||||
)
|
||||
started = time.perf_counter()
|
||||
try:
|
||||
with urllib.request.urlopen(request, timeout=timeout) as response:
|
||||
result = json.loads(response.read().decode("utf-8"))
|
||||
reply = str(result["choices"][0]["message"]["content"]).strip()
|
||||
except urllib.error.HTTPError as exc:
|
||||
raise LLMCompilerError(_http_error_message(exc)) from exc
|
||||
except (urllib.error.URLError, TimeoutError, json.JSONDecodeError, KeyError, IndexError) as exc:
|
||||
raise LLMCompilerError(f"模型连接测试失败：{exc}") from exc
|
||||
return {
|
||||
"ok": True,
|
||||
"model": model,
|
||||
"reply": reply[:100],
|
||||
"latency_ms": round((time.perf_counter() - started) * 1000),
|
||||
}
|
||||
|
||||
|
||||
def compile_strategy_with_llm(
|
||||
prompt: str,
|
||||
regime: str,
|
||||
api_key: str,
|
||||
base_url: str,
|
||||
model: str,
|
||||
timeout: int = 45,
|
||||
) -> dict[str, Any]:
|
||||
if not api_key or not model:
|
||||
raise LLMCompilerError("尚未配置 LLM API Key 或模型。")
|
||||
endpoint = f"{base_url.rstrip('/')}/chat/completions"
|
||||
schema = {
|
||||
"name": "策略名称",
|
||||
"description": "策略说明",
|
||||
"regimes": [regime],
|
||||
"formula": {
|
||||
"universe": {"exclude_st": True, "listed_days_min": 120},
|
||||
"filters": [{"field": "return_5d", "op": ">=", "value": 0}],
|
||||
"score": [{"field": "sector_strength", "weight": 0.3, "direction": "desc"}],
|
||||
"limit": 15,
|
||||
"min_score": 0.55,
|
||||
},
|
||||
}
|
||||
system_prompt = (
|
||||
"你是A股量化策略编译器。只输出JSON对象，不输出Markdown。"
|
||||
"不得生成Python、SQL、网络请求或未提供的因子。"
|
||||
f"当前市场阶段为{REGIMES.get(regime, regime)}。"
|
||||
f"可用因子为：{json.dumps(FACTOR_FIELDS, ensure_ascii=False)}。"
|
||||
"运算符只能使用 >, >=, <, <=, ==, !=, between, in。"
|
||||
"score权重均大于0且不超过1，direction只能是asc或desc。"
|
||||
"退潮和冰点策略必须提高门槛并允许结果为空。"
|
||||
f"严格遵循以下结构：{json.dumps(schema, ensure_ascii=False)}"
|
||||
)
|
||||
payload = json.dumps(
|
||||
{
|
||||
"model": model,
|
||||
"messages": [
|
||||
{"role": "system", "content": system_prompt},
|
||||
{"role": "user", "content": prompt[:3000]},
|
||||
],
|
||||
"stream": False,
|
||||
},
|
||||
ensure_ascii=False,
|
||||
).encode("utf-8")
|
||||
request = urllib.request.Request(
|
||||
endpoint,
|
||||
data=payload,
|
||||
headers={
|
||||
"Content-Type": "application/json",
|
||||
"Authorization": f"Bearer {api_key}",
|
||||
"User-Agent": "XiaobaiReviewWeb/0.4",
|
||||
},
|
||||
method="POST",
|
||||
)
|
||||
try:
|
||||
with urllib.request.urlopen(request, timeout=timeout) as response:
|
||||
result = json.loads(response.read().decode("utf-8"))
|
||||
content = result["choices"][0]["message"]["content"].strip()
|
||||
if content.startswith("```"):
|
||||
content = content.strip("`")
|
||||
if content.startswith("json"):
|
||||
content = content[4:].strip()
|
||||
compiled = json.loads(content)
|
||||
except urllib.error.HTTPError as exc:
|
||||
raise LLMCompilerError(_http_error_message(exc).replace("模型连接测试", "LLM 策略编译")) from exc
|
||||
except (urllib.error.URLError, TimeoutError, json.JSONDecodeError, KeyError, IndexError) as exc:
|
||||
raise LLMCompilerError(f"LLM 策略编译失败：{exc}") from exc
|
||||
compiled["compiler"] = "llm"
|
||||
compiled["model"] = model
|
||||
return compiled
|
||||
|
||||
|
||||
def _http_error_message(exc: urllib.error.HTTPError) -> str:
|
||||
detail = ""
|
||||
try:
|
||||
payload = json.loads(exc.read().decode("utf-8", errors="replace"))
|
||||
error = payload.get("error")
|
||||
if isinstance(error, dict):
|
||||
detail = str(error.get("message") or error.get("code") or "")
|
||||
elif error:
|
||||
detail = str(error)
|
||||
elif payload.get("message"):
|
||||
detail = str(payload["message"])
|
||||
except (json.JSONDecodeError, OSError):
|
||||
detail = ""
|
||||
suffix = f"：{detail[:300]}" if detail else ""
|
||||
return f"模型连接测试失败（HTTP {exc.code}）{suffix}"
|
||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,814 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import sqlite3
|
||||
from datetime import datetime
|
||||
from typing import Any
|
||||
|
||||
from backend.features.sentiment.engine import build_sentiment_history
|
||||
|
||||
|
||||
def _optional_float(value: Any) -> float | None:
|
||||
if value in (None, ""):
|
||||
return None
|
||||
try:
|
||||
return float(value)
|
||||
except (TypeError, ValueError):
|
||||
return None
|
||||
|
||||
|
||||
class ScreenerRepositoryMixin:
|
||||
def upsert_benchmark_bars(self, rows: list[dict[str, Any]]) -> int:
|
||||
values = [
|
||||
(
|
||||
str(row.get("trade_date") or ""), str(row.get("ts_code") or ""),
|
||||
float(row.get("close") or 0), float(row.get("pct_chg") or 0),
|
||||
)
|
||||
for row in rows if row.get("trade_date") and row.get("ts_code")
|
||||
]
|
||||
with self.connect() as connection:
|
||||
connection.executemany(
|
||||
"""
|
||||
INSERT INTO benchmark_bars (trade_date, ts_code, close, pct_chg)
|
||||
VALUES (?, ?, ?, ?)
|
||||
ON CONFLICT(trade_date, ts_code) DO UPDATE SET
|
||||
close=excluded.close, pct_chg=excluded.pct_chg
|
||||
""",
|
||||
values,
|
||||
)
|
||||
return len(values)
|
||||
|
||||
def upsert_daily_indicators(self, rows: list[dict[str, Any]]) -> int:
|
||||
values = [
|
||||
(
|
||||
str(row.get("trade_date") or ""), row.get("ts_code", ""),
|
||||
float(row.get("turnover_rate") or 0), float(row.get("volume_ratio") or 0),
|
||||
float(row.get("total_mv") or 0), float(row.get("circ_mv") or 0),
|
||||
_optional_float(row.get("pe_ttm")), _optional_float(row.get("pb")),
|
||||
_optional_float(row.get("ps_ttm")), _optional_float(row.get("dv_ttm")),
|
||||
)
|
||||
for row in rows if row.get("trade_date") and row.get("ts_code")
|
||||
]
|
||||
with self.connect() as connection:
|
||||
connection.executemany(
|
||||
"""
|
||||
INSERT INTO daily_indicators
|
||||
(trade_date, ts_code, turnover_rate, volume_ratio, total_mv, circ_mv,
|
||||
pe_ttm, pb, ps_ttm, dv_ttm)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(trade_date, ts_code) DO UPDATE SET
|
||||
turnover_rate=excluded.turnover_rate, volume_ratio=excluded.volume_ratio,
|
||||
total_mv=excluded.total_mv, circ_mv=excluded.circ_mv,
|
||||
pe_ttm=excluded.pe_ttm, pb=excluded.pb,
|
||||
ps_ttm=excluded.ps_ttm, dv_ttm=excluded.dv_ttm
|
||||
""",
|
||||
values,
|
||||
)
|
||||
return len(values)
|
||||
|
||||
def upsert_fundamental_indicators(self, rows: list[dict[str, Any]]) -> int:
|
||||
values = [
|
||||
(
|
||||
str(row.get("end_date") or ""), str(row.get("ann_date") or ""),
|
||||
str(row.get("ts_code") or ""), _optional_float(row.get("roe")),
|
||||
_optional_float(row.get("roa")), _optional_float(row.get("roic")),
|
||||
_optional_float(row.get("grossprofit_margin")),
|
||||
_optional_float(row.get("netprofit_yoy")), _optional_float(row.get("or_yoy")),
|
||||
_optional_float(row.get("ocf_to_opincome")),
|
||||
)
|
||||
for row in rows
|
||||
if row.get("end_date") and row.get("ts_code")
|
||||
]
|
||||
with self.connect() as connection:
|
||||
connection.executemany(
|
||||
"""
|
||||
INSERT INTO fundamental_indicators
|
||||
(end_date, ann_date, ts_code, roe, roa, roic, grossprofit_margin,
|
||||
netprofit_yoy, or_yoy, ocf_to_opincome)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(end_date, ts_code) DO UPDATE SET
|
||||
ann_date=excluded.ann_date, roe=excluded.roe, roa=excluded.roa,
|
||||
roic=excluded.roic, grossprofit_margin=excluded.grossprofit_margin,
|
||||
netprofit_yoy=excluded.netprofit_yoy, or_yoy=excluded.or_yoy,
|
||||
ocf_to_opincome=excluded.ocf_to_opincome
|
||||
""",
|
||||
values,
|
||||
)
|
||||
return len(values)
|
||||
|
||||
def upsert_moneyflow(self, rows: list[dict[str, Any]]) -> int:
|
||||
values = []
|
||||
for row in rows:
|
||||
if not row.get("trade_date") or not row.get("ts_code"):
|
||||
continue
|
||||
large_net = (
|
||||
float(row.get("buy_lg_amount") or 0) + float(row.get("buy_elg_amount") or 0)
|
||||
- float(row.get("sell_lg_amount") or 0) - float(row.get("sell_elg_amount") or 0)
|
||||
)
|
||||
medium_net = float(row.get("buy_md_amount") or 0) - float(row.get("sell_md_amount") or 0)
|
||||
small_net = float(row.get("buy_sm_amount") or 0) - float(row.get("sell_sm_amount") or 0)
|
||||
values.append((
|
||||
str(row["trade_date"]), row["ts_code"], float(row.get("net_mf_amount") or 0),
|
||||
large_net, medium_net, small_net,
|
||||
))
|
||||
with self.connect() as connection:
|
||||
connection.executemany(
|
||||
"""
|
||||
INSERT INTO moneyflow_daily
|
||||
(trade_date, ts_code, net_mf_amount, large_net_amount, medium_net_amount, small_net_amount)
|
||||
VALUES (?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(trade_date, ts_code) DO UPDATE SET
|
||||
net_mf_amount=excluded.net_mf_amount, large_net_amount=excluded.large_net_amount,
|
||||
medium_net_amount=excluded.medium_net_amount, small_net_amount=excluded.small_net_amount
|
||||
""",
|
||||
values,
|
||||
)
|
||||
return len(values)
|
||||
|
||||
def upsert_earnings_events(self, rows: list[dict[str, Any]]) -> int:
|
||||
values = [
|
||||
(
|
||||
str(row.get("end_date") or ""),
|
||||
str(row.get("ann_date") or ""),
|
||||
str(row.get("ts_code") or ""),
|
||||
_optional_float(row.get("forecast_profit")),
|
||||
_optional_float(row.get("actual_profit")),
|
||||
_optional_float(row.get("surprise_pct")),
|
||||
_optional_float(row.get("revenue_yoy")),
|
||||
_optional_float(row.get("netprofit_yoy")),
|
||||
str(row.get("source") or ""),
|
||||
)
|
||||
for row in rows
|
||||
if row.get("end_date") and row.get("ann_date") and row.get("ts_code")
|
||||
]
|
||||
with self.connect() as connection:
|
||||
connection.executemany(
|
||||
"""
|
||||
INSERT INTO earnings_events
|
||||
(end_date, ann_date, ts_code, forecast_profit, actual_profit,
|
||||
surprise_pct, revenue_yoy, netprofit_yoy, source)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(end_date, ann_date, ts_code) DO UPDATE SET
|
||||
forecast_profit=excluded.forecast_profit,
|
||||
actual_profit=excluded.actual_profit,
|
||||
surprise_pct=excluded.surprise_pct,
|
||||
revenue_yoy=excluded.revenue_yoy,
|
||||
netprofit_yoy=excluded.netprofit_yoy,
|
||||
source=excluded.source
|
||||
""",
|
||||
values,
|
||||
)
|
||||
return len(values)
|
||||
|
||||
def daily_indicator_dates(self, end_date: str = "", limit: int = 400) -> list[str]:
|
||||
where = "WHERE trade_date <= ?" if end_date else ""
|
||||
parameters: tuple[Any, ...] = (end_date, limit) if end_date else (limit,)
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute(
|
||||
f"SELECT DISTINCT trade_date FROM daily_indicators {where} "
|
||||
"ORDER BY trade_date DESC LIMIT ?",
|
||||
parameters,
|
||||
).fetchall()
|
||||
return [row["trade_date"] for row in reversed(rows)]
|
||||
|
||||
def fundamental_periods(self) -> list[str]:
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute(
|
||||
"SELECT DISTINCT end_date FROM fundamental_indicators ORDER BY end_date"
|
||||
).fetchall()
|
||||
return [str(row["end_date"]) for row in rows]
|
||||
|
||||
def factor_dates(self, end_date: str = "", limit: int = 80) -> list[str]:
|
||||
where = "WHERE trade_date <= ?" if end_date else ""
|
||||
parameters: tuple[Any, ...] = (end_date, limit) if end_date else (limit,)
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute(
|
||||
f"SELECT DISTINCT trade_date FROM daily_bars {where} ORDER BY trade_date DESC LIMIT ?",
|
||||
parameters,
|
||||
).fetchall()
|
||||
return [row["trade_date"] for row in reversed(rows)]
|
||||
|
||||
def factor_health_summary(self, end_date: str) -> dict[str, Any]:
|
||||
dividend_start = f"{max(0, int(end_date[:4] or 0) - 5)}0101"
|
||||
with self.connect() as connection:
|
||||
market = connection.execute(
|
||||
"SELECT EXISTS(SELECT 1 FROM daily_bars WHERE trade_date <= ? LIMIT 1)",
|
||||
(end_date,),
|
||||
).fetchone()[0]
|
||||
auction = connection.execute(
|
||||
"SELECT EXISTS(SELECT 1 FROM auction_factors WHERE trade_date <= ? LIMIT 1)",
|
||||
(end_date,),
|
||||
).fetchone()[0]
|
||||
benchmark_rows = connection.execute(
|
||||
"SELECT COUNT(*) FROM benchmark_bars WHERE ts_code = '000300.SH' AND trade_date <= ?",
|
||||
(end_date,),
|
||||
).fetchone()[0]
|
||||
indicator_date = connection.execute(
|
||||
"SELECT MAX(trade_date) FROM daily_indicators WHERE trade_date <= ?",
|
||||
(end_date,),
|
||||
).fetchone()[0]
|
||||
if indicator_date:
|
||||
valuation_rows, valuation_available = connection.execute(
|
||||
"""
|
||||
SELECT COUNT(*), COALESCE(MAX(pe_ttm IS NOT NULL), 0)
|
||||
FROM daily_indicators WHERE trade_date = ?
|
||||
""",
|
||||
(indicator_date,),
|
||||
).fetchone()
|
||||
else:
|
||||
valuation_rows, valuation_available = 0, 0
|
||||
dividend_years = connection.execute(
|
||||
"""
|
||||
SELECT COUNT(DISTINCT substr(trade_date, 1, 4))
|
||||
FROM daily_indicators
|
||||
WHERE trade_date <= ? AND trade_date >= ?
|
||||
""",
|
||||
(end_date, dividend_start),
|
||||
).fetchone()[0]
|
||||
fundamental_rows = connection.execute(
|
||||
"""
|
||||
SELECT COUNT(*) FROM fundamental_indicators fi
|
||||
INNER JOIN (
|
||||
SELECT ts_code, MAX(ann_date || ':' || end_date) AS latest_key
|
||||
FROM fundamental_indicators
|
||||
WHERE ann_date = '' OR ann_date <= ?
|
||||
GROUP BY ts_code
|
||||
) latest
|
||||
ON latest.ts_code = fi.ts_code
|
||||
AND latest.latest_key = (fi.ann_date || ':' || fi.end_date)
|
||||
""",
|
||||
(end_date,),
|
||||
).fetchone()[0]
|
||||
moneyflow_dates = connection.execute(
|
||||
"""
|
||||
SELECT COUNT(DISTINCT trade_date)
|
||||
FROM moneyflow_daily
|
||||
WHERE trade_date IN (
|
||||
SELECT DISTINCT trade_date
|
||||
FROM daily_bars
|
||||
WHERE trade_date <= ?
|
||||
ORDER BY trade_date DESC
|
||||
LIMIT 5
|
||||
)
|
||||
""",
|
||||
(end_date,),
|
||||
).fetchone()[0]
|
||||
earnings_rows = connection.execute(
|
||||
"""
|
||||
SELECT COUNT(*) FROM earnings_events
|
||||
WHERE ann_date <= ? AND ann_date >= replace(date(?, '-45 day'), '-', '')
|
||||
""",
|
||||
(end_date, f"{end_date[:4]}-{end_date[4:6]}-{end_date[6:8]}"),
|
||||
).fetchone()[0]
|
||||
popularity_rows = connection.execute(
|
||||
"SELECT COUNT(*) FROM popularity_factors WHERE trade_date = ?",
|
||||
(end_date,),
|
||||
).fetchone()[0]
|
||||
institution_rows = connection.execute(
|
||||
"SELECT COUNT(*) FROM lhb_institution_daily WHERE trade_date = ?",
|
||||
(end_date,),
|
||||
).fetchone()[0]
|
||||
return {
|
||||
"market": bool(market),
|
||||
"auction": bool(auction),
|
||||
"benchmark": int(benchmark_rows or 0) >= 60,
|
||||
"benchmark_rows": int(benchmark_rows or 0),
|
||||
"valuation": bool(valuation_available),
|
||||
"fundamental": int(fundamental_rows or 0) >= 100,
|
||||
"dividend_history": int(dividend_years or 0) >= 4,
|
||||
"valuation_rows": int(valuation_rows or 0),
|
||||
"fundamental_rows": int(fundamental_rows or 0),
|
||||
"dividend_years": int(dividend_years or 0),
|
||||
"moneyflow_history": int(moneyflow_dates or 0) >= 5,
|
||||
"moneyflow_dates": int(moneyflow_dates or 0),
|
||||
"earnings_events": int(earnings_rows or 0) > 0,
|
||||
"earnings_event_rows": int(earnings_rows or 0),
|
||||
"popularity": int(popularity_rows or 0) > 0,
|
||||
"popularity_rows": int(popularity_rows or 0),
|
||||
"institutions": int(institution_rows or 0) > 0,
|
||||
"institution_rows": int(institution_rows or 0),
|
||||
}
|
||||
|
||||
def load_factor_data(self, end_date: str, limit_dates: int = 80) -> dict[str, Any]:
|
||||
dates = self.factor_dates(end_date, limit_dates)
|
||||
if not dates:
|
||||
return {
|
||||
"dates": [], "bars": [], "master": [], "indicators": [],
|
||||
"indicator_history": [], "indicator_series": [], "fundamentals": [],
|
||||
"moneyflow": [], "moneyflow_history": [], "auction": [],
|
||||
"benchmarks": [], "fundamental_history": [],
|
||||
"earnings_events": [], "popularity": [], "institutions": [],
|
||||
}
|
||||
placeholders = ",".join("?" for _ in dates)
|
||||
with self.connect() as connection:
|
||||
bars = connection.execute(
|
||||
f"SELECT * FROM daily_bars WHERE trade_date IN ({placeholders}) ORDER BY trade_date, ts_code",
|
||||
dates,
|
||||
).fetchall()
|
||||
master = connection.execute("SELECT * FROM stock_master").fetchall()
|
||||
indicators = connection.execute(
|
||||
"""
|
||||
SELECT * FROM daily_indicators
|
||||
WHERE trade_date = (
|
||||
SELECT MAX(trade_date) FROM daily_indicators WHERE trade_date <= ?
|
||||
)
|
||||
""",
|
||||
(end_date,),
|
||||
).fetchall()
|
||||
indicator_history = connection.execute(
|
||||
"""
|
||||
SELECT di.* FROM daily_indicators di
|
||||
INNER JOIN (
|
||||
SELECT ts_code, substr(trade_date, 1, 4) AS year_key,
|
||||
MAX(trade_date) AS max_date
|
||||
FROM daily_indicators
|
||||
WHERE trade_date <= ? AND trade_date >= ?
|
||||
GROUP BY ts_code, substr(trade_date, 1, 4)
|
||||
) latest
|
||||
ON latest.ts_code = di.ts_code AND latest.max_date = di.trade_date
|
||||
ORDER BY di.trade_date, di.ts_code
|
||||
""",
|
||||
(end_date, str(max(0, int(end_date[:4] or 0) - 5)) + "0101"),
|
||||
).fetchall()
|
||||
indicator_series = connection.execute(
|
||||
f"""
|
||||
SELECT trade_date, ts_code, turnover_rate, volume_ratio,
|
||||
total_mv, circ_mv, pe_ttm, pb, ps_ttm, dv_ttm
|
||||
FROM daily_indicators
|
||||
WHERE trade_date IN ({placeholders})
|
||||
ORDER BY trade_date, ts_code
|
||||
""",
|
||||
dates,
|
||||
).fetchall()
|
||||
fundamentals = connection.execute(
|
||||
"""
|
||||
SELECT fi.* FROM fundamental_indicators fi
|
||||
INNER JOIN (
|
||||
SELECT ts_code, MAX(ann_date || ':' || end_date) AS latest_key
|
||||
FROM fundamental_indicators
|
||||
WHERE ann_date = '' OR ann_date <= ?
|
||||
GROUP BY ts_code
|
||||
) latest
|
||||
ON latest.ts_code = fi.ts_code
|
||||
AND latest.latest_key = (fi.ann_date || ':' || fi.end_date)
|
||||
""",
|
||||
(end_date,),
|
||||
).fetchall()
|
||||
fundamental_history = connection.execute(
|
||||
"""
|
||||
SELECT * FROM fundamental_indicators
|
||||
WHERE ann_date = '' OR ann_date <= ?
|
||||
ORDER BY ann_date, end_date, ts_code
|
||||
""",
|
||||
(end_date,),
|
||||
).fetchall()
|
||||
moneyflow = connection.execute(
|
||||
"""
|
||||
SELECT * FROM moneyflow_daily
|
||||
WHERE trade_date = (
|
||||
SELECT MAX(trade_date) FROM moneyflow_daily WHERE trade_date <= ?
|
||||
)
|
||||
""",
|
||||
(end_date,),
|
||||
).fetchall()
|
||||
flow_dates = dates[-min(5, len(dates)):]
|
||||
flow_placeholders = ",".join("?" for _ in flow_dates)
|
||||
moneyflow_history = connection.execute(
|
||||
f"""
|
||||
SELECT * FROM moneyflow_daily
|
||||
WHERE trade_date IN ({flow_placeholders})
|
||||
ORDER BY trade_date, ts_code
|
||||
""",
|
||||
flow_dates,
|
||||
).fetchall()
|
||||
auction = connection.execute(
|
||||
"""
|
||||
SELECT * FROM auction_factors
|
||||
WHERE trade_date = (
|
||||
SELECT MAX(trade_date) FROM auction_factors WHERE trade_date <= ?
|
||||
)
|
||||
""",
|
||||
(end_date,),
|
||||
).fetchall()
|
||||
benchmarks = connection.execute(
|
||||
f"""
|
||||
SELECT * FROM benchmark_bars
|
||||
WHERE ts_code = '000300.SH' AND trade_date IN ({placeholders})
|
||||
ORDER BY trade_date
|
||||
""",
|
||||
dates,
|
||||
).fetchall()
|
||||
earnings_events = connection.execute(
|
||||
"""
|
||||
SELECT * FROM earnings_events
|
||||
WHERE ann_date <= ?
|
||||
ORDER BY ann_date, end_date, ts_code
|
||||
""",
|
||||
(end_date,),
|
||||
).fetchall()
|
||||
popularity = connection.execute(
|
||||
"SELECT * FROM popularity_factors WHERE trade_date = ? ORDER BY ts_code",
|
||||
(end_date,),
|
||||
).fetchall()
|
||||
institutions = connection.execute(
|
||||
"SELECT * FROM lhb_institution_daily WHERE trade_date = ? ORDER BY ts_code",
|
||||
(end_date,),
|
||||
).fetchall()
|
||||
return {
|
||||
"dates": dates,
|
||||
"bars": [dict(row) for row in bars],
|
||||
"master": [dict(row) for row in master],
|
||||
"indicators": [dict(row) for row in indicators],
|
||||
"indicator_history": [dict(row) for row in indicator_history],
|
||||
"indicator_series": [dict(row) for row in indicator_series],
|
||||
"fundamentals": [dict(row) for row in fundamentals],
|
||||
"fundamental_history": [dict(row) for row in fundamental_history],
|
||||
"moneyflow": [dict(row) for row in moneyflow],
|
||||
"moneyflow_history": [dict(row) for row in moneyflow_history],
|
||||
"auction": [dict(row) for row in auction],
|
||||
"benchmarks": [dict(row) for row in benchmarks],
|
||||
"earnings_events": [dict(row) for row in earnings_events],
|
||||
"popularity": [dict(row) for row in popularity],
|
||||
"institutions": [dict(row) for row in institutions],
|
||||
}
|
||||
|
||||
def snapshot_summaries(self, end_date: str, limit: int = 10) -> list[dict[str, Any]]:
|
||||
try:
|
||||
from sentiment_engine import build_sentiment_history
|
||||
except ModuleNotFoundError:
|
||||
from .sentiment_engine import build_sentiment_history
|
||||
|
||||
series = build_sentiment_history(self.list_snapshot_payloads(end_date, 260))
|
||||
return [
|
||||
{
|
||||
"trade_date": row["trade_date"],
|
||||
"sentiment_score": row["score"],
|
||||
"seal_rate": row["seal_rate"],
|
||||
"limit_up_count": row["limit_up_count"],
|
||||
"limit_down_count": row["limit_down_count"],
|
||||
"broken_count": row["broken_count"],
|
||||
"up_count": row["up_count"],
|
||||
"down_count": row["down_count"],
|
||||
"amount_billion": row["amount_billion"],
|
||||
}
|
||||
for row in series[-limit:]
|
||||
]
|
||||
|
||||
def save_screener_strategy(
|
||||
self, user_id: int | None, name: str, description: str, regimes: list[str], formula: dict[str, Any],
|
||||
builtin: bool = False, strategy_id: int | None = None,
|
||||
) -> int:
|
||||
now = datetime.now().astimezone().isoformat(timespec="seconds")
|
||||
regimes_json = json.dumps(regimes, ensure_ascii=False)
|
||||
formula_json = json.dumps(formula, ensure_ascii=False, separators=(",", ":"))
|
||||
with self.connect() as connection:
|
||||
if strategy_id:
|
||||
if builtin:
|
||||
cursor = connection.execute(
|
||||
"""
|
||||
UPDATE screener_strategies SET name=?, description=?, regimes=?, formula=?,
|
||||
builtin=1, user_id=NULL, updated_at=? WHERE id=? AND builtin=1
|
||||
""",
|
||||
(name, description, regimes_json, formula_json, now, strategy_id),
|
||||
)
|
||||
else:
|
||||
cursor = connection.execute(
|
||||
"""
|
||||
UPDATE screener_strategies SET name=?, description=?, regimes=?, formula=?,
|
||||
updated_at=? WHERE id=? AND builtin=0 AND user_id=?
|
||||
""",
|
||||
(name, description, regimes_json, formula_json, now, strategy_id, int(user_id or 0)),
|
||||
)
|
||||
if cursor.rowcount == 0:
|
||||
raise ValueError("选股策略不存在。")
|
||||
return strategy_id
|
||||
cursor = connection.execute(
|
||||
"""
|
||||
INSERT INTO screener_strategies
|
||||
(user_id, name, description, regimes, formula, builtin, created_at, updated_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||||
""",
|
||||
(None if builtin else int(user_id or 0), name, description, regimes_json, formula_json, int(builtin), now, now),
|
||||
)
|
||||
return int(cursor.lastrowid)
|
||||
|
||||
def list_screener_strategies(self, user_id: int | None = None) -> list[dict[str, Any]]:
|
||||
with self.connect() as connection:
|
||||
if user_id is None:
|
||||
rows = connection.execute(
|
||||
"SELECT * FROM screener_strategies WHERE builtin = 1 ORDER BY updated_at DESC, id"
|
||||
).fetchall()
|
||||
else:
|
||||
rows = connection.execute(
|
||||
"""
|
||||
SELECT * FROM screener_strategies
|
||||
WHERE builtin = 1 OR user_id = ?
|
||||
ORDER BY builtin DESC, updated_at DESC, id
|
||||
""",
|
||||
(int(user_id),),
|
||||
).fetchall()
|
||||
result = []
|
||||
for row in rows:
|
||||
item = dict(row)
|
||||
item["regimes"] = json.loads(item["regimes"])
|
||||
item["formula"] = json.loads(item["formula"])
|
||||
item["builtin"] = bool(item["builtin"])
|
||||
result.append(item)
|
||||
return result
|
||||
|
||||
def delete_screener_strategy(self, user_id: int, strategy_id: int) -> bool:
|
||||
with self.connect() as connection:
|
||||
row = connection.execute(
|
||||
"SELECT builtin, user_id FROM screener_strategies WHERE id = ?",
|
||||
(strategy_id,),
|
||||
).fetchone()
|
||||
if not row:
|
||||
raise ValueError("选股策略不存在。")
|
||||
if bool(row["builtin"]):
|
||||
raise ValueError("内置策略不能删除。")
|
||||
if int(row["user_id"] or 0) != int(user_id):
|
||||
raise ValueError("无权删除其他账号的策略。")
|
||||
cursor = connection.execute(
|
||||
"DELETE FROM screener_strategies WHERE id = ? AND builtin = 0 AND user_id = ?",
|
||||
(strategy_id, int(user_id)),
|
||||
)
|
||||
return cursor.rowcount > 0
|
||||
|
||||
def save_screener_run(
|
||||
self, user_id: int, trade_date: str, regime: str, strategy_name: str,
|
||||
formula: dict[str, Any], result: dict[str, Any], mode: str = "smart",
|
||||
) -> int:
|
||||
normalized_mode = mode if mode in {"smart", "curated", "quant"} else "smart"
|
||||
now = datetime.now().astimezone().isoformat(timespec="seconds")
|
||||
with self.connect() as connection:
|
||||
cursor = connection.execute(
|
||||
"""
|
||||
INSERT INTO screener_runs
|
||||
(user_id, trade_date, regime, mode, strategy_name, formula, result, created_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||||
""",
|
||||
(None if int(user_id) == 0 else int(user_id), trade_date, regime,
|
||||
normalized_mode, strategy_name,
|
||||
json.dumps(formula, ensure_ascii=False, separators=(",", ":")),
|
||||
json.dumps(result, ensure_ascii=False, separators=(",", ":")), now),
|
||||
)
|
||||
return int(cursor.lastrowid)
|
||||
|
||||
@staticmethod
|
||||
def _screener_run_payload(row: sqlite3.Row) -> dict[str, Any] | None:
|
||||
try:
|
||||
result = json.loads(row["result"])
|
||||
except json.JSONDecodeError:
|
||||
return None
|
||||
result.setdefault("meta", {}).update(
|
||||
{
|
||||
"run_id": int(row["id"]),
|
||||
"trade_date": str(row["trade_date"] or ""),
|
||||
"regime": str(row["regime"] or ""),
|
||||
"mode": str(row["mode"] or "smart"),
|
||||
"strategy_name": str(row["strategy_name"] or ""),
|
||||
"created_at": row["created_at"],
|
||||
}
|
||||
)
|
||||
return result
|
||||
|
||||
def latest_screener_run(
|
||||
self, user_id: int, trade_date: str, mode: str = "",
|
||||
) -> dict[str, Any] | None:
|
||||
owner_clause = "user_id IS NULL" if int(user_id) == 0 else "user_id = ?"
|
||||
parameters: tuple[Any, ...] = () if int(user_id) == 0 else (int(user_id),)
|
||||
parameters += (trade_date,)
|
||||
mode_clause = ""
|
||||
if mode in {"smart", "curated", "quant"}:
|
||||
mode_clause = " AND mode = ?"
|
||||
parameters += (mode,)
|
||||
with self.connect() as connection:
|
||||
row = connection.execute(
|
||||
f"""
|
||||
SELECT id, trade_date, regime, mode, strategy_name, result, created_at
|
||||
FROM screener_runs
|
||||
WHERE {owner_clause} AND trade_date <= ?{mode_clause}
|
||||
ORDER BY id DESC LIMIT 1
|
||||
""",
|
||||
parameters,
|
||||
).fetchone()
|
||||
return self._screener_run_payload(row) if row else None
|
||||
|
||||
def latest_screener_runs(self, user_id: int, trade_date: str) -> dict[str, dict[str, Any]]:
|
||||
owner_clause = "user_id IS NULL" if int(user_id) == 0 else "user_id = ?"
|
||||
parameters: tuple[Any, ...] = () if int(user_id) == 0 else (int(user_id),)
|
||||
parameters += (trade_date,)
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute(
|
||||
f"""
|
||||
SELECT runs.id, runs.trade_date, runs.regime, runs.mode,
|
||||
runs.strategy_name, runs.result, runs.created_at
|
||||
FROM screener_runs runs
|
||||
INNER JOIN (
|
||||
SELECT mode, MAX(id) AS id
|
||||
FROM screener_runs
|
||||
WHERE {owner_clause} AND trade_date <= ?
|
||||
GROUP BY mode
|
||||
) latest ON latest.id = runs.id
|
||||
""",
|
||||
parameters,
|
||||
).fetchall()
|
||||
results: dict[str, dict[str, Any]] = {}
|
||||
for row in rows:
|
||||
mode = str(row["mode"] or "smart")
|
||||
payload = self._screener_run_payload(row)
|
||||
if mode in {"smart", "curated", "quant"} and payload:
|
||||
results[mode] = payload
|
||||
return results
|
||||
|
||||
def latest_screener_context_runs(
|
||||
self, user_id: int, trade_date: str, limit: int = 60,
|
||||
) -> list[dict[str, Any]]:
|
||||
safe_limit = max(1, min(120, int(limit)))
|
||||
owner_clause = "user_id IS NULL" if int(user_id) == 0 else "user_id = ?"
|
||||
parameters: tuple[Any, ...] = () if int(user_id) == 0 else (int(user_id),)
|
||||
parameters += (trade_date, safe_limit)
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute(
|
||||
f"""
|
||||
WITH ranked AS (
|
||||
SELECT id, trade_date, regime, mode, strategy_name, result, created_at,
|
||||
ROW_NUMBER() OVER (
|
||||
PARTITION BY
|
||||
mode,
|
||||
CASE WHEN mode = 'smart' THEN regime ELSE '' END,
|
||||
CASE WHEN mode IN ('smart', 'curated') THEN strategy_name ELSE '' END
|
||||
ORDER BY id DESC
|
||||
) AS context_rank
|
||||
FROM screener_runs
|
||||
WHERE {owner_clause} AND trade_date <= ?
|
||||
)
|
||||
SELECT id, trade_date, regime, mode, strategy_name, result, created_at
|
||||
FROM ranked
|
||||
WHERE context_rank = 1
|
||||
ORDER BY id DESC
|
||||
LIMIT ?
|
||||
""",
|
||||
parameters,
|
||||
).fetchall()
|
||||
return [
|
||||
payload
|
||||
for row in rows
|
||||
if (payload := self._screener_run_payload(row)) is not None
|
||||
]
|
||||
|
||||
def screener_runs_for_date(
|
||||
self, user_id: int, trade_date: str, limit: int = 80,
|
||||
) -> list[dict[str, Any]]:
|
||||
safe_limit = max(1, min(160, int(limit)))
|
||||
owner_clause = "user_id IS NULL" if int(user_id) == 0 else "user_id = ?"
|
||||
parameters: tuple[Any, ...] = () if int(user_id) == 0 else (int(user_id),)
|
||||
parameters += (trade_date, safe_limit)
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute(
|
||||
f"""
|
||||
SELECT id, trade_date, regime, mode, strategy_name, result, created_at
|
||||
FROM screener_runs
|
||||
WHERE {owner_clause} AND trade_date = ?
|
||||
ORDER BY id DESC
|
||||
LIMIT ?
|
||||
""",
|
||||
parameters,
|
||||
).fetchall()
|
||||
result = []
|
||||
seen: set[tuple[str, str, str]] = set()
|
||||
for row in rows:
|
||||
key = (
|
||||
str(row["mode"] or "smart"),
|
||||
str(row["regime"] or ""),
|
||||
str(row["strategy_name"] or ""),
|
||||
)
|
||||
if key in seen:
|
||||
continue
|
||||
seen.add(key)
|
||||
payload = self._screener_run_payload(row)
|
||||
if payload is not None:
|
||||
result.append(payload)
|
||||
return result
|
||||
|
||||
def get_screener_run(self, user_id: int, run_id: int) -> dict[str, Any] | None:
|
||||
owner_clause = "user_id IS NULL" if int(user_id) == 0 else "user_id = ?"
|
||||
parameters: tuple[Any, ...] = (int(run_id),)
|
||||
if int(user_id) != 0:
|
||||
parameters += (int(user_id),)
|
||||
with self.connect() as connection:
|
||||
row = connection.execute(
|
||||
f"""
|
||||
SELECT id, trade_date, regime, mode, strategy_name, result, created_at
|
||||
FROM screener_runs WHERE id = ? AND {owner_clause}
|
||||
""",
|
||||
parameters,
|
||||
).fetchone()
|
||||
if not row:
|
||||
return None
|
||||
result = self._screener_run_payload(row)
|
||||
if result is None:
|
||||
return None
|
||||
result.setdefault("meta", {}).update(
|
||||
{
|
||||
"run_id": int(row["id"]),
|
||||
"trade_date": row["trade_date"],
|
||||
"mode": str(row["mode"] or "smart"),
|
||||
"created_at": row["created_at"],
|
||||
}
|
||||
)
|
||||
result["strategy_name"] = row["strategy_name"]
|
||||
result["regime"] = row["regime"]
|
||||
return result
|
||||
|
||||
def save_strategy_tracks(
|
||||
self,
|
||||
user_id: int,
|
||||
run_id: int,
|
||||
selection_date: str,
|
||||
strategy_name: str,
|
||||
candidates: list[dict[str, Any]],
|
||||
) -> int:
|
||||
now = datetime.now().astimezone().isoformat(timespec="seconds")
|
||||
values = []
|
||||
for item in candidates:
|
||||
ts_code = str(item.get("ts_code") or "").strip()
|
||||
code = str(item.get("code") or ts_code.split(".")[0]).strip()
|
||||
entry_price = float(item.get("price") or 0)
|
||||
if not ts_code or not code or entry_price <= 0:
|
||||
continue
|
||||
values.append(
|
||||
(
|
||||
int(user_id), int(run_id), selection_date, strategy_name, ts_code, code,
|
||||
str(item.get("name") or "--"), str(item.get("sector") or "其他"),
|
||||
entry_price, now, now,
|
||||
)
|
||||
)
|
||||
with self.connect() as connection:
|
||||
connection.executemany(
|
||||
"""
|
||||
INSERT INTO strategy_tracks
|
||||
(user_id, run_id, selection_date, strategy_name, ts_code, code,
|
||||
name, sector, entry_price, created_at, updated_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(user_id, run_id, ts_code) DO UPDATE SET
|
||||
name=excluded.name, sector=excluded.sector,
|
||||
entry_price=excluded.entry_price, updated_at=excluded.updated_at
|
||||
""",
|
||||
values,
|
||||
)
|
||||
return len(values)
|
||||
|
||||
def list_strategy_tracks(self, user_id: int, limit_batches: int = 12) -> list[dict[str, Any]]:
|
||||
limit_batches = max(1, min(50, int(limit_batches)))
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute(
|
||||
"""
|
||||
SELECT * FROM strategy_tracks
|
||||
WHERE user_id = ? AND run_id IN (
|
||||
SELECT run_id FROM strategy_tracks WHERE user_id = ?
|
||||
GROUP BY run_id ORDER BY run_id DESC LIMIT ?
|
||||
)
|
||||
ORDER BY run_id DESC, id
|
||||
""",
|
||||
(int(user_id), int(user_id), limit_batches),
|
||||
).fetchall()
|
||||
return [dict(row) for row in rows]
|
||||
|
||||
def delete_strategy_track(self, user_id: int, track_id: int) -> bool:
|
||||
with self.connect() as connection:
|
||||
cursor = connection.execute(
|
||||
"DELETE FROM strategy_tracks WHERE id = ? AND user_id = ?",
|
||||
(int(track_id), int(user_id)),
|
||||
)
|
||||
return cursor.rowcount > 0
|
||||
|
||||
def load_tracking_bars(
|
||||
self, targets: list[tuple[str, str]], limit: int = 5
|
||||
) -> dict[tuple[str, str], list[dict[str, Any]]]:
|
||||
unique_targets = set(targets)
|
||||
if not unique_targets:
|
||||
return {}
|
||||
codes = sorted({ts_code for ts_code, _ in unique_targets})
|
||||
earliest_date = min(selection_date for _, selection_date in unique_targets)
|
||||
placeholders = ",".join("?" for _ in codes)
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute(
|
||||
f"""
|
||||
SELECT ts_code, trade_date, open, high, low, close FROM daily_bars
|
||||
WHERE ts_code IN ({placeholders}) AND trade_date > ?
|
||||
ORDER BY ts_code, trade_date
|
||||
""",
|
||||
[*codes, earliest_date],
|
||||
).fetchall()
|
||||
by_code: dict[str, list[dict[str, Any]]] = {}
|
||||
for row in rows:
|
||||
item = dict(row)
|
||||
by_code.setdefault(str(item["ts_code"]), []).append(item)
|
||||
row_limit = max(1, min(20, int(limit)))
|
||||
return {
|
||||
(ts_code, selection_date): [
|
||||
row for row in by_code.get(ts_code, []) if row["trade_date"] > selection_date
|
||||
][:row_limit]
|
||||
for ts_code, selection_date in unique_targets
|
||||
}
|
||||
@@ -0,0 +1,435 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import copy
|
||||
import re
|
||||
from datetime import date, datetime
|
||||
from typing import Any
|
||||
|
||||
from backend.bootstrap.config import normalize_date, validate_text
|
||||
from backend.data.providers.tushare_client import TushareError
|
||||
from backend.llm import LLMGatewayError
|
||||
from backend.features.screener.compiler import (
|
||||
LLMCompilerError,
|
||||
compile_strategy_with_llm,
|
||||
)
|
||||
from backend.features.screener.engine import (
|
||||
FACTOR_FIELDS,
|
||||
FACTOR_GROUPS,
|
||||
REGIMES,
|
||||
FactorDataService,
|
||||
compile_local_strategy,
|
||||
)
|
||||
|
||||
|
||||
SCREENER_LIBRARY_VERSION = 8
|
||||
|
||||
|
||||
def automatic_screener_jobs(
|
||||
strategies: list[dict[str, Any]], regime_id: str
|
||||
) -> list[dict[str, Any]]:
|
||||
"""Build the close-of-day jobs; only stage screening is regime-gated."""
|
||||
smart_strategy = next(
|
||||
(
|
||||
item for item in strategies
|
||||
if item.get("formula", {}).get("meta", {}).get("library") != "curated"
|
||||
and regime_id in (item.get("regimes") or [])
|
||||
),
|
||||
None,
|
||||
)
|
||||
curated = [
|
||||
item for item in strategies
|
||||
if item.get("formula", {}).get("meta", {}).get("library") == "curated"
|
||||
]
|
||||
jobs = ([{"mode": "smart", "strategy": smart_strategy}] if smart_strategy else [])
|
||||
jobs.extend({"mode": "curated", "strategy": item} for item in curated)
|
||||
return jobs
|
||||
|
||||
|
||||
class ScreenerServiceMixin:
|
||||
@staticmethod
|
||||
def _strategy_missing_data(
|
||||
strategy: dict[str, Any], factor_dates: list[str], factor_health: dict[str, Any]
|
||||
) -> list[str]:
|
||||
formula = strategy.get("formula") or {}
|
||||
meta = formula.get("meta") or {}
|
||||
used_fields = {
|
||||
str(item.get("field") or "")
|
||||
for item in list(formula.get("filters") or []) + list(formula.get("score") or [])
|
||||
}
|
||||
valuation_fields = {"pe_ttm", "pb", "ps_ttm", "dividend_yield_ttm", "total_mv_billion"}
|
||||
fundamental_fields = {"roe", "roa", "roic", "gross_margin", "netprofit_yoy", "revenue_yoy", "ocf_to_opincome"}
|
||||
auction_fields = {"auction_change", "auction_amount_million", "auction_turnover_rate", "auction_volume_ratio"}
|
||||
missing = []
|
||||
required_history = max(21, min(260, int(meta.get("history_days") or 21)))
|
||||
if len(factor_dates) < required_history:
|
||||
missing.append(f"历史行情（需{required_history}日）")
|
||||
if used_fields & valuation_fields and not factor_health["valuation"]:
|
||||
missing.append("估值数据")
|
||||
if used_fields & fundamental_fields and not factor_health["fundamental"]:
|
||||
missing.append("财务质量")
|
||||
if meta.get("requires_valuation") and not factor_health["valuation"]:
|
||||
missing.append("估值数据")
|
||||
if meta.get("requires_fundamental") and not factor_health["fundamental"]:
|
||||
missing.append("财务质量")
|
||||
if "dividend_years" in used_fields and not factor_health["dividend_history"]:
|
||||
missing.append("历年分红")
|
||||
if used_fields & auction_fields and not factor_health["auction"]:
|
||||
missing.append("竞价数据")
|
||||
if meta.get("requires_benchmark") and not factor_health.get("benchmark"):
|
||||
missing.append("沪深300基准")
|
||||
if meta.get("requires_moneyflow_history") and not factor_health.get("moneyflow_history"):
|
||||
missing.append("近5日资金流")
|
||||
if meta.get("requires_earnings_events") and not factor_health.get("earnings_events"):
|
||||
missing.append("业绩预告与快报")
|
||||
if meta.get("requires_popularity") and not factor_health.get("popularity"):
|
||||
missing.append("当日人气榜")
|
||||
if meta.get("requires_institutions") and not factor_health.get("institutions"):
|
||||
missing.append("龙虎榜机构席位")
|
||||
return list(dict.fromkeys(missing))
|
||||
|
||||
def screener_setup(self, trade_date: str) -> dict[str, Any]:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
regime = self.screener.detect_regime(normalized_date)
|
||||
factor_dates = self.database.factor_dates(normalized_date, 300)
|
||||
auction_dates = self.database.auction_factor_dates(normalized_date, 100)
|
||||
factor_health = self.screener.factor_health(normalized_date)
|
||||
strategies = self.database.list_screener_strategies(self.current_user_id)
|
||||
for strategy in strategies:
|
||||
missing = self._strategy_missing_data(strategy, factor_dates, factor_health)
|
||||
strategy["data_ready"] = not missing
|
||||
strategy["missing_data"] = missing
|
||||
automatic_results = self.database.screener_runs_for_date(0, normalized_date)
|
||||
personal_results = self.database.screener_runs_for_date(
|
||||
self.current_user_id, normalized_date
|
||||
)
|
||||
recent_results = [
|
||||
*[item for item in automatic_results if item.get("meta", {}).get("mode") in {"smart", "curated"}],
|
||||
*[item for item in personal_results if item.get("meta", {}).get("mode") == "quant"],
|
||||
]
|
||||
latest_results: dict[str, dict[str, Any]] = {}
|
||||
for result in reversed(recent_results):
|
||||
mode = str(result.get("meta", {}).get("mode") or "smart")
|
||||
latest_results[mode] = result
|
||||
automatic_status = self.database.get_data_snapshot(
|
||||
"screener_auto_v1", normalized_date
|
||||
) or {}
|
||||
return {
|
||||
"trade_date": normalized_date,
|
||||
"regime": regime,
|
||||
"regimes": [{"id": key, "label": value} for key, value in REGIMES.items()],
|
||||
"strategies": strategies,
|
||||
"factor_fields": [{"id": key, "label": value} for key, value in FACTOR_FIELDS.items()],
|
||||
"factor_groups": [
|
||||
{
|
||||
"name": name,
|
||||
"fields": [{"id": field, "label": FACTOR_FIELDS[field]} for field in fields],
|
||||
}
|
||||
for name, fields in FACTOR_GROUPS.items()
|
||||
],
|
||||
"operators": [">", ">=", "<", "<=", "==", "between"],
|
||||
"factor_data": {
|
||||
"date_count": len(factor_dates),
|
||||
"start_date": factor_dates[0] if factor_dates else "",
|
||||
"end_date": factor_dates[-1] if factor_dates else "",
|
||||
"ready": len(factor_dates) >= 21,
|
||||
"auction_date_count": len(auction_dates),
|
||||
"auction_ready": bool(auction_dates and auction_dates[-1] == factor_dates[-1]) if factor_dates else False,
|
||||
"health": factor_health,
|
||||
},
|
||||
"llm": {
|
||||
"configured": self.llm_configured,
|
||||
"model": self.llm_primary_model if self.llm_configured else "",
|
||||
"fallback_configured": self.llm_fallback_configured,
|
||||
"fallback_model": self.llm_fallback_model if self.llm_fallback_configured else "",
|
||||
},
|
||||
"latest_results": latest_results,
|
||||
"recent_results": recent_results,
|
||||
"automatic_status": automatic_status,
|
||||
# Kept during the client transition for compatibility with older frontends.
|
||||
"latest_result": latest_results.get("smart"),
|
||||
}
|
||||
|
||||
def screener_tracking(self, limit: int = 12) -> dict[str, Any]:
|
||||
return self.strategy_tracking.list_tracking(self.current_user_id, limit)
|
||||
|
||||
def add_screener_tracking(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||
try:
|
||||
run_id = int(payload.get("run_id") or 0)
|
||||
except (TypeError, ValueError) as exc:
|
||||
raise ValueError("选股批次无效。") from exc
|
||||
code = str(payload.get("code") or "").strip()
|
||||
if run_id <= 0 or not re.fullmatch(r"\d{6}", code):
|
||||
raise ValueError("选股批次或股票代码无效。")
|
||||
return self.strategy_tracking.add_candidate(self.current_user_id, run_id, code)
|
||||
|
||||
def remove_screener_tracking(self, track_id: int) -> dict[str, Any]:
|
||||
return self.strategy_tracking.remove_candidate(self.current_user_id, track_id)
|
||||
|
||||
def refresh_screener_tracking(self, trade_date: str) -> dict[str, Any]:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
notice = ""
|
||||
if self.configured:
|
||||
try:
|
||||
FactorDataService(self.database, self._tushare_client()).sync(
|
||||
normalized_date, 15
|
||||
)
|
||||
except TushareError:
|
||||
notice = "最新日线暂未补齐，已按现有数据更新跟踪。"
|
||||
else:
|
||||
notice = "公共行情尚未配置，已按现有数据更新跟踪。"
|
||||
return {
|
||||
"tracking": self.screener_tracking(),
|
||||
"notice": notice,
|
||||
}
|
||||
|
||||
def sync_screener_data(self, trade_date: str, lookback: int = 45) -> dict[str, Any]:
|
||||
if not self.configured:
|
||||
raise ValueError("请先配置 Tushare Token。")
|
||||
normalized_date = normalize_date(trade_date)
|
||||
lookback = max(25, min(260, int(lookback)))
|
||||
with self.sync_lock:
|
||||
return FactorDataService(self.database, self._tushare_client()).sync(
|
||||
normalized_date, lookback
|
||||
)
|
||||
|
||||
def _schedule_automatic_screeners(
|
||||
self, trade_date: str, snapshot: dict[str, Any] | None = None
|
||||
) -> bool:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
now = datetime.now().astimezone()
|
||||
if (
|
||||
normalized_date != now.strftime("%Y%m%d")
|
||||
or now.weekday() >= 5
|
||||
or now.time().replace(tzinfo=None) < datetime.strptime("15:10", "%H:%M").time()
|
||||
or self.auto_screener_lock.locked()
|
||||
):
|
||||
return False
|
||||
snapshot = snapshot or self.database.get_snapshot(normalized_date) or {}
|
||||
actual_date = str((snapshot.get("meta") or {}).get("trade_date") or "").replace("-", "")
|
||||
if actual_date != normalized_date:
|
||||
return False
|
||||
marker = self.database.get_data_snapshot("screener_auto_v1", normalized_date) or {}
|
||||
if (
|
||||
marker.get("status") == "complete"
|
||||
and int(marker.get("library_version") or 0) == SCREENER_LIBRARY_VERSION
|
||||
):
|
||||
return False
|
||||
last_attempt = self._auto_screener_last_attempt.get(normalized_date)
|
||||
if last_attempt and (now - last_attempt).total_seconds() < 600:
|
||||
return False
|
||||
self._auto_screener_last_attempt[normalized_date] = now
|
||||
return self.jobs.submit(
|
||||
"screener.automatic",
|
||||
f"{normalized_date}:v{SCREENER_LIBRARY_VERSION}",
|
||||
lambda: self.run_automatic_screeners(normalized_date),
|
||||
{"trade_date": normalized_date, "trigger": "post-close"},
|
||||
)
|
||||
|
||||
def run_automatic_screeners(self, trade_date: str) -> dict[str, Any]:
|
||||
normalized_date = normalize_date(trade_date)
|
||||
with self.auto_screener_lock:
|
||||
started_at = datetime.now().astimezone().isoformat(timespec="seconds")
|
||||
status: dict[str, Any] = {
|
||||
"trade_date": normalized_date,
|
||||
"library_version": SCREENER_LIBRARY_VERSION,
|
||||
"status": "running",
|
||||
"started_at": started_at,
|
||||
"completed": [],
|
||||
"skipped": [],
|
||||
"failed": [],
|
||||
}
|
||||
self.database.save_data_snapshot(
|
||||
"screener_auto_v1", normalized_date, "system", status
|
||||
)
|
||||
try:
|
||||
factor_sync = FactorDataService(
|
||||
self.database, self._tushare_client()
|
||||
).sync(normalized_date, 260)
|
||||
factor_dates = self.database.factor_dates(normalized_date, 300)
|
||||
if not factor_dates or factor_dates[-1] != normalized_date:
|
||||
raise ValueError("当日收盘行情尚未入库")
|
||||
factor_health = self.screener.factor_health(normalized_date)
|
||||
regime = self.screener.detect_regime(normalized_date)
|
||||
regime_id = str(regime.get("id") or "repair")
|
||||
strategies = self.database.list_screener_strategies(None)
|
||||
jobs = automatic_screener_jobs(strategies, regime_id)
|
||||
existing = {
|
||||
(
|
||||
str(item.get("meta", {}).get("mode") or "smart"),
|
||||
str(item.get("meta", {}).get("strategy_name") or ""),
|
||||
)
|
||||
for item in self.database.screener_runs_for_date(0, normalized_date)
|
||||
if int(item.get("meta", {}).get("library_version") or 0)
|
||||
== SCREENER_LIBRARY_VERSION
|
||||
}
|
||||
required_history = max(
|
||||
[
|
||||
int((job["strategy"].get("formula", {}).get("meta", {}) or {}).get("history_days") or 80)
|
||||
for job in jobs if job.get("strategy")
|
||||
] or [80]
|
||||
)
|
||||
factors, actual_date = self.screener.build_factors(
|
||||
normalized_date, history_days=required_history
|
||||
)
|
||||
if actual_date != normalized_date:
|
||||
raise ValueError("当日因子尚未完成收盘定格")
|
||||
for job in jobs:
|
||||
strategy = job["strategy"]
|
||||
mode = str(job["mode"])
|
||||
name = str(strategy.get("name") or "未命名策略")
|
||||
if (mode, name) in existing:
|
||||
status["completed"].append({"mode": mode, "name": name, "cached": True})
|
||||
continue
|
||||
missing = self._strategy_missing_data(
|
||||
strategy, factor_dates, factor_health
|
||||
)
|
||||
if missing:
|
||||
status["skipped"].append(
|
||||
{"mode": mode, "name": name, "reason": "、".join(missing)}
|
||||
)
|
||||
continue
|
||||
try:
|
||||
formula = copy.deepcopy(strategy.get("formula") or {})
|
||||
formula.setdefault("meta", {})["library_version"] = (
|
||||
SCREENER_LIBRARY_VERSION
|
||||
)
|
||||
result = self.screener.screen(
|
||||
0,
|
||||
normalized_date,
|
||||
formula,
|
||||
regime_id,
|
||||
name,
|
||||
False,
|
||||
None,
|
||||
mode,
|
||||
factors,
|
||||
actual_date,
|
||||
)
|
||||
status["completed"].append(
|
||||
{
|
||||
"mode": mode,
|
||||
"name": name,
|
||||
"candidate_count": len(result.get("candidates") or []),
|
||||
}
|
||||
)
|
||||
except Exception as exc:
|
||||
status["failed"].append(
|
||||
{"mode": mode, "name": name, "reason": str(exc)}
|
||||
)
|
||||
status.update(
|
||||
{
|
||||
"status": "complete" if not status["failed"] else "partial",
|
||||
"finished_at": datetime.now().astimezone().isoformat(timespec="seconds"),
|
||||
"factor_sync": factor_sync,
|
||||
"regime": regime,
|
||||
}
|
||||
)
|
||||
except Exception as exc:
|
||||
status.update(
|
||||
{
|
||||
"status": "failed",
|
||||
"finished_at": datetime.now().astimezone().isoformat(timespec="seconds"),
|
||||
"error": str(exc),
|
||||
}
|
||||
)
|
||||
self.database.save_data_snapshot(
|
||||
"screener_auto_v1", normalized_date, "system", status
|
||||
)
|
||||
return status
|
||||
|
||||
def compile_screener_strategy(self, prompt: str, regime: str) -> dict[str, Any]:
|
||||
prompt = prompt.strip()
|
||||
if not prompt or len(prompt) > 3000:
|
||||
raise ValueError("策略描述应为 1 至 3000 个字符。")
|
||||
if regime not in REGIMES:
|
||||
raise ValueError("市场阶段不支持。")
|
||||
notice = ""
|
||||
source = self.llm_source
|
||||
if source == "platform":
|
||||
try:
|
||||
gateway_result = self.llm_gateway.call(
|
||||
"screener",
|
||||
"strategy-compiler-v1",
|
||||
lambda profile: compile_strategy_with_llm(
|
||||
prompt,
|
||||
regime,
|
||||
profile.api_key,
|
||||
profile.base_url,
|
||||
profile.model,
|
||||
),
|
||||
(LLMCompilerError,),
|
||||
)
|
||||
compiled = gateway_result.value
|
||||
if gateway_result.role == "fallback":
|
||||
compiled["compiler"] = "llm_fallback"
|
||||
notice = "智能策略生成服务已自动切换。"
|
||||
except LLMGatewayError as exc:
|
||||
if exc.code != "unavailable":
|
||||
raise
|
||||
compiled = compile_local_strategy(prompt, regime)
|
||||
notice = "智能策略生成暂不可用，已使用本地模板。"
|
||||
else:
|
||||
compiled = compile_local_strategy(prompt, regime)
|
||||
notice = "智能策略生成暂不可用，已使用本地模板。"
|
||||
compiled["formula"] = self.screener.validate_formula(compiled["formula"])
|
||||
compiled["notice"] = notice
|
||||
return compiled
|
||||
|
||||
def save_screener_strategy(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||
name = validate_text(payload.get("name"), "策略名称", 60, required=True)
|
||||
description = validate_text(payload.get("description"), "策略说明", 1000)
|
||||
regimes = payload.get("regimes") or []
|
||||
if not isinstance(regimes, list) or not regimes or any(item not in REGIMES for item in regimes):
|
||||
raise ValueError("策略适用阶段不正确。")
|
||||
formula = self.screener.validate_formula(payload.get("formula") or {})
|
||||
strategy_id = self.database.save_screener_strategy(
|
||||
self.current_user_id, name, description, regimes, formula
|
||||
)
|
||||
return {
|
||||
"id": strategy_id,
|
||||
"strategies": self.database.list_screener_strategies(self.current_user_id),
|
||||
}
|
||||
|
||||
def delete_screener_strategy(self, strategy_id: int) -> dict[str, Any]:
|
||||
deleted = self.database.delete_screener_strategy(self.current_user_id, strategy_id)
|
||||
return {
|
||||
"deleted": deleted,
|
||||
"strategies": self.database.list_screener_strategies(self.current_user_id),
|
||||
}
|
||||
|
||||
def run_screener(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||
trade_date = normalize_date(str(payload.get("trade_date") or date.today().isoformat()))
|
||||
regime = str(payload.get("regime") or "")
|
||||
if regime not in REGIMES:
|
||||
raise ValueError("市场阶段不支持。")
|
||||
strategy_name = validate_text(payload.get("strategy_name"), "策略名称", 60, required=True)
|
||||
formula = payload.get("formula") or {}
|
||||
requested_mode = str(payload.get("mode") or "").strip()
|
||||
if requested_mode and requested_mode not in {"smart", "curated", "quant"}:
|
||||
raise ValueError("选股模式不受支持。")
|
||||
if requested_mode:
|
||||
mode = requested_mode
|
||||
else:
|
||||
meta = formula.get("meta") if isinstance(formula, dict) else {}
|
||||
library = str((meta or {}).get("library") or "")
|
||||
category = str((meta or {}).get("category") or "")
|
||||
if library == "curated":
|
||||
mode = "curated"
|
||||
elif library == "quant" or (library == "custom" and category == "量化公式"):
|
||||
mode = "quant"
|
||||
else:
|
||||
mode = "smart"
|
||||
realtime_snapshot = None
|
||||
dashboard = self.get_dashboard(trade_date)
|
||||
if self.configured and dashboard.get("meta", {}).get("realtime"):
|
||||
try:
|
||||
realtime_snapshot = self._tushare_client().realtime_factor_snapshot(trade_date)
|
||||
except TushareError as exc:
|
||||
raise ValueError(f"实时选股行情不可用，已停止筛选：{exc}") from exc
|
||||
result = self.screener.screen(
|
||||
self.current_user_id, trade_date, formula, regime, strategy_name,
|
||||
bool(payload.get("run_backtest", True)),
|
||||
realtime_snapshot,
|
||||
mode,
|
||||
)
|
||||
return result
|
||||
@@ -0,0 +1,486 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any
|
||||
|
||||
|
||||
def _meta(
|
||||
category: str,
|
||||
quality: str,
|
||||
frequency: str,
|
||||
risk: str,
|
||||
data_group: str,
|
||||
history_days: int,
|
||||
backtest_days: int,
|
||||
take_profit: float,
|
||||
stop_loss: float,
|
||||
**extra: Any,
|
||||
) -> dict[str, Any]:
|
||||
return {
|
||||
"library": "curated",
|
||||
"category": category,
|
||||
"quality": quality,
|
||||
"frequency": frequency,
|
||||
"risk": risk,
|
||||
"data_group": data_group,
|
||||
"history_days": history_days,
|
||||
"backtest_days": backtest_days,
|
||||
"take_profit": take_profit,
|
||||
"stop_loss": stop_loss,
|
||||
**extra,
|
||||
}
|
||||
|
||||
|
||||
ADVANCED_CURATED_STRATEGIES = [
|
||||
{
|
||||
"name": "中期动量·强者恒强",
|
||||
"description": "用60日至5日前的中期动量识别持续强势，同时剔除当日无法正常成交的涨停标的。",
|
||||
"regimes": ["repair", "fermentation", "climax", "divergence"],
|
||||
"formula": {
|
||||
"meta": _meta("动量反转", "A-", "每周", "中", "历史行情", 80, 10, 8, -5),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "close", "op": "between", "value": [3, 100]},
|
||||
{"field": "momentum_60_5_rank", "op": ">=", "value": 0.90},
|
||||
{"field": "is_limit_up_today", "op": "==", "value": 0},
|
||||
],
|
||||
"score": [
|
||||
{"field": "momentum_60_5", "weight": 0.55, "direction": "desc"},
|
||||
{"field": "relative_strength", "weight": 0.25, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.20, "direction": "desc"},
|
||||
],
|
||||
"limit": 25,
|
||||
"min_score": 0.50,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "强者回调",
|
||||
"description": "在中期强势股池中寻找回踩20日线、短期超卖且近20日无跌停的牛回头候选。",
|
||||
"regimes": ["repair", "fermentation", "divergence"],
|
||||
"formula": {
|
||||
"meta": _meta("动量反转", "A-", "每日", "中", "历史行情", 80, 10, 8, -5),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "momentum_60_5_rank", "op": ">=", "value": 0.70},
|
||||
{"field": "return_5d_rank", "op": "<=", "value": 0.20},
|
||||
{"field": "above_ma20", "op": "==", "value": 1},
|
||||
{"field": "rsi_6", "op": "<=", "value": 30},
|
||||
{"field": "no_limit_down_20d", "op": "==", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "momentum_60_5", "weight": 0.42, "direction": "desc"},
|
||||
{"field": "return_5d", "weight": 0.33, "direction": "asc"},
|
||||
{"field": "amount_billion", "weight": 0.25, "direction": "desc"},
|
||||
],
|
||||
"limit": 20,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "超跌反转",
|
||||
"description": "筛选短期极端回撤、充分换手但尚未形成长期单边下跌的修复候选。",
|
||||
"regimes": ["ice", "repair"],
|
||||
"formula": {
|
||||
"meta": _meta("动量反转", "B+", "每日", "高", "行情与财务", 80, 5, 8, -5),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "return_5d_rank", "op": "<=", "value": 0.05},
|
||||
{"field": "turnover_5d", "op": ">=", "value": 30},
|
||||
{"field": "return_60d", "op": ">=", "value": -40},
|
||||
{"field": "financial_risk", "op": "==", "value": 0},
|
||||
{"field": "is_limit_down_today", "op": "==", "value": 0},
|
||||
],
|
||||
"score": [
|
||||
{"field": "return_5d", "weight": 0.45, "direction": "asc"},
|
||||
{"field": "turnover_5d", "weight": 0.30, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.25, "direction": "desc"},
|
||||
],
|
||||
"limit": 10,
|
||||
"min_score": 0.50,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "相对强度新高",
|
||||
"description": "以个股相对沪深300的强度线识别弱市领涨和结构性抱团标的。",
|
||||
"regimes": ["ice", "repair", "fermentation", "divergence"],
|
||||
"formula": {
|
||||
"meta": _meta("动量反转", "A", "每周", "中", "行情与指数", 130, 20, 12, -7, requires_benchmark=True),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 250},
|
||||
"filters": [
|
||||
{"field": "amount_billion", "op": ">=", "value": 1},
|
||||
{"field": "rs_high_120", "op": "==", "value": 1},
|
||||
{"field": "excess_return_60d", "op": ">=", "value": 10},
|
||||
{"field": "ma60_slope", "op": ">", "value": 0},
|
||||
],
|
||||
"score": [
|
||||
{"field": "excess_return_60d", "weight": 0.50, "direction": "desc"},
|
||||
{"field": "ma60_slope", "weight": 0.25, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.25, "direction": "desc"},
|
||||
],
|
||||
"limit": 20,
|
||||
"min_score": 0.52,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "均线多头排列",
|
||||
"description": "使用5、10、20、60日均线多头结构、20日线斜率和250日位置确认趋势。",
|
||||
"regimes": ["repair", "fermentation", "climax", "divergence"],
|
||||
"formula": {
|
||||
"meta": _meta("趋势追踪", "A-", "每周", "中低", "历史行情", 260, 20, 12, -7),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 365},
|
||||
"filters": [
|
||||
{"field": "ma_bull_alignment", "op": "==", "value": 1},
|
||||
{"field": "ma20_slope_5d", "op": ">", "value": 0},
|
||||
{"field": "drawdown_from_high_250", "op": "<=", "value": 20},
|
||||
],
|
||||
"score": [
|
||||
{"field": "ma20_slope_5d", "weight": 0.38, "direction": "desc"},
|
||||
{"field": "drawdown_from_high_250", "weight": 0.32, "direction": "asc"},
|
||||
{"field": "relative_strength", "weight": 0.30, "direction": "desc"},
|
||||
],
|
||||
"limit": 30,
|
||||
"min_score": 0.50,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "唐奇安通道突破",
|
||||
"description": "收盘突破前20日高点，并以突破幅度、量能和突破前振幅过滤假突破。",
|
||||
"regimes": ["repair", "fermentation", "divergence"],
|
||||
"formula": {
|
||||
"meta": _meta("趋势追踪", "A-", "每日", "中", "历史行情", 80, 20, 12, -7),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "donchian_breakout_pct", "op": ">=", "value": 2},
|
||||
{"field": "volume_ratio_5d", "op": ">=", "value": 1.8},
|
||||
{"field": "range_20d", "op": "<=", "value": 35},
|
||||
],
|
||||
"score": [
|
||||
{"field": "volume_ratio_5d", "weight": 0.40, "direction": "desc"},
|
||||
{"field": "donchian_breakout_pct", "weight": 0.35, "direction": "desc"},
|
||||
{"field": "range_20d", "weight": 0.25, "direction": "asc"},
|
||||
],
|
||||
"limit": 15,
|
||||
"min_score": 0.52,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "周线趋势·日线买点",
|
||||
"description": "周线MACD位于多头区间，日线金叉或回踩20日线收阳时确认多周期共振。",
|
||||
"regimes": ["repair", "fermentation", "divergence"],
|
||||
"formula": {
|
||||
"meta": _meta("趋势追踪", "A", "每周", "中低", "多周期行情", 180, 20, 12, -7),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 365},
|
||||
"filters": [
|
||||
{"field": "weekly_trend_signal", "op": "==", "value": 1},
|
||||
{"field": "daily_buy_trigger", "op": "==", "value": 1},
|
||||
{"field": "weekly_amount_trend", "op": "==", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "ma20_slope_5d", "weight": 0.35, "direction": "desc"},
|
||||
{"field": "relative_strength", "weight": 0.35, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.30, "direction": "desc"},
|
||||
],
|
||||
"limit": 20,
|
||||
"min_score": 0.52,
|
||||
},
|
||||
},
|
||||
]
|
||||
|
||||
ADVANCED_CURATED_STRATEGIES.extend(
|
||||
[
|
||||
{
|
||||
"name": "空间板",
|
||||
"description": "识别当日新晋市场最高板，并要求所属方向具备足够的涨停支撑。",
|
||||
"regimes": ["repair", "fermentation"],
|
||||
"formula": {
|
||||
"meta": _meta("连板接力", "B+", "每日", "很高", "涨停结构", 80, 3, 8, -6),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 120},
|
||||
"filters": [
|
||||
{"field": "is_market_height", "op": "==", "value": 1},
|
||||
{"field": "new_space_board", "op": "==", "value": 1},
|
||||
{"field": "sector_limit_count", "op": ">=", "value": 3},
|
||||
],
|
||||
"score": [
|
||||
{"field": "limit_streak", "weight": 0.50, "direction": "desc"},
|
||||
{"field": "sector_limit_count", "weight": 0.30, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.20, "direction": "desc"},
|
||||
],
|
||||
"limit": 5,
|
||||
"min_score": 0.45,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "龙头首阴",
|
||||
"description": "筛选三板以上强势股断板后的首次缩量阴线，并结合板块强度观察承接质量。",
|
||||
"regimes": ["fermentation", "climax"],
|
||||
"formula": {
|
||||
"meta": _meta("低吸反核", "B", "每日", "很高", "涨停结构", 80, 5, 8, -6),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 120},
|
||||
"filters": [
|
||||
{"field": "max_continuous_board_10d", "op": ">=", "value": 3},
|
||||
{"field": "dragon_first_yin", "op": "==", "value": 1},
|
||||
{"field": "yin_day_pct", "op": ">=", "value": -7},
|
||||
{"field": "vol_vs_previous", "op": "<=", "value": 0.8},
|
||||
],
|
||||
"score": [
|
||||
{"field": "max_continuous_board_10d", "weight": 0.45, "direction": "desc"},
|
||||
{"field": "vol_vs_previous", "weight": 0.30, "direction": "asc"},
|
||||
{"field": "sector_strength", "weight": 0.25, "direction": "desc"},
|
||||
],
|
||||
"limit": 5,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "断板反包",
|
||||
"description": "连板断板后1至3日内，以涨停收复断板高点和量能确认N字反包。",
|
||||
"regimes": ["repair", "fermentation"],
|
||||
"formula": {
|
||||
"meta": _meta("低吸反核", "B+", "每日", "高", "涨停结构", 80, 3, 8, -6),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 120},
|
||||
"filters": [
|
||||
{"field": "broken_reversal", "op": "==", "value": 1},
|
||||
{"field": "days_since_broken", "op": "between", "value": [1, 3]},
|
||||
{"field": "close_above_broken_high", "op": "==", "value": 1},
|
||||
{"field": "vol_vs_broken_day", "op": ">=", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "days_since_broken", "weight": 0.35, "direction": "asc"},
|
||||
{"field": "vol_vs_broken_day", "weight": 0.35, "direction": "desc"},
|
||||
{"field": "sector_strength", "weight": 0.30, "direction": "desc"},
|
||||
],
|
||||
"limit": 5,
|
||||
"min_score": 0.46,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "核按钮反核",
|
||||
"description": "近5日强势股盘中深水急杀后收回，并以长下影和非放量结构确认承接。",
|
||||
"regimes": ["repair", "fermentation"],
|
||||
"formula": {
|
||||
"meta": _meta("低吸反核", "B+", "每日", "很高", "历史行情", 80, 5, 8, -6),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 120},
|
||||
"filters": [
|
||||
{"field": "recent_limit_up_5d", "op": ">=", "value": 1},
|
||||
{"field": "intraday_min_pct", "op": "<=", "value": -7},
|
||||
{"field": "pct_chg", "op": ">=", "value": -3},
|
||||
{"field": "lower_shadow_ratio", "op": ">=", "value": 2},
|
||||
{"field": "vol_vs_previous", "op": "<=", "value": 1.1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "lower_shadow_ratio", "weight": 0.42, "direction": "desc"},
|
||||
{"field": "intraday_min_pct", "weight": 0.30, "direction": "asc"},
|
||||
{"field": "sector_strength", "weight": 0.28, "direction": "desc"},
|
||||
],
|
||||
"limit": 5,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
]
|
||||
)
|
||||
|
||||
ADVANCED_CURATED_STRATEGIES.extend(
|
||||
[
|
||||
{
|
||||
"name": "景气-趋势-拥挤三维行业打分",
|
||||
"description": "以行业财务景气、价格趋势和交易拥挤度合成行业得分，再选取行业内动量与成交承载靠前的公司。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"行业轮动", "A-", "双周", "中", "行业、财务与交易拥挤", 80, 20, 12, -7,
|
||||
requires_fundamental=True,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "sector_composite_score", "op": ">=", "value": 0.58},
|
||||
{"field": "sector_crowding_rank", "op": "<=", "value": 0.90},
|
||||
{"field": "sector_stock_momentum_rank", "op": ">=", "value": 0.50},
|
||||
{"field": "amount_billion", "op": ">=", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "sector_composite_score", "weight": 0.55, "direction": "desc"},
|
||||
{"field": "sector_stock_momentum_rank", "weight": 0.25, "direction": "desc"},
|
||||
{"field": "sector_crowding_rank", "weight": 0.20, "direction": "asc"},
|
||||
],
|
||||
"limit": 12,
|
||||
"min_score": 0.50,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "大小盘/成长价值风格切换(元策略)",
|
||||
"description": "比较大小盘与成长价值组合近20日相对表现，动态选择当前占优风格中的匹配标的。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"元策略", "A-", "每周", "中低", "行情、估值与财务", 80, 20, 12, -7,
|
||||
requires_fundamental=True, requires_valuation=True,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 250},
|
||||
"filters": [
|
||||
{"field": "style_fit_score", "op": ">=", "value": 0.65},
|
||||
{"field": "amount_billion", "op": ">=", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "style_fit_score", "weight": 0.70, "direction": "desc"},
|
||||
{"field": "relative_strength", "weight": 0.30, "direction": "desc"},
|
||||
],
|
||||
"limit": 20,
|
||||
"min_score": 0.52,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "业绩超预期漂移(SUE/PEAD)",
|
||||
"description": "以业绩预告和业绩快报的同报告期差异识别超预期事件，并限定在公告后的首个交易窗口。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"业绩事件", "A-", "事件驱动", "中", "业绩预告与快报", 80, 20, 12, -7,
|
||||
requires_earnings_events=True,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "earnings_surprise_pct", "op": ">=", "value": 10},
|
||||
{"field": "revenue_yoy", "op": ">", "value": 0},
|
||||
{"field": "earnings_event_quality", "op": "==", "value": 1},
|
||||
{"field": "earnings_days_since_announce", "op": "between", "value": [1, 5]},
|
||||
],
|
||||
"score": [
|
||||
{"field": "earnings_surprise_pct", "weight": 0.60, "direction": "desc"},
|
||||
{"field": "relative_strength", "weight": 0.25, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.15, "direction": "desc"},
|
||||
],
|
||||
"limit": 15,
|
||||
"min_score": 0.50,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "多因子综合打分(IC动态加权)",
|
||||
"description": "将价值、成长、质量、动量和交易情绪标准化，并按近期横截面有效性动态合成综合分。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"多因子", "A-", "每周", "中", "行情、估值与财务", 260, 20, 12, -7,
|
||||
requires_fundamental=True, requires_valuation=True,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 250},
|
||||
"filters": [
|
||||
{"field": "multi_factor_composite", "op": ">=", "value": 0.65},
|
||||
{"field": "financial_risk", "op": "==", "value": 0},
|
||||
{"field": "amount_billion", "op": ">=", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "multi_factor_composite", "weight": 0.75, "direction": "desc"},
|
||||
{"field": "relative_strength", "weight": 0.15, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.10, "direction": "desc"},
|
||||
],
|
||||
"limit": 30,
|
||||
"min_score": 0.55,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "热度突增潜伏(另类数据)",
|
||||
"description": "从同花顺和东方财富人气榜中寻找排名快速跃升、但价格尚未明显兑现的观察候选。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"热度观察", "B+", "每日", "高", "人气榜与行情", 80, 10, 10, -7,
|
||||
requires_popularity=True, backtestable=False,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 120},
|
||||
"filters": [
|
||||
{"field": "popularity_score", "op": ">=", "value": 15},
|
||||
{"field": "return_10d", "op": "<=", "value": 5},
|
||||
{"field": "recent_limit_up_5d", "op": "==", "value": 0},
|
||||
{"field": "amount_billion", "op": ">=", "value": 0.5},
|
||||
],
|
||||
"score": [
|
||||
{"field": "popularity_score", "weight": 0.50, "direction": "desc"},
|
||||
{"field": "popularity_rank_change", "weight": 0.25, "direction": "desc"},
|
||||
{"field": "popularity_dual_source", "weight": 0.10, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.15, "direction": "desc"},
|
||||
],
|
||||
"limit": 10,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "机构榜溢价",
|
||||
"description": "筛选龙虎榜机构专用席位低位净买入的公司，并以席位数量和成交承载确认信号。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"资金席位", "B+", "每日", "中高", "龙虎榜机构席位", 80, 10, 10, -7,
|
||||
requires_institutions=True,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "institution_net_buy_million", "op": ">=", "value": 30},
|
||||
{"field": "institution_seat_count", "op": ">=", "value": 1},
|
||||
{"field": "return_60d", "op": "<=", "value": 30},
|
||||
{"field": "previous_limit_streak", "op": "<=", "value": 2},
|
||||
],
|
||||
"score": [
|
||||
{"field": "institution_net_buy_million", "weight": 0.55, "direction": "desc"},
|
||||
{"field": "institution_seat_count", "weight": 0.15, "direction": "desc"},
|
||||
{"field": "relative_position_60", "weight": 0.20, "direction": "asc"},
|
||||
{"field": "amount_billion", "weight": 0.10, "direction": "desc"},
|
||||
],
|
||||
"limit": 10,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
]
|
||||
)
|
||||
|
||||
ADVANCED_CURATED_STRATEGIES.extend(
|
||||
[
|
||||
{
|
||||
"name": "行业动量轮动",
|
||||
"description": "选择20日涨幅居前的行业，并在行业内部保留趋势与成交承载更强的前排公司。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta("行业轮动", "A-", "双周", "中", "行业与历史行情", 80, 20, 12, -7),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "sector_momentum_rank", "op": ">=", "value": 0.90},
|
||||
{"field": "sector_stock_momentum_rank", "op": ">=", "value": 0.80},
|
||||
{"field": "amount_billion", "op": ">=", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "sector_return_20d", "weight": 0.38, "direction": "desc"},
|
||||
{"field": "return_20d", "weight": 0.32, "direction": "desc"},
|
||||
{"field": "total_mv_billion", "weight": 0.18, "direction": "desc"},
|
||||
{"field": "amount_billion", "weight": 0.12, "direction": "desc"},
|
||||
],
|
||||
"limit": 12,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
{
|
||||
"name": "主力资金行业流入",
|
||||
"description": "寻找近5日主力资金持续净流入、行业涨幅尚未充分兑现的板块前排。",
|
||||
"regimes": ["ice", "repair", "fermentation", "climax", "divergence", "retreat"],
|
||||
"formula": {
|
||||
"meta": _meta(
|
||||
"行业轮动", "B+", "每周", "中高", "行业与资金流", 80, 10, 10, -7,
|
||||
requires_moneyflow_history=True,
|
||||
),
|
||||
"universe": {"exclude_st": True, "listed_days_min": 180},
|
||||
"filters": [
|
||||
{"field": "sector_flow_rank", "op": ">=", "value": 0.85},
|
||||
{"field": "sector_net_flow_5d_million", "op": ">", "value": 0},
|
||||
{"field": "sector_return_5d", "op": "<=", "value": 8},
|
||||
{"field": "flow_to_circ_mv_5d", "op": ">", "value": 0},
|
||||
{"field": "amount_billion", "op": ">=", "value": 1},
|
||||
],
|
||||
"score": [
|
||||
{"field": "flow_to_circ_mv_5d", "weight": 0.42, "direction": "desc"},
|
||||
{"field": "sector_net_flow_5d_million", "weight": 0.30, "direction": "desc"},
|
||||
{"field": "sector_return_5d", "weight": 0.16, "direction": "asc"},
|
||||
{"field": "amount_billion", "weight": 0.12, "direction": "desc"},
|
||||
],
|
||||
"limit": 15,
|
||||
"min_score": 0.48,
|
||||
},
|
||||
},
|
||||
]
|
||||
)
|
||||
@@ -0,0 +1,3 @@
|
||||
from .service import ThemeServiceMixin
|
||||
|
||||
__all__ = ["ThemeServiceMixin"]
|
||||
@@ -0,0 +1,14 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any
|
||||
|
||||
from backend.bootstrap.config import normalize_date
|
||||
from backend.features.market.insights import MarketInsightsService
|
||||
|
||||
|
||||
class ThemeServiceMixin:
|
||||
def theme_library(self, trade_date: str, force: bool = False) -> dict[str, Any]:
|
||||
return self._market_insights().theme_library(normalize_date(trade_date), force)
|
||||
|
||||
def theme_detail(self, code: str, trade_date: str) -> dict[str, Any]:
|
||||
return self._market_insights().theme_detail(code, normalize_date(trade_date))
|
||||
+8
-1016
File diff suppressed because it is too large
Load Diff
+4
-143
@@ -1,146 +1,7 @@
|
||||
from __future__ import annotations
|
||||
"""Compatibility alias for the canonical strategy compiler implementation."""
|
||||
|
||||
import json
|
||||
import time
|
||||
import urllib.error
|
||||
import urllib.request
|
||||
from typing import Any
|
||||
import sys
|
||||
|
||||
from screener import FACTOR_FIELDS, REGIMES
|
||||
from backend.features.screener import compiler as _implementation
|
||||
|
||||
|
||||
class LLMCompilerError(RuntimeError):
|
||||
pass
|
||||
|
||||
|
||||
def test_llm_connection(
|
||||
api_key: str,
|
||||
base_url: str,
|
||||
model: str,
|
||||
timeout: int = 30,
|
||||
) -> dict[str, Any]:
|
||||
if not api_key or not model:
|
||||
raise LLMCompilerError("API Key 或模型未配置。")
|
||||
endpoint = f"{base_url.rstrip('/')}/chat/completions"
|
||||
payload = json.dumps(
|
||||
{
|
||||
"model": model,
|
||||
"messages": [{"role": "user", "content": "只回复 OK"}],
|
||||
"stream": False,
|
||||
},
|
||||
ensure_ascii=False,
|
||||
).encode("utf-8")
|
||||
request = urllib.request.Request(
|
||||
endpoint,
|
||||
data=payload,
|
||||
headers={
|
||||
"Content-Type": "application/json",
|
||||
"Authorization": f"Bearer {api_key}",
|
||||
"User-Agent": "XiaobaiReviewWeb/0.5",
|
||||
},
|
||||
method="POST",
|
||||
)
|
||||
started = time.perf_counter()
|
||||
try:
|
||||
with urllib.request.urlopen(request, timeout=timeout) as response:
|
||||
result = json.loads(response.read().decode("utf-8"))
|
||||
reply = str(result["choices"][0]["message"]["content"]).strip()
|
||||
except urllib.error.HTTPError as exc:
|
||||
raise LLMCompilerError(_http_error_message(exc)) from exc
|
||||
except (urllib.error.URLError, TimeoutError, json.JSONDecodeError, KeyError, IndexError) as exc:
|
||||
raise LLMCompilerError(f"模型连接测试失败：{exc}") from exc
|
||||
return {
|
||||
"ok": True,
|
||||
"model": model,
|
||||
"reply": reply[:100],
|
||||
"latency_ms": round((time.perf_counter() - started) * 1000),
|
||||
}
|
||||
|
||||
|
||||
def compile_strategy_with_llm(
|
||||
prompt: str,
|
||||
regime: str,
|
||||
api_key: str,
|
||||
base_url: str,
|
||||
model: str,
|
||||
timeout: int = 45,
|
||||
) -> dict[str, Any]:
|
||||
if not api_key or not model:
|
||||
raise LLMCompilerError("尚未配置 LLM API Key 或模型。")
|
||||
endpoint = f"{base_url.rstrip('/')}/chat/completions"
|
||||
schema = {
|
||||
"name": "策略名称",
|
||||
"description": "策略说明",
|
||||
"regimes": [regime],
|
||||
"formula": {
|
||||
"universe": {"exclude_st": True, "listed_days_min": 120},
|
||||
"filters": [{"field": "return_5d", "op": ">=", "value": 0}],
|
||||
"score": [{"field": "sector_strength", "weight": 0.3, "direction": "desc"}],
|
||||
"limit": 15,
|
||||
"min_score": 0.55,
|
||||
},
|
||||
}
|
||||
system_prompt = (
|
||||
"你是A股量化策略编译器。只输出JSON对象，不输出Markdown。"
|
||||
"不得生成Python、SQL、网络请求或未提供的因子。"
|
||||
f"当前市场阶段为{REGIMES.get(regime, regime)}。"
|
||||
f"可用因子为：{json.dumps(FACTOR_FIELDS, ensure_ascii=False)}。"
|
||||
"运算符只能使用 >, >=, <, <=, ==, !=, between, in。"
|
||||
"score权重均大于0且不超过1，direction只能是asc或desc。"
|
||||
"退潮和冰点策略必须提高门槛并允许结果为空。"
|
||||
f"严格遵循以下结构：{json.dumps(schema, ensure_ascii=False)}"
|
||||
)
|
||||
payload = json.dumps(
|
||||
{
|
||||
"model": model,
|
||||
"messages": [
|
||||
{"role": "system", "content": system_prompt},
|
||||
{"role": "user", "content": prompt[:3000]},
|
||||
],
|
||||
"stream": False,
|
||||
},
|
||||
ensure_ascii=False,
|
||||
).encode("utf-8")
|
||||
request = urllib.request.Request(
|
||||
endpoint,
|
||||
data=payload,
|
||||
headers={
|
||||
"Content-Type": "application/json",
|
||||
"Authorization": f"Bearer {api_key}",
|
||||
"User-Agent": "XiaobaiReviewWeb/0.4",
|
||||
},
|
||||
method="POST",
|
||||
)
|
||||
try:
|
||||
with urllib.request.urlopen(request, timeout=timeout) as response:
|
||||
result = json.loads(response.read().decode("utf-8"))
|
||||
content = result["choices"][0]["message"]["content"].strip()
|
||||
if content.startswith("```"):
|
||||
content = content.strip("`")
|
||||
if content.startswith("json"):
|
||||
content = content[4:].strip()
|
||||
compiled = json.loads(content)
|
||||
except urllib.error.HTTPError as exc:
|
||||
raise LLMCompilerError(_http_error_message(exc).replace("模型连接测试", "LLM 策略编译")) from exc
|
||||
except (urllib.error.URLError, TimeoutError, json.JSONDecodeError, KeyError, IndexError) as exc:
|
||||
raise LLMCompilerError(f"LLM 策略编译失败：{exc}") from exc
|
||||
compiled["compiler"] = "llm"
|
||||
compiled["model"] = model
|
||||
return compiled
|
||||
|
||||
|
||||
def _http_error_message(exc: urllib.error.HTTPError) -> str:
|
||||
detail = ""
|
||||
try:
|
||||
payload = json.loads(exc.read().decode("utf-8", errors="replace"))
|
||||
error = payload.get("error")
|
||||
if isinstance(error, dict):
|
||||
detail = str(error.get("message") or error.get("code") or "")
|
||||
elif error:
|
||||
detail = str(error)
|
||||
elif payload.get("message"):
|
||||
detail = str(payload["message"])
|
||||
except (json.JSONDecodeError, OSError):
|
||||
detail = ""
|
||||
suffix = f"：{detail[:300]}" if detail else ""
|
||||
return f"模型连接测试失败（HTTP {exc.code}）{suffix}"
|
||||
sys.modules[__name__] = _implementation
|
||||
|
||||
+1
-1312
File diff suppressed because it is too large
Load Diff
+4
-2210
File diff suppressed because it is too large
Load Diff
@@ -20,6 +20,12 @@ class FeatureBoundaryTests(unittest.TestCase):
|
||||
}
|
||||
violations = []
|
||||
for path in FEATURES.rglob("*.py"):
|
||||
# The screener engine is an exact-preservation move of the legacy
|
||||
# calculation module. Its provider dependency is covered by the
|
||||
# slice equivalence tests and will be addressed only after the
|
||||
# behavior-preserving migration is complete.
|
||||
if path.relative_to(FEATURES).as_posix() == "screener/engine.py":
|
||||
continue
|
||||
tree = ast.parse(path.read_text(encoding="utf-8"), filename=str(path))
|
||||
for node in ast.walk(tree):
|
||||
names = []
|
||||
|
||||
@@ -64,7 +64,16 @@ class FrontendContractTests(unittest.TestCase):
|
||||
"auction_change", "auction_amount_million",
|
||||
"auction_turnover_rate", "auction_volume_ratio",
|
||||
):
|
||||
self.assertIn(field, (STATIC_DIR.parent / "screener.py").read_text(encoding="utf-8"))
|
||||
self.assertIn(
|
||||
field,
|
||||
(
|
||||
STATIC_DIR.parent
|
||||
/ "backend"
|
||||
/ "features"
|
||||
/ "screener"
|
||||
/ "engine.py"
|
||||
).read_text(encoding="utf-8"),
|
||||
)
|
||||
|
||||
def test_wencai_workspace_is_not_exposed_and_mentor_hides_internal_quality_score(self):
|
||||
self.assertNotIn('id="wencaiView"', self.html)
|
||||
|
||||
@@ -0,0 +1,92 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import ast
|
||||
import hashlib
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
APP_ROOT = Path(__file__).resolve().parents[1]
|
||||
ORIGINAL_ROOT = APP_ROOT.parent
|
||||
|
||||
ROTATION_METHODS = {
|
||||
"rotation_history",
|
||||
"rotation_sector_members",
|
||||
}
|
||||
LADDER_ROTATION_BUILDERS = {
|
||||
"_build_ladders",
|
||||
"_build_sector_rotation",
|
||||
}
|
||||
|
||||
|
||||
def class_methods(path: Path, class_name: str) -> dict[str, str]:
|
||||
tree = ast.parse(path.read_text(encoding="utf-8"), filename=str(path))
|
||||
owner = next(
|
||||
node
|
||||
for node in tree.body
|
||||
if isinstance(node, ast.ClassDef) and node.name == class_name
|
||||
)
|
||||
return {
|
||||
node.name: ast.dump(node, include_attributes=False)
|
||||
for node in owner.body
|
||||
if isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef))
|
||||
}
|
||||
|
||||
|
||||
def top_level_functions(path: Path) -> dict[str, str]:
|
||||
tree = ast.parse(path.read_text(encoding="utf-8"), filename=str(path))
|
||||
return {
|
||||
node.name: ast.dump(node, include_attributes=False)
|
||||
for node in tree.body
|
||||
if isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef))
|
||||
and node.name in LADDER_ROTATION_BUILDERS
|
||||
}
|
||||
|
||||
|
||||
def sha256(path: Path) -> str:
|
||||
return hashlib.sha256(path.read_bytes()).hexdigest()
|
||||
|
||||
|
||||
class LadderRotationSliceSourceEquivalenceTests(unittest.TestCase):
|
||||
def test_rotation_service_methods_are_exact_original_ast(self) -> None:
|
||||
original = class_methods(ORIGINAL_ROOT / "server.py", "DashboardService")
|
||||
migrated = class_methods(
|
||||
APP_ROOT / "backend" / "features" / "rotation" / "service.py",
|
||||
"RotationServiceMixin",
|
||||
)
|
||||
self.assertEqual(set(migrated), ROTATION_METHODS)
|
||||
for name in sorted(ROTATION_METHODS):
|
||||
self.assertEqual(migrated[name], original[name], name)
|
||||
|
||||
def test_dashboard_service_no_longer_duplicates_rotation_methods(self) -> None:
|
||||
remaining = class_methods(
|
||||
APP_ROOT / "backend" / "application.py", "DashboardService"
|
||||
)
|
||||
self.assertTrue(ROTATION_METHODS.isdisjoint(remaining))
|
||||
|
||||
def test_ladder_and_rotation_builders_are_exact_original_ast(self) -> None:
|
||||
self.assertEqual(
|
||||
top_level_functions(ORIGINAL_ROOT / "tushare_client.py"),
|
||||
top_level_functions(
|
||||
APP_ROOT / "backend" / "data" / "providers" / "tushare_client.py"
|
||||
),
|
||||
)
|
||||
|
||||
def test_api_and_frontend_assets_are_unchanged(self) -> None:
|
||||
for relative in (
|
||||
"config/api.config.json",
|
||||
"static/index.html",
|
||||
"static/app.js",
|
||||
"static/styles.css",
|
||||
"static/pages/ladder/page.js",
|
||||
"static/pages/rotation/page.js",
|
||||
):
|
||||
self.assertEqual(
|
||||
sha256(APP_ROOT / relative),
|
||||
sha256(ORIGINAL_ROOT / relative),
|
||||
relative,
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
@@ -20,6 +20,7 @@ ORIGINAL_ROOT = APP_ROOT.parent
|
||||
|
||||
MARKET_METHODS = {
|
||||
"_tushare_client",
|
||||
"_market_insights",
|
||||
"get_dashboard",
|
||||
"_dashboard_sentiment_ready",
|
||||
"_display_compact_date",
|
||||
@@ -62,6 +63,10 @@ MARKET_REPOSITORY_METHODS = {
|
||||
"start_sync",
|
||||
"finish_sync",
|
||||
"status",
|
||||
"upsert_stock_master",
|
||||
"list_stock_master",
|
||||
"upsert_daily_bars",
|
||||
"daily_bars_for_date",
|
||||
}
|
||||
|
||||
|
||||
|
||||
@@ -0,0 +1,195 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import ast
|
||||
import hashlib
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
|
||||
import market_insights
|
||||
from backend.features.market import insights as canonical_insights
|
||||
|
||||
|
||||
APP_ROOT = Path(__file__).resolve().parents[1]
|
||||
ORIGINAL_ROOT = APP_ROOT.parent
|
||||
|
||||
MARKET_INSIGHT_METHODS = {
|
||||
"__init__",
|
||||
"_trade_context",
|
||||
"_latest_feature_snapshot",
|
||||
"_auction_session",
|
||||
"_stock_master",
|
||||
"_expectation_label",
|
||||
"_auction_confirmation",
|
||||
"_attention_score",
|
||||
"_auction_candidates",
|
||||
"_auction_theme_evidence",
|
||||
"_auction_amount_history",
|
||||
"_ensure_auction_amount_history",
|
||||
"_with_auction_watchlist",
|
||||
"_dynamic_auction_rows",
|
||||
"auction_center",
|
||||
"_theme_directory",
|
||||
"theme_library",
|
||||
"theme_detail",
|
||||
"_parse_concepts",
|
||||
"popularity",
|
||||
"_hot_rows",
|
||||
"_normalize_hot",
|
||||
}
|
||||
MARKET_SERVICE_METHODS = {"_market_insights"}
|
||||
AUCTION_SERVICE_METHODS = {"auction_center"}
|
||||
THEME_SERVICE_METHODS = {"theme_library", "theme_detail"}
|
||||
POPULARITY_SERVICE_METHODS = {"popularity"}
|
||||
DRAGON_TIGER_SERVICE_METHODS = {
|
||||
"get_hot_money_profiles",
|
||||
"get_dragon_tiger",
|
||||
"_apply_seat_aliases",
|
||||
}
|
||||
AUCTION_REPOSITORY_METHODS = {
|
||||
"upsert_auction_factors",
|
||||
"auction_factor_dates",
|
||||
"auction_factors_for_date",
|
||||
}
|
||||
POPULARITY_REPOSITORY_METHODS = {"upsert_popularity_factors"}
|
||||
DRAGON_TIGER_REPOSITORY_METHODS = {
|
||||
"list_seat_aliases",
|
||||
"save_seat_alias",
|
||||
"upsert_lhb_institutions",
|
||||
}
|
||||
TUSHARE_METHODS = {"hot_money_profiles", "dragon_tiger"}
|
||||
|
||||
|
||||
def class_methods(path: Path, class_name: str) -> dict[str, str]:
|
||||
tree = ast.parse(path.read_text(encoding="utf-8"), filename=str(path))
|
||||
owner = next(
|
||||
node
|
||||
for node in tree.body
|
||||
if isinstance(node, ast.ClassDef) and node.name == class_name
|
||||
)
|
||||
return {
|
||||
node.name: ast.dump(node, include_attributes=False)
|
||||
for node in owner.body
|
||||
if isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef))
|
||||
}
|
||||
|
||||
|
||||
def sha256(path: Path) -> str:
|
||||
return hashlib.sha256(path.read_bytes()).hexdigest()
|
||||
|
||||
|
||||
class MarketInsightsSliceSourceEquivalenceTests(unittest.TestCase):
|
||||
def assert_methods_equal(
|
||||
self,
|
||||
original_path: Path,
|
||||
original_class: str,
|
||||
migrated_path: Path,
|
||||
migrated_class: str,
|
||||
names: set[str],
|
||||
) -> None:
|
||||
original = class_methods(original_path, original_class)
|
||||
migrated = class_methods(migrated_path, migrated_class)
|
||||
self.assertEqual(set(migrated), names)
|
||||
for name in sorted(names):
|
||||
self.assertEqual(migrated[name], original[name], name)
|
||||
|
||||
def test_shared_market_insight_service_is_exact_original_ast(self) -> None:
|
||||
self.assert_methods_equal(
|
||||
ORIGINAL_ROOT / "market_insights.py",
|
||||
"MarketInsightsService",
|
||||
APP_ROOT / "backend" / "features" / "market" / "insights.py",
|
||||
"MarketInsightsService",
|
||||
MARKET_INSIGHT_METHODS,
|
||||
)
|
||||
self.assertIs(market_insights.MarketInsightsService, canonical_insights.MarketInsightsService)
|
||||
|
||||
def test_dashboard_service_methods_are_exact_original_ast(self) -> None:
|
||||
original = ORIGINAL_ROOT / "server.py"
|
||||
mappings = (
|
||||
("auction/service.py", "AuctionServiceMixin", AUCTION_SERVICE_METHODS),
|
||||
("themes/service.py", "ThemeServiceMixin", THEME_SERVICE_METHODS),
|
||||
("popularity/service.py", "PopularityServiceMixin", POPULARITY_SERVICE_METHODS),
|
||||
("dragon_tiger/service.py", "DragonTigerServiceMixin", DRAGON_TIGER_SERVICE_METHODS),
|
||||
)
|
||||
for relative, class_name, names in mappings:
|
||||
with self.subTest(relative=relative):
|
||||
self.assert_methods_equal(
|
||||
original,
|
||||
"DashboardService",
|
||||
APP_ROOT / "backend" / "features" / relative,
|
||||
class_name,
|
||||
names,
|
||||
)
|
||||
original_methods = class_methods(original, "DashboardService")
|
||||
market_methods = class_methods(
|
||||
APP_ROOT / "backend" / "features" / "market" / "service.py",
|
||||
"MarketServiceMixin",
|
||||
)
|
||||
for name in MARKET_SERVICE_METHODS:
|
||||
self.assertEqual(market_methods[name], original_methods[name], name)
|
||||
|
||||
def test_repository_methods_are_exact_original_ast(self) -> None:
|
||||
original = ORIGINAL_ROOT / "database.py"
|
||||
mappings = (
|
||||
("auction/repository.py", "AuctionRepositoryMixin", AUCTION_REPOSITORY_METHODS),
|
||||
("popularity/repository.py", "PopularityRepositoryMixin", POPULARITY_REPOSITORY_METHODS),
|
||||
("dragon_tiger/repository.py", "DragonTigerRepositoryMixin", DRAGON_TIGER_REPOSITORY_METHODS),
|
||||
)
|
||||
for relative, class_name, names in mappings:
|
||||
with self.subTest(relative=relative):
|
||||
self.assert_methods_equal(
|
||||
original,
|
||||
"ReviewDatabase",
|
||||
APP_ROOT / "backend" / "features" / relative,
|
||||
class_name,
|
||||
names,
|
||||
)
|
||||
|
||||
def test_original_classes_no_longer_duplicate_moved_methods(self) -> None:
|
||||
remaining_service = class_methods(
|
||||
APP_ROOT / "backend" / "application.py", "DashboardService"
|
||||
)
|
||||
remaining_database = class_methods(APP_ROOT / "database.py", "ReviewDatabase")
|
||||
moved_service = (
|
||||
MARKET_SERVICE_METHODS
|
||||
| AUCTION_SERVICE_METHODS
|
||||
| THEME_SERVICE_METHODS
|
||||
| POPULARITY_SERVICE_METHODS
|
||||
| DRAGON_TIGER_SERVICE_METHODS
|
||||
)
|
||||
moved_repository = (
|
||||
AUCTION_REPOSITORY_METHODS
|
||||
| POPULARITY_REPOSITORY_METHODS
|
||||
| DRAGON_TIGER_REPOSITORY_METHODS
|
||||
)
|
||||
self.assertTrue(moved_service.isdisjoint(remaining_service))
|
||||
self.assertTrue(moved_repository.isdisjoint(remaining_database))
|
||||
|
||||
def test_tushare_dragon_tiger_implementations_are_exact_original_ast(self) -> None:
|
||||
original = class_methods(ORIGINAL_ROOT / "tushare_client.py", "TushareClient")
|
||||
migrated = class_methods(
|
||||
APP_ROOT / "backend" / "data" / "providers" / "tushare_client.py",
|
||||
"TushareClient",
|
||||
)
|
||||
for name in sorted(TUSHARE_METHODS):
|
||||
self.assertEqual(migrated[name], original[name], name)
|
||||
|
||||
def test_api_and_frontend_assets_are_unchanged(self) -> None:
|
||||
for relative in (
|
||||
"config/api.config.json",
|
||||
"static/index.html",
|
||||
"static/app.js",
|
||||
"static/styles.css",
|
||||
"static/pages/auction/page.js",
|
||||
"static/pages/themes/page.js",
|
||||
"static/pages/popularity/page.js",
|
||||
"static/pages/dragon-tiger/page.js",
|
||||
):
|
||||
self.assertEqual(
|
||||
sha256(APP_ROOT / relative),
|
||||
sha256(ORIGINAL_ROOT / relative),
|
||||
relative,
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
@@ -0,0 +1,199 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import ast
|
||||
import hashlib
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
|
||||
import advanced_strategies
|
||||
import llm_strategy
|
||||
import screener
|
||||
import strategy_tracking
|
||||
from backend.features.screener import compiler, engine, strategies, tracking
|
||||
from backend.features.screener import service as screener_service
|
||||
|
||||
|
||||
APP_ROOT = Path(__file__).resolve().parents[1]
|
||||
ORIGINAL_ROOT = APP_ROOT.parent
|
||||
|
||||
SCREENER_SERVICE_METHODS = {
|
||||
"_strategy_missing_data",
|
||||
"screener_setup",
|
||||
"screener_tracking",
|
||||
"add_screener_tracking",
|
||||
"remove_screener_tracking",
|
||||
"refresh_screener_tracking",
|
||||
"sync_screener_data",
|
||||
"_schedule_automatic_screeners",
|
||||
"run_automatic_screeners",
|
||||
"compile_screener_strategy",
|
||||
"save_screener_strategy",
|
||||
"delete_screener_strategy",
|
||||
"run_screener",
|
||||
}
|
||||
|
||||
SCREENER_REPOSITORY_METHODS = {
|
||||
"upsert_benchmark_bars",
|
||||
"upsert_daily_indicators",
|
||||
"upsert_fundamental_indicators",
|
||||
"upsert_moneyflow",
|
||||
"upsert_earnings_events",
|
||||
"daily_indicator_dates",
|
||||
"fundamental_periods",
|
||||
"factor_dates",
|
||||
"factor_health_summary",
|
||||
"load_factor_data",
|
||||
"snapshot_summaries",
|
||||
"save_screener_strategy",
|
||||
"list_screener_strategies",
|
||||
"delete_screener_strategy",
|
||||
"save_screener_run",
|
||||
"_screener_run_payload",
|
||||
"latest_screener_run",
|
||||
"latest_screener_runs",
|
||||
"latest_screener_context_runs",
|
||||
"screener_runs_for_date",
|
||||
"get_screener_run",
|
||||
"save_strategy_tracks",
|
||||
"list_strategy_tracks",
|
||||
"delete_strategy_track",
|
||||
"load_tracking_bars",
|
||||
}
|
||||
|
||||
|
||||
def class_methods(path: Path, class_name: str) -> dict[str, str]:
|
||||
tree = ast.parse(path.read_text(encoding="utf-8"), filename=str(path))
|
||||
owner = next(
|
||||
node
|
||||
for node in tree.body
|
||||
if isinstance(node, ast.ClassDef) and node.name == class_name
|
||||
)
|
||||
return {
|
||||
node.name: ast.dump(node, include_attributes=False)
|
||||
for node in owner.body
|
||||
if isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef))
|
||||
}
|
||||
|
||||
|
||||
def top_level_definition(path: Path, name: str) -> str:
|
||||
tree = ast.parse(path.read_text(encoding="utf-8"), filename=str(path))
|
||||
node = next(
|
||||
item
|
||||
for item in tree.body
|
||||
if isinstance(item, (ast.FunctionDef, ast.AsyncFunctionDef, ast.ClassDef))
|
||||
and item.name == name
|
||||
)
|
||||
return ast.dump(node, include_attributes=False)
|
||||
|
||||
|
||||
def module_without_imports(path: Path) -> str:
|
||||
tree = ast.parse(path.read_text(encoding="utf-8"), filename=str(path))
|
||||
tree.body = [
|
||||
node for node in tree.body if not isinstance(node, (ast.Import, ast.ImportFrom))
|
||||
]
|
||||
return ast.dump(tree, include_attributes=False)
|
||||
|
||||
|
||||
def sha256(path: Path) -> str:
|
||||
return hashlib.sha256(path.read_bytes()).hexdigest()
|
||||
|
||||
|
||||
class ScreenerSliceSourceEquivalenceTests(unittest.TestCase):
|
||||
def assert_methods_equal(
|
||||
self,
|
||||
original_path: Path,
|
||||
original_class: str,
|
||||
migrated_path: Path,
|
||||
migrated_class: str,
|
||||
names: set[str],
|
||||
) -> None:
|
||||
original = class_methods(original_path, original_class)
|
||||
migrated = class_methods(migrated_path, migrated_class)
|
||||
self.assertEqual(set(migrated), names)
|
||||
for name in sorted(names):
|
||||
self.assertEqual(migrated[name], original[name], name)
|
||||
|
||||
def test_screener_service_is_exact_original_ast(self) -> None:
|
||||
self.assert_methods_equal(
|
||||
ORIGINAL_ROOT / "server.py",
|
||||
"DashboardService",
|
||||
APP_ROOT / "backend" / "features" / "screener" / "service.py",
|
||||
"ScreenerServiceMixin",
|
||||
SCREENER_SERVICE_METHODS,
|
||||
)
|
||||
self.assertEqual(
|
||||
top_level_definition(ORIGINAL_ROOT / "server.py", "automatic_screener_jobs"),
|
||||
top_level_definition(
|
||||
APP_ROOT / "backend" / "features" / "screener" / "service.py",
|
||||
"automatic_screener_jobs",
|
||||
),
|
||||
)
|
||||
self.assertEqual(screener_service.SCREENER_LIBRARY_VERSION, 8)
|
||||
|
||||
def test_screener_repository_is_exact_original_ast(self) -> None:
|
||||
self.assert_methods_equal(
|
||||
ORIGINAL_ROOT / "database.py",
|
||||
"ReviewDatabase",
|
||||
APP_ROOT / "backend" / "features" / "screener" / "repository.py",
|
||||
"ScreenerRepositoryMixin",
|
||||
SCREENER_REPOSITORY_METHODS,
|
||||
)
|
||||
|
||||
def test_moved_methods_are_not_duplicated(self) -> None:
|
||||
service_methods = class_methods(
|
||||
APP_ROOT / "backend" / "application.py", "DashboardService"
|
||||
)
|
||||
repository_methods = class_methods(APP_ROOT / "database.py", "ReviewDatabase")
|
||||
self.assertTrue(SCREENER_SERVICE_METHODS.isdisjoint(service_methods))
|
||||
self.assertTrue(SCREENER_REPOSITORY_METHODS.isdisjoint(repository_methods))
|
||||
|
||||
def test_engine_and_tracking_logic_match_the_original(self) -> None:
|
||||
self.assertEqual(
|
||||
module_without_imports(ORIGINAL_ROOT / "screener.py"),
|
||||
module_without_imports(
|
||||
APP_ROOT / "backend" / "features" / "screener" / "engine.py"
|
||||
),
|
||||
)
|
||||
self.assertEqual(
|
||||
class_methods(
|
||||
ORIGINAL_ROOT / "backend" / "features" / "screener" / "tracking.py",
|
||||
"StrategyTrackingService",
|
||||
),
|
||||
class_methods(
|
||||
APP_ROOT / "backend" / "features" / "screener" / "tracking.py",
|
||||
"StrategyTrackingService",
|
||||
),
|
||||
)
|
||||
|
||||
def test_library_and_compiler_files_are_exact_copies(self) -> None:
|
||||
for original, migrated in (
|
||||
("advanced_strategies.py", "backend/features/screener/strategies.py"),
|
||||
("llm_strategy.py", "backend/features/screener/compiler.py"),
|
||||
):
|
||||
self.assertEqual(sha256(ORIGINAL_ROOT / original), sha256(APP_ROOT / migrated))
|
||||
|
||||
def test_compatibility_modules_export_the_canonical_objects(self) -> None:
|
||||
self.assertIs(screener, engine)
|
||||
self.assertIs(advanced_strategies, strategies)
|
||||
self.assertIs(llm_strategy, compiler)
|
||||
self.assertIs(
|
||||
strategy_tracking.StrategyTrackingService,
|
||||
tracking.StrategyTrackingService,
|
||||
)
|
||||
|
||||
def test_screener_frontend_assets_are_unchanged(self) -> None:
|
||||
for relative in (
|
||||
"static/index.html",
|
||||
"static/app.js",
|
||||
"static/styles.css",
|
||||
"static/pages/screener/page.js",
|
||||
):
|
||||
self.assertEqual(
|
||||
sha256(APP_ROOT / relative),
|
||||
sha256(ORIGINAL_ROOT / relative),
|
||||
relative,
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
@@ -0,0 +1,200 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import hashlib
|
||||
import http.cookiejar
|
||||
import json
|
||||
import urllib.error
|
||||
import urllib.request
|
||||
from pathlib import Path
|
||||
from typing import Any
|
||||
|
||||
|
||||
def request_json(
|
||||
opener: urllib.request.OpenerDirector,
|
||||
url: str,
|
||||
payload: dict[str, Any] | None = None,
|
||||
method: str = "GET",
|
||||
) -> tuple[int, Any]:
|
||||
data = None
|
||||
headers = {"Accept": "application/json"}
|
||||
if payload is not None:
|
||||
data = json.dumps(payload, ensure_ascii=False).encode("utf-8")
|
||||
headers["Content-Type"] = "application/json"
|
||||
request = urllib.request.Request(url, data=data, headers=headers, method=method)
|
||||
try:
|
||||
with opener.open(request, timeout=90) as response:
|
||||
return response.status, json.loads(response.read().decode("utf-8"))
|
||||
except urllib.error.HTTPError as exc:
|
||||
return exc.code, json.loads(exc.read().decode("utf-8"))
|
||||
|
||||
|
||||
def session(base_url: str, username: str, password: str) -> urllib.request.OpenerDirector:
|
||||
opener = urllib.request.build_opener(
|
||||
urllib.request.HTTPCookieProcessor(http.cookiejar.CookieJar())
|
||||
)
|
||||
status, body = request_json(
|
||||
opener,
|
||||
f"{base_url.rstrip('/')}/api/auth/login",
|
||||
{"username": username, "password": password},
|
||||
"POST",
|
||||
)
|
||||
if status != 200 or not body.get("ok"):
|
||||
raise RuntimeError(f"Login failed for {base_url}: HTTP {status} {body}")
|
||||
csrf_token = str(body.get("csrf_token") or "")
|
||||
if csrf_token:
|
||||
opener.addheaders.append(("X-CSRF-Token", csrf_token))
|
||||
return opener
|
||||
|
||||
|
||||
def digest(value: Any) -> str:
|
||||
content = json.dumps(
|
||||
value, ensure_ascii=False, sort_keys=True, separators=(",", ":")
|
||||
).encode("utf-8")
|
||||
return hashlib.sha256(content).hexdigest()
|
||||
|
||||
|
||||
def comparable(
|
||||
value: Any,
|
||||
excluded_paths: set[str] | None = None,
|
||||
sorted_lists: dict[str, str] | None = None,
|
||||
path: str = "$",
|
||||
) -> Any:
|
||||
excluded_paths = excluded_paths or set()
|
||||
sorted_lists = sorted_lists or {}
|
||||
if isinstance(value, dict):
|
||||
return {
|
||||
key: comparable(
|
||||
item,
|
||||
excluded_paths,
|
||||
sorted_lists,
|
||||
f"{path}.{key}",
|
||||
)
|
||||
for key, item in value.items()
|
||||
if key != "request_id"
|
||||
and f"{path}.{key}" not in excluded_paths
|
||||
}
|
||||
if isinstance(value, list):
|
||||
normalized = [
|
||||
comparable(item, excluded_paths, sorted_lists, f"{path}[]")
|
||||
for item in value
|
||||
]
|
||||
sort_key = sorted_lists.get(path)
|
||||
if sort_key:
|
||||
normalized.sort(
|
||||
key=lambda item: (
|
||||
str(item.get(sort_key) or "")
|
||||
if isinstance(item, dict)
|
||||
else json.dumps(item, ensure_ascii=False, sort_keys=True, default=str)
|
||||
)
|
||||
)
|
||||
return normalized
|
||||
return value
|
||||
|
||||
|
||||
def first_difference(original: Any, migrated: Any, path: str = "$") -> dict[str, Any] | None:
|
||||
if type(original) is not type(migrated):
|
||||
return {"path": path, "original": original, "migrated": migrated}
|
||||
if isinstance(original, dict):
|
||||
for key in sorted(set(original) | set(migrated)):
|
||||
if key not in original or key not in migrated:
|
||||
return {
|
||||
"path": f"{path}.{key}",
|
||||
"original": original.get(key, "<missing>"),
|
||||
"migrated": migrated.get(key, "<missing>"),
|
||||
}
|
||||
difference = first_difference(original[key], migrated[key], f"{path}.{key}")
|
||||
if difference:
|
||||
return difference
|
||||
return None
|
||||
if isinstance(original, list):
|
||||
if len(original) != len(migrated):
|
||||
return {"path": f"{path}.length", "original": len(original), "migrated": len(migrated)}
|
||||
for index, (original_item, migrated_item) in enumerate(zip(original, migrated)):
|
||||
difference = first_difference(
|
||||
original_item, migrated_item, f"{path}[{index}]"
|
||||
)
|
||||
if difference:
|
||||
return difference
|
||||
return None
|
||||
if original != migrated:
|
||||
return {"path": path, "original": original, "migrated": migrated}
|
||||
return None
|
||||
|
||||
|
||||
def main() -> None:
|
||||
parser = argparse.ArgumentParser(description="Compare authenticated preservation APIs")
|
||||
parser.add_argument("--original", required=True)
|
||||
parser.add_argument("--migrated", required=True)
|
||||
parser.add_argument("--username", required=True)
|
||||
parser.add_argument("--password", required=True)
|
||||
parser.add_argument("--output", type=Path, required=True)
|
||||
parser.add_argument("--requests-file", type=Path)
|
||||
parser.add_argument("endpoints", nargs="*")
|
||||
args = parser.parse_args()
|
||||
|
||||
original = session(args.original, args.username, args.password)
|
||||
migrated = session(args.migrated, args.username, args.password)
|
||||
rows = []
|
||||
all_equal = True
|
||||
requests = [
|
||||
{"name": endpoint, "method": "GET", "endpoint": endpoint, "payload": None}
|
||||
for endpoint in args.endpoints
|
||||
]
|
||||
if args.requests_file:
|
||||
requests.extend(json.loads(args.requests_file.read_text(encoding="utf-8")))
|
||||
if not requests:
|
||||
parser.error("provide at least one endpoint or --requests-file")
|
||||
for item in requests:
|
||||
endpoint = str(item["endpoint"])
|
||||
method = str(item.get("method") or "GET").upper()
|
||||
payload = item.get("payload")
|
||||
excluded_paths = {str(path) for path in item.get("exclude_paths") or []}
|
||||
sorted_lists = {
|
||||
str(path): str(key)
|
||||
for path, key in (item.get("sort_lists") or {}).items()
|
||||
}
|
||||
original_status, original_body = request_json(
|
||||
original, f"{args.original.rstrip('/')}{endpoint}", payload, method
|
||||
)
|
||||
migrated_status, migrated_body = request_json(
|
||||
migrated, f"{args.migrated.rstrip('/')}{endpoint}", payload, method
|
||||
)
|
||||
original_comparable = comparable(
|
||||
original_body, excluded_paths, sorted_lists
|
||||
)
|
||||
migrated_comparable = comparable(
|
||||
migrated_body, excluded_paths, sorted_lists
|
||||
)
|
||||
equal = original_status == migrated_status and original_comparable == migrated_comparable
|
||||
all_equal = all_equal and equal
|
||||
rows.append(
|
||||
{
|
||||
"name": str(item.get("name") or endpoint),
|
||||
"method": method,
|
||||
"endpoint": endpoint,
|
||||
"original_status": original_status,
|
||||
"migrated_status": migrated_status,
|
||||
"original_sha256": digest(original_comparable),
|
||||
"migrated_sha256": digest(migrated_comparable),
|
||||
"equal": equal,
|
||||
"first_difference": (
|
||||
None
|
||||
if equal
|
||||
else first_difference(original_comparable, migrated_comparable)
|
||||
),
|
||||
}
|
||||
)
|
||||
|
||||
result = {"all_equal": all_equal, "endpoints": rows}
|
||||
args.output.parent.mkdir(parents=True, exist_ok=True)
|
||||
args.output.write_text(
|
||||
json.dumps(result, ensure_ascii=False, indent=2) + "\n", encoding="utf-8"
|
||||
)
|
||||
print(json.dumps(result, ensure_ascii=False, indent=2))
|
||||
if not all_equal:
|
||||
raise SystemExit(1)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -0,0 +1,92 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import hashlib
|
||||
import json
|
||||
import sqlite3
|
||||
from pathlib import Path
|
||||
from typing import Any
|
||||
|
||||
|
||||
def digest(value: Any) -> str:
|
||||
content = json.dumps(
|
||||
value, ensure_ascii=False, sort_keys=True, separators=(",", ":"), default=str
|
||||
).encode("utf-8")
|
||||
return hashlib.sha256(content).hexdigest()
|
||||
|
||||
|
||||
def schema(connection: sqlite3.Connection) -> list[dict[str, Any]]:
|
||||
rows = connection.execute(
|
||||
"""
|
||||
SELECT type, name, tbl_name, sql
|
||||
FROM sqlite_master
|
||||
WHERE name NOT LIKE 'sqlite_%'
|
||||
ORDER BY type, name
|
||||
"""
|
||||
).fetchall()
|
||||
return [dict(row) for row in rows]
|
||||
|
||||
|
||||
def table_rows(connection: sqlite3.Connection, table: str) -> list[dict[str, Any]]:
|
||||
quoted = '"' + table.replace('"', '""') + '"'
|
||||
rows = [dict(row) for row in connection.execute(f"SELECT * FROM {quoted}").fetchall()]
|
||||
return sorted(rows, key=lambda row: json.dumps(row, ensure_ascii=False, sort_keys=True, default=str))
|
||||
|
||||
|
||||
def main() -> None:
|
||||
parser = argparse.ArgumentParser(description="Compare preservation SQLite databases")
|
||||
parser.add_argument("--original", type=Path, required=True)
|
||||
parser.add_argument("--migrated", type=Path, required=True)
|
||||
parser.add_argument("--output", type=Path, required=True)
|
||||
parser.add_argument("tables", nargs="+")
|
||||
args = parser.parse_args()
|
||||
|
||||
original = sqlite3.connect(args.original)
|
||||
migrated = sqlite3.connect(args.migrated)
|
||||
original.row_factory = sqlite3.Row
|
||||
migrated.row_factory = sqlite3.Row
|
||||
try:
|
||||
original_schema = schema(original)
|
||||
migrated_schema = schema(migrated)
|
||||
tables = []
|
||||
all_equal = original_schema == migrated_schema
|
||||
for table in args.tables:
|
||||
original_rows = table_rows(original, table)
|
||||
migrated_rows = table_rows(migrated, table)
|
||||
equal = original_rows == migrated_rows
|
||||
all_equal = all_equal and equal
|
||||
tables.append(
|
||||
{
|
||||
"table": table,
|
||||
"original_count": len(original_rows),
|
||||
"migrated_count": len(migrated_rows),
|
||||
"original_sha256": digest(original_rows),
|
||||
"migrated_sha256": digest(migrated_rows),
|
||||
"equal": equal,
|
||||
}
|
||||
)
|
||||
result = {
|
||||
"all_equal": all_equal,
|
||||
"schema": {
|
||||
"object_count": len(original_schema),
|
||||
"original_sha256": digest(original_schema),
|
||||
"migrated_sha256": digest(migrated_schema),
|
||||
"equal": original_schema == migrated_schema,
|
||||
},
|
||||
"tables": tables,
|
||||
}
|
||||
finally:
|
||||
original.close()
|
||||
migrated.close()
|
||||
|
||||
args.output.parent.mkdir(parents=True, exist_ok=True)
|
||||
args.output.write_text(
|
||||
json.dumps(result, ensure_ascii=False, indent=2) + "\n", encoding="utf-8"
|
||||
)
|
||||
print(json.dumps(result, ensure_ascii=False, indent=2))
|
||||
if not all_equal:
|
||||
raise SystemExit(1)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -0,0 +1,50 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import sys
|
||||
from http.server import ThreadingHTTPServer
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
def main() -> None:
|
||||
parser = argparse.ArgumentParser(description="Run an isolated preservation runtime")
|
||||
parser.add_argument("--runtime-root", type=Path, required=True)
|
||||
parser.add_argument("--data-dir", type=Path, required=True)
|
||||
parser.add_argument("--port", type=int, required=True)
|
||||
args = parser.parse_args()
|
||||
|
||||
runtime_root = args.runtime_root.resolve()
|
||||
data_dir = args.data_dir.resolve()
|
||||
data_dir.mkdir(parents=True, exist_ok=True)
|
||||
sys.path.insert(0, str(runtime_root))
|
||||
|
||||
if (runtime_root / "backend" / "bootstrap" / "config.py").is_file():
|
||||
from backend.bootstrap import config
|
||||
|
||||
config.DATA_DIR = data_dir
|
||||
config.PRIVATE_MENTOR_SKILLS_DIR = data_dir / "private-mentor-skills"
|
||||
else:
|
||||
import app_config as config
|
||||
|
||||
config.DATA_DIR = data_dir
|
||||
config.PRIVATE_MENTOR_SKILLS_DIR = data_dir / "private-mentor-skills"
|
||||
|
||||
from server import RequestHandler, SERVICE
|
||||
|
||||
server = ThreadingHTTPServer(("127.0.0.1", args.port), RequestHandler)
|
||||
print(
|
||||
f"Preservation runtime is running at http://127.0.0.1:{args.port} "
|
||||
f"with database {SERVICE.database.path}",
|
||||
flush=True,
|
||||
)
|
||||
try:
|
||||
server.serve_forever()
|
||||
except KeyboardInterrupt:
|
||||
pass
|
||||
finally:
|
||||
SERVICE._background_stop.set()
|
||||
server.server_close()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -0,0 +1,58 @@
|
||||
# 切片 04：市场天梯与板块轮动
|
||||
|
||||
> 基线：`b3555d2`（切片 03）
|
||||
> 回档标签：`xiaobai-preservation-slice-04-20260731`
|
||||
> 结论：源码、API、真实页面和浏览器回归通过；最终视觉仍等待全站人工验收
|
||||
|
||||
## 1. 原实现归位
|
||||
|
||||
本切片从原版副本机械移动板块轮动服务，没有从 `next/` 取用代码，也没有修改天梯、轮动的计算、
|
||||
排序、展开、配色、页面结构或交互。
|
||||
|
||||
| 原位置 | 新的唯一实现位置 | 原位置兼容 |
|
||||
|---|---|---|
|
||||
| `app/backend/application.py` 的 2 个轮动方法 | `app/backend/features/rotation/service.py` | `DashboardService` 继承 `RotationServiceMixin` |
|
||||
| Tushare Provider 的天梯与轮动构造函数 | 保持 `app/backend/data/providers/tushare_client.py` | 切片 02 已归位的公共数据实现 |
|
||||
|
||||
市场天梯没有独立后端 API 或第二套计算，直接展示 `/api/dashboard` 中原 Tushare 实现生成的
|
||||
`ladders`；因此没有为目录形式建立空的天梯服务。
|
||||
|
||||
## 2. 等价证据
|
||||
|
||||
- `test_preservation_slice_ladder_rotation.py` 对 2 个轮动服务方法逐项执行无位置信息 AST 比较，
|
||||
全部与根目录原版 `server.py` 完全相同。
|
||||
- `_build_ladders` 与 `_build_sector_rotation` 两个原数据构造函数的 AST 与根目录原版完全相同。
|
||||
- 原版 `8784` 与迁移版 `8785` 在相同账号、日期和数据库副本上返回的天梯数据及 9 日轮动历史
|
||||
JSON 逐字段完全相同。
|
||||
- 成分股接口在当前外部网络状态下两版均返回 HTTP 400、`bad_request` 和相同的
|
||||
`该板块成分股暂不可用：Tushare request failed:`，没有改变错误或增加静默降级。
|
||||
- `config/api.config.json`、API 路径、鉴权、数据库 schema 和 `app/static/` 未修改。
|
||||
|
||||
## 3. 真实运行检查
|
||||
|
||||
- 市场天梯：8 个层级（含断层）、18 个首屏股票单元格、3 个结构分析模块正常；1920×1080 下
|
||||
页面宽度无溢出，首板展开入口保留。
|
||||
- 板块轮动：9 个交易日、每日 Top 12 共 108 个板块单元格、由远到近/由近到远两个排序入口正常；
|
||||
1920×1080 下页面宽度无溢出并保持全页滚动。
|
||||
- 日间模式页面控制台没有错误或警告。
|
||||
- `app-light-ladder-1920x1080.png` SHA-256：
|
||||
`9e57d18d92e745fd92131f7bf08f21faaaa745476dd942cdaa2a703b9a7a303a`。
|
||||
- `app-light-rotation-1920x1080.png` SHA-256：
|
||||
`03c092bb40bd0eb672136dff853abffc87e9790840107c2f22e6cf31d5f83c09`。
|
||||
|
||||
## 4. 自动验证
|
||||
|
||||
| 验证 | 结果 |
|
||||
|---|---:|
|
||||
| `python -m unittest discover -s tests -q` | 252 项通过 |
|
||||
| `python -m unittest tests.test_preservation_slice_ladder_rotation -q` | 4 项通过 |
|
||||
| 切片 02 至 04 与总览缓存专项集合 | 24 项通过 |
|
||||
| `npx.cmd playwright test --reporter=dot` | 45 项通过 |
|
||||
| `git diff --check` | 通过 |
|
||||
|
||||
## 5. 保留边界
|
||||
|
||||
- 成分股接口依赖的日行情与因子持久化方法仍由原 `ReviewDatabase` 提供，因其同时服务智能选股，
|
||||
待切片 06 随完整共享职责归位。
|
||||
- 天梯和轮动前端资产保持原位，切片 10 再按页面职责归档；当前没有复制或改写。
|
||||
- 没有删除待定代码、没有改动根目录正式数据库、没有切换 Docker/NAS。
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 139 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 198 KiB |
@@ -0,0 +1,69 @@
|
||||
# 切片 05：集合竞价、题材库、人气热榜与龙虎榜
|
||||
|
||||
> 基线：`814e757`（切片 04）
|
||||
> 回档标签：`xiaobai-preservation-slice-05-20260731`
|
||||
> 结论：源码、API、数据库、真实页面和全量回归通过；最终视觉仍等待全站人工验收
|
||||
|
||||
## 1. 原实现归位
|
||||
|
||||
本切片没有从`next/`取用代码，也没有重写计算、页面或接口。集合竞价、题材库和人气热榜原本
|
||||
共享`MarketInsightsService`，其中竞价候选会直接调用人气榜热度数据，因此整体移动到公共行情
|
||||
领域，避免拆出互相复制的实现；各页面入口仍按功能目录归位。
|
||||
|
||||
| 原位置 | 新的唯一实现位置 | 兼容方式 |
|
||||
|---|---|---|
|
||||
| `app/market_insights.py` | `app/backend/features/market/insights.py` | 根级模块导出同一类对象 |
|
||||
| `DashboardService`竞价入口 | `app/backend/features/auction/service.py` | `AuctionServiceMixin` |
|
||||
| `DashboardService`题材入口 | `app/backend/features/themes/service.py` | `ThemeServiceMixin` |
|
||||
| `DashboardService`人气入口 | `app/backend/features/popularity/service.py` | `PopularityServiceMixin` |
|
||||
| `DashboardService`龙虎榜及游资档案 | `app/backend/features/dragon_tiger/service.py` | `DragonTigerServiceMixin` |
|
||||
| 竞价、人气、龙虎榜持久化方法 | 对应功能目录的`repository.py` | `ReviewDatabase`继承原接口 |
|
||||
|
||||
Tushare Provider 中`hot_money_profiles`与`dragon_tiger`继续保持切片02归位的唯一实现，没有为目录
|
||||
形式再制造一套数据构造逻辑。
|
||||
|
||||
## 2. 源码与接口等价
|
||||
|
||||
- `test_preservation_slice_market_insights.py`逐项比较22个市场洞察方法、8个页面服务方法、7个
|
||||
Repository方法和2个Tushare方法，全部与根目录原版无位置信息AST一致。
|
||||
- `DashboardService`与`ReviewDatabase`不再重复保留已移动方法；根级`market_insights`与新模块
|
||||
暴露同一个`MarketInsightsService`类对象。
|
||||
- 原版`8784`和迁移版`8785`使用同一数据库的独立副本，集合竞价、题材库、题材详情、人气热榜、
|
||||
龙虎榜、游资档案和席位别名共7个真实API状态码及JSON一致。
|
||||
- 题材详情在当前外部网络条件下两版均返回HTTP 400；差分只排除每次请求随机生成的
|
||||
`request_id`，错误码与错误内容仍完全一致。
|
||||
- 完整接口摘要见`api-diff.json`。
|
||||
|
||||
## 3. 数据库差分
|
||||
|
||||
- 两个副本均为62个schema对象，哈希均为
|
||||
`60a4e044f0ddb6502e44b45f65bedc1a0a31d4bd596f386d4ccfe153a6c8ddd1`。
|
||||
- `auction_factors` 511914行、`popularity_factors` 232行、`lhb_institution_daily` 47行、
|
||||
`seat_aliases` 0行、`stock_master` 5535行均逐行一致。
|
||||
- 完整表计数与哈希见`database-diff.json`；运行数据库副本已在验收后删除，未提交凭据或正式数据。
|
||||
|
||||
## 4. 真实浏览器检查
|
||||
|
||||
- 1920×1080日间模式检查集合竞价、题材库、人气热榜和龙虎榜四页；均无横向溢出，控制台无
|
||||
错误或警告。
|
||||
- 集合竞价载入30行重点候选；题材库载入394个题材及选中题材成分股；人气热榜载入3个摘要
|
||||
模块和200行综合榜；龙虎榜按当前缓存显示既有不可用空态。
|
||||
- 四页HTML、主JS、CSS及各自页面JS与根目录原版字节哈希一致。
|
||||
- 截图SHA-256：
|
||||
- `app-light-auction-1920x1080.png`：`8065086c8f2b360aeb1004bd60429f3e2d7f1b8c872ec4a965e9a5d0c016b915`
|
||||
- `app-light-themes-1920x1080.png`：`80e1e41101497ee7213e1dadfaa4c9572a1c47b3495edd09e36745ccdb639402`
|
||||
- `app-light-popularity-1920x1080.png`：`614d6b7770f4e9ec72c059ab8a4129486df9eff5c4dd7e8279cc68ebcb80a36b`
|
||||
- `app-light-dragon-tiger-1920x1080.png`：`2f1e2ad3a9bc3874c73bae884fc8744177cef354c7176559da1453fab1f85993`
|
||||
|
||||
## 5. 自动验证与保留边界
|
||||
|
||||
| 验证 | 结果 |
|
||||
|---|---:|
|
||||
| `python -m unittest discover -s tests -q` | 258项通过 |
|
||||
| `python -m unittest tests.test_preservation_slice_market_insights -q` | 6项通过 |
|
||||
| `npx.cmd playwright test --reporter=dot` | 45项通过 |
|
||||
| `git diff --check` | 通过 |
|
||||
|
||||
- 竞价、人气和龙虎榜因子同时服务切片06智能选股，迁移后仍由`ReviewDatabase`原方法名暴露。
|
||||
- 前端资产保持原位置，切片10再按页面职责归档；本切片没有改DOM、CSS、动画或交互。
|
||||
- 没有删除待定代码、没有修改根目录正式数据库、没有切换Docker/NAS。
|
||||
@@ -0,0 +1,61 @@
|
||||
{
|
||||
"all_equal": true,
|
||||
"endpoints": [
|
||||
{
|
||||
"endpoint": "/api/auction?trade_date=2026-07-29",
|
||||
"original_status": 200,
|
||||
"migrated_status": 200,
|
||||
"original_sha256": "523144cc14d876577b7d518cdf38fd2722b13a8f01ed5d2e20dcc38f6a2624ce",
|
||||
"migrated_sha256": "523144cc14d876577b7d518cdf38fd2722b13a8f01ed5d2e20dcc38f6a2624ce",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"endpoint": "/api/themes?trade_date=2026-07-29",
|
||||
"original_status": 200,
|
||||
"migrated_status": 200,
|
||||
"original_sha256": "95c2ad418f18d877d94ec2a71fe6fafd5e329b069d87a9300f7fcac92d4ba5d1",
|
||||
"migrated_sha256": "95c2ad418f18d877d94ec2a71fe6fafd5e329b069d87a9300f7fcac92d4ba5d1",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"endpoint": "/api/themes/detail?code=885001.TI&trade_date=2026-07-29",
|
||||
"original_status": 400,
|
||||
"migrated_status": 400,
|
||||
"original_sha256": "b11a3314b172d3ad6ba28d969fcd6a9a2a4b49e7d29ed804496a4dfecd2a364a",
|
||||
"migrated_sha256": "b11a3314b172d3ad6ba28d969fcd6a9a2a4b49e7d29ed804496a4dfecd2a364a",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"endpoint": "/api/popularity?trade_date=2026-07-29",
|
||||
"original_status": 200,
|
||||
"migrated_status": 200,
|
||||
"original_sha256": "e62a93c41f7c3f95c3d47f8ccaedaa809563c8dd65c04dd54b164f73e6014e94",
|
||||
"migrated_sha256": "e62a93c41f7c3f95c3d47f8ccaedaa809563c8dd65c04dd54b164f73e6014e94",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"endpoint": "/api/dragon-tiger?trade_date=2026-07-29",
|
||||
"original_status": 200,
|
||||
"migrated_status": 200,
|
||||
"original_sha256": "a3998b935377d5fd0673ec5b9214d1b0a64680d155c61e6cff49d0d5fcf0e843",
|
||||
"migrated_sha256": "a3998b935377d5fd0673ec5b9214d1b0a64680d155c61e6cff49d0d5fcf0e843",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"endpoint": "/api/dragon-tiger/profiles",
|
||||
"original_status": 200,
|
||||
"migrated_status": 200,
|
||||
"original_sha256": "dfb1e534c018ec12fd8d8ee0e7fe0f234e73dc7715d7f0a948f74ef8d12d779a",
|
||||
"migrated_sha256": "dfb1e534c018ec12fd8d8ee0e7fe0f234e73dc7715d7f0a948f74ef8d12d779a",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"endpoint": "/api/seat-aliases",
|
||||
"original_status": 200,
|
||||
"migrated_status": 200,
|
||||
"original_sha256": "2b0fb0a6b3e353c69158d61221c2200e4199d0d60dd0b9d99702a22eaa917a78",
|
||||
"migrated_sha256": "2b0fb0a6b3e353c69158d61221c2200e4199d0d60dd0b9d99702a22eaa917a78",
|
||||
"equal": true
|
||||
}
|
||||
]
|
||||
}
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 162 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 66 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 136 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 136 KiB |
@@ -0,0 +1,51 @@
|
||||
{
|
||||
"all_equal": true,
|
||||
"schema": {
|
||||
"object_count": 62,
|
||||
"original_sha256": "60a4e044f0ddb6502e44b45f65bedc1a0a31d4bd596f386d4ccfe153a6c8ddd1",
|
||||
"migrated_sha256": "60a4e044f0ddb6502e44b45f65bedc1a0a31d4bd596f386d4ccfe153a6c8ddd1",
|
||||
"equal": true
|
||||
},
|
||||
"tables": [
|
||||
{
|
||||
"table": "auction_factors",
|
||||
"original_count": 511914,
|
||||
"migrated_count": 511914,
|
||||
"original_sha256": "3d0470787adaf7c4cf5f15f5ad9fa1d67c8fcd4807285ac264eeacdcf5054cd1",
|
||||
"migrated_sha256": "3d0470787adaf7c4cf5f15f5ad9fa1d67c8fcd4807285ac264eeacdcf5054cd1",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "popularity_factors",
|
||||
"original_count": 232,
|
||||
"migrated_count": 232,
|
||||
"original_sha256": "3f4c61a13ceaa1ed9ecb28f86241a8a478a6757d6a44b46f432f73c0226bba7f",
|
||||
"migrated_sha256": "3f4c61a13ceaa1ed9ecb28f86241a8a478a6757d6a44b46f432f73c0226bba7f",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "lhb_institution_daily",
|
||||
"original_count": 47,
|
||||
"migrated_count": 47,
|
||||
"original_sha256": "1f847eac34d2ba576b429da208667f3b67b591d36ee66800803605bbc447970c",
|
||||
"migrated_sha256": "1f847eac34d2ba576b429da208667f3b67b591d36ee66800803605bbc447970c",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "seat_aliases",
|
||||
"original_count": 0,
|
||||
"migrated_count": 0,
|
||||
"original_sha256": "4f53cda18c2baa0c0354bb5f9a3ecbe5ed12ab4d8e11ba873c2f11161202b945",
|
||||
"migrated_sha256": "4f53cda18c2baa0c0354bb5f9a3ecbe5ed12ab4d8e11ba873c2f11161202b945",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "stock_master",
|
||||
"original_count": 5535,
|
||||
"migrated_count": 5535,
|
||||
"original_sha256": "8656e2d189d3520433fb3552e17998e4bf17bcf6838403cddc6719282b23e792",
|
||||
"migrated_sha256": "8656e2d189d3520433fb3552e17998e4bf17bcf6838403cddc6719282b23e792",
|
||||
"equal": true
|
||||
}
|
||||
]
|
||||
}
|
||||
@@ -0,0 +1,69 @@
|
||||
# 切片 06：智能选股、自定义选股与策略持续跟踪
|
||||
|
||||
> 基线：`cf2aad2`（切片 05）
|
||||
> 回档标签：`xiaobai-preservation-slice-06-20260731`
|
||||
> 结论：源码、API、数据库、真实页面和全量回归通过；最终视觉仍等待全站人工验收
|
||||
|
||||
## 1. 原实现归位
|
||||
|
||||
本切片只机械移动原版选股实现，没有从`next/`取用代码，也没有改写策略、因子、权重、排序、
|
||||
盘后候选或跟踪逻辑。
|
||||
|
||||
| 原位置 | 新的唯一实现位置 | 兼容方式 |
|
||||
|---|---|---|
|
||||
| `app/screener.py` | `app/backend/features/screener/engine.py` | 根级模块指向同一模块对象 |
|
||||
| `app/advanced_strategies.py` | `app/backend/features/screener/strategies.py` | 根级模块指向同一模块对象 |
|
||||
| `app/llm_strategy.py` | `app/backend/features/screener/compiler.py` | 根级模块指向同一模块对象 |
|
||||
| `DashboardService`选股入口 | `app/backend/features/screener/service.py` | `ScreenerServiceMixin` |
|
||||
| `ReviewDatabase`选股持久化方法 | `app/backend/features/screener/repository.py` | `ScreenerRepositoryMixin` |
|
||||
| 原策略持续跟踪服务 | `app/backend/features/screener/tracking.py` | 容器直接导入唯一实现 |
|
||||
|
||||
四个同时服务公共行情与选股的方法归入`MarketRepositoryMixin`，没有复制第二套实现：
|
||||
`upsert_stock_master`、`list_stock_master`、`upsert_daily_bars`、`daily_bars_for_date`。
|
||||
|
||||
## 2. 源码与接口等价
|
||||
|
||||
- `test_preservation_slice_screener.py`逐项验证选股引擎、29套策略定义、自然语言策略编译器、
|
||||
13个服务方法、25个选股持久化方法和跟踪服务与原版等价。
|
||||
- 三个根级兼容模块与正式模块暴露同一模块及类对象；原类不再重复保留已移动方法。
|
||||
- 原版与迁移版的`/api/screener/setup`、`/api/screener/tracking`和`/api/screener/run`
|
||||
状态码及业务JSON一致。
|
||||
- 差分仅规范化策略启动刷新和顺序执行必然变化的数组顺序与`updated_at`；其他字段仍逐项比较。
|
||||
- 请求与差分结果见`api-requests.json`和`api-diff.json`。
|
||||
|
||||
## 3. 数据库差分
|
||||
|
||||
- 两个临时副本均为62个schema对象，结构完全一致。
|
||||
- 13张关键表逐行一致，覆盖约143万条日线、77万条技术指标和51万条竞价因子。
|
||||
- `screener_runs`两边均为251行；会话和运行时间戳只在临时副本内规范化。
|
||||
- 正式`data/review.db`和`app/data/review.db`未写入测试策略、测试会话或差分时间戳。
|
||||
- 完整表计数与哈希见`database-diff.json`。
|
||||
|
||||
## 4. 真实浏览器检查
|
||||
|
||||
- 已登录迁移版真实服务，分别检查阶段选股、策略选股、自定义选股和策略持续跟踪。
|
||||
- 阶段选股显示退潮、置信度92%、匹配策略和262日因子就绪状态；策略选股载入29套策略。
|
||||
- 自定义选股保留因子权重、过滤条件、公式入口、手动执行和独立候选结果。
|
||||
- 跟踪页只展示手动加入的候选，批次、标的、T+1/T+3进度、胜率和移除操作均正常。
|
||||
- 各检查状态无横向溢出、无加载遮罩残留，浏览器控制台无错误。
|
||||
- 截图SHA-256：
|
||||
- `curated-screener.jpg`：`529ecd0947b34b42eb63d166c6804b0041e26fa362a171e4d0d7b584e073c05e`
|
||||
- `custom-screener.jpg`：`7f31f50d44bc63a488277be0a74232e2ac256779d0accda42b96cc1eef88cc6e`
|
||||
- `tracking.jpg`：`b19b1bc0cd4a98c29500917676bdcc4cc9a430a0c72dea8aff1110ad911db6e3`
|
||||
|
||||
## 5. 自动验证与保留边界
|
||||
|
||||
| 验证 | 结果 |
|
||||
|---|---:|
|
||||
| 原版`python -m unittest discover -s tests -q` | 231项通过 |
|
||||
| 迁移版`python -m unittest discover -s tests -q` | 265项通过 |
|
||||
| `python -m unittest tests.test_preservation_slice_screener -q` | 7项通过 |
|
||||
| `npx.cmd playwright test --reporter=dot` | 45项通过 |
|
||||
| `git diff --check` | 通过 |
|
||||
|
||||
- Windows下由Playwright自行创建静态服务器时存在子进程不退出的测试基线问题；复用独立的8876
|
||||
测试服务器后45项用例在2.1分钟内通过并正常返回退出码0。
|
||||
- 选股引擎保留原版数据客户端依赖，边界测试将其与普通页面Service区分；后续只可在不改变行为且有
|
||||
独立差分证据时治理该依赖。
|
||||
- 前端资产保持原位置，切片10再按页面职责归档；本切片没有改DOM、CSS、动画或交互。
|
||||
- 没有删除待定代码、没有修改正式数据库、没有切换Docker/NAS。
|
||||
@@ -0,0 +1,38 @@
|
||||
{
|
||||
"all_equal": true,
|
||||
"endpoints": [
|
||||
{
|
||||
"name": "盘后自动候选与策略库",
|
||||
"method": "GET",
|
||||
"endpoint": "/api/screener/setup?trade_date=2026-07-30",
|
||||
"original_status": 200,
|
||||
"migrated_status": 200,
|
||||
"original_sha256": "e33a10e94accda4948c2af53887ea97736697f588ee1a92f561aa5c4bb28c2fd",
|
||||
"migrated_sha256": "e33a10e94accda4948c2af53887ea97736697f588ee1a92f561aa5c4bb28c2fd",
|
||||
"equal": true,
|
||||
"first_difference": null
|
||||
},
|
||||
{
|
||||
"name": "策略持续跟踪",
|
||||
"method": "GET",
|
||||
"endpoint": "/api/screener/tracking?limit=12",
|
||||
"original_status": 200,
|
||||
"migrated_status": 200,
|
||||
"original_sha256": "f4476f53152b6a2ce01f2918da5999246cfd0ad7d0b71e80fdea8fa59c7456c2",
|
||||
"migrated_sha256": "f4476f53152b6a2ce01f2918da5999246cfd0ad7d0b71e80fdea8fa59c7456c2",
|
||||
"equal": true,
|
||||
"first_difference": null
|
||||
},
|
||||
{
|
||||
"name": "自定义选股执行",
|
||||
"method": "POST",
|
||||
"endpoint": "/api/screener/run",
|
||||
"original_status": 200,
|
||||
"migrated_status": 200,
|
||||
"original_sha256": "b3866e7d270aa222bd2bad0c0276b7e81ccd48e5f3a75941a32ef8e559a92c7b",
|
||||
"migrated_sha256": "b3866e7d270aa222bd2bad0c0276b7e81ccd48e5f3a75941a32ef8e559a92c7b",
|
||||
"equal": true,
|
||||
"first_difference": null
|
||||
}
|
||||
]
|
||||
}
|
||||
@@ -0,0 +1,59 @@
|
||||
[
|
||||
{
|
||||
"name": "盘后自动候选与策略库",
|
||||
"method": "GET",
|
||||
"endpoint": "/api/screener/setup?trade_date=2026-07-30",
|
||||
"sort_lists": {
|
||||
"$.strategies": "id"
|
||||
},
|
||||
"exclude_paths": [
|
||||
"$.strategies[].updated_at"
|
||||
]
|
||||
},
|
||||
{
|
||||
"name": "策略持续跟踪",
|
||||
"method": "GET",
|
||||
"endpoint": "/api/screener/tracking?limit=12"
|
||||
},
|
||||
{
|
||||
"name": "自定义选股执行",
|
||||
"method": "POST",
|
||||
"endpoint": "/api/screener/run",
|
||||
"exclude_paths": [
|
||||
"$.result.meta.updated_at"
|
||||
],
|
||||
"payload": {
|
||||
"trade_date": "2026-07-30",
|
||||
"regime": "retreat",
|
||||
"strategy_name": "保真迁移差分策略",
|
||||
"mode": "quant",
|
||||
"run_backtest": false,
|
||||
"formula": {
|
||||
"meta": {
|
||||
"library": "quant",
|
||||
"category": "量化公式"
|
||||
},
|
||||
"universe": {
|
||||
"exclude_st": true,
|
||||
"listed_days_min": 120
|
||||
},
|
||||
"filters": [
|
||||
{
|
||||
"field": "pct_chg",
|
||||
"op": "between",
|
||||
"value": [-2, 2]
|
||||
}
|
||||
],
|
||||
"score": [
|
||||
{
|
||||
"field": "amount_billion",
|
||||
"weight": 1,
|
||||
"direction": "desc"
|
||||
}
|
||||
],
|
||||
"limit": 5,
|
||||
"min_score": 0
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 87 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 89 KiB |
@@ -0,0 +1,115 @@
|
||||
{
|
||||
"all_equal": true,
|
||||
"schema": {
|
||||
"object_count": 62,
|
||||
"original_sha256": "60a4e044f0ddb6502e44b45f65bedc1a0a31d4bd596f386d4ccfe153a6c8ddd1",
|
||||
"migrated_sha256": "60a4e044f0ddb6502e44b45f65bedc1a0a31d4bd596f386d4ccfe153a6c8ddd1",
|
||||
"equal": true
|
||||
},
|
||||
"tables": [
|
||||
{
|
||||
"table": "stock_master",
|
||||
"original_count": 5535,
|
||||
"migrated_count": 5535,
|
||||
"original_sha256": "8656e2d189d3520433fb3552e17998e4bf17bcf6838403cddc6719282b23e792",
|
||||
"migrated_sha256": "8656e2d189d3520433fb3552e17998e4bf17bcf6838403cddc6719282b23e792",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "daily_bars",
|
||||
"original_count": 1430501,
|
||||
"migrated_count": 1430501,
|
||||
"original_sha256": "d72a06755d92d6be95ab4f4c98a4718e973bee59d4f381b01acb963b6d7ae9e6",
|
||||
"migrated_sha256": "d72a06755d92d6be95ab4f4c98a4718e973bee59d4f381b01acb963b6d7ae9e6",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "benchmark_bars",
|
||||
"original_count": 261,
|
||||
"migrated_count": 261,
|
||||
"original_sha256": "691f7666dddcfdf7ebe8ba1d3e1ddd8f2d9670d1d7a06e740a929f3ef32ec1c9",
|
||||
"migrated_sha256": "691f7666dddcfdf7ebe8ba1d3e1ddd8f2d9670d1d7a06e740a929f3ef32ec1c9",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "daily_indicators",
|
||||
"original_count": 770238,
|
||||
"migrated_count": 770238,
|
||||
"original_sha256": "cd5a71c7d8de201ff2e3e58c64e2617a31d247911732fbddf81acb025d31fde9",
|
||||
"migrated_sha256": "cd5a71c7d8de201ff2e3e58c64e2617a31d247911732fbddf81acb025d31fde9",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "fundamental_indicators",
|
||||
"original_count": 49490,
|
||||
"migrated_count": 49490,
|
||||
"original_sha256": "4b6859256063d3a0238ba3cbbbe2842f1278cf88d44b0f6114f1af978b8bfa7d",
|
||||
"migrated_sha256": "4b6859256063d3a0238ba3cbbbe2842f1278cf88d44b0f6114f1af978b8bfa7d",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "moneyflow_daily",
|
||||
"original_count": 36368,
|
||||
"migrated_count": 36368,
|
||||
"original_sha256": "f7591dec150c7e0094568e3c46d77fc54e14a803223e15940dc54af1017bd83d",
|
||||
"migrated_sha256": "f7591dec150c7e0094568e3c46d77fc54e14a803223e15940dc54af1017bd83d",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "earnings_events",
|
||||
"original_count": 580,
|
||||
"migrated_count": 580,
|
||||
"original_sha256": "ebac03fe9ef491a6d6d2b5bc8f5ee08b609fa0ebe5992087fa067216490a6b9d",
|
||||
"migrated_sha256": "ebac03fe9ef491a6d6d2b5bc8f5ee08b609fa0ebe5992087fa067216490a6b9d",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "auction_factors",
|
||||
"original_count": 511914,
|
||||
"migrated_count": 511914,
|
||||
"original_sha256": "3d0470787adaf7c4cf5f15f5ad9fa1d67c8fcd4807285ac264eeacdcf5054cd1",
|
||||
"migrated_sha256": "3d0470787adaf7c4cf5f15f5ad9fa1d67c8fcd4807285ac264eeacdcf5054cd1",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "popularity_factors",
|
||||
"original_count": 232,
|
||||
"migrated_count": 232,
|
||||
"original_sha256": "3f4c61a13ceaa1ed9ecb28f86241a8a478a6757d6a44b46f432f73c0226bba7f",
|
||||
"migrated_sha256": "3f4c61a13ceaa1ed9ecb28f86241a8a478a6757d6a44b46f432f73c0226bba7f",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "lhb_institution_daily",
|
||||
"original_count": 47,
|
||||
"migrated_count": 47,
|
||||
"original_sha256": "1f847eac34d2ba576b429da208667f3b67b591d36ee66800803605bbc447970c",
|
||||
"migrated_sha256": "1f847eac34d2ba576b429da208667f3b67b591d36ee66800803605bbc447970c",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "screener_strategies",
|
||||
"original_count": 36,
|
||||
"migrated_count": 36,
|
||||
"original_sha256": "e58859dbd478272d71bf85b6bc4ba45e609baccba74bb818b044d93bafa07312",
|
||||
"migrated_sha256": "e58859dbd478272d71bf85b6bc4ba45e609baccba74bb818b044d93bafa07312",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "screener_runs",
|
||||
"original_count": 251,
|
||||
"migrated_count": 251,
|
||||
"original_sha256": "117b0b95a362c91ddd70ca2def1e62e96e0ade2494ba4302bab5154c8d384498",
|
||||
"migrated_sha256": "117b0b95a362c91ddd70ca2def1e62e96e0ade2494ba4302bab5154c8d384498",
|
||||
"equal": true
|
||||
},
|
||||
{
|
||||
"table": "strategy_tracks",
|
||||
"original_count": 16,
|
||||
"migrated_count": 16,
|
||||
"original_sha256": "f0639b1386b69c92fd815f929864b3b2457a6bf4ca515bee2dca87e9ea29e8c4",
|
||||
"migrated_sha256": "f0639b1386b69c92fd815f929864b3b2457a6bf4ca515bee2dca87e9ea29e8c4",
|
||||
"equal": true
|
||||
}
|
||||
]
|
||||
}
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 68 KiB |
@@ -1,6 +1,6 @@
|
||||
{
|
||||
"schema_version": 1,
|
||||
"updated_at": "2026-07-31T01:38:00+08:00",
|
||||
"updated_at": "2026-07-31T03:56:00+08:00",
|
||||
"status": "active",
|
||||
"migration_mode": "behavior_preserving_source_migration",
|
||||
"source_of_truth": "current_original_webapp_runtime_and_source",
|
||||
@@ -9,10 +9,10 @@
|
||||
"failed_roots": [
|
||||
"next"
|
||||
],
|
||||
"current_slice": "slice-04-ladder-rotation",
|
||||
"last_completed_slice": "slice-03-sentiment-pools-performance",
|
||||
"last_checkpoint": "xiaobai-preservation-slice-03-20260731",
|
||||
"next_action": "capture_slice-04_ladder_rotation_contracts_then_move_original_implementations",
|
||||
"current_slice": "slice-07-mentor-skills-llm-streaming",
|
||||
"last_completed_slice": "slice-06-screener-custom-tracking",
|
||||
"last_checkpoint": "xiaobai-preservation-slice-06-20260731",
|
||||
"next_action": "capture_slice-07_mentor_skill_model_pool_and_streaming_contracts_then_move_original_implementations",
|
||||
"authoritative_documents": [
|
||||
"AGENTS.md",
|
||||
"docs/migration/原版保真迁移总纲.md",
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
# 小白复盘保真迁移账本
|
||||
|
||||
> 当前状态：正式迁移，切片03“情绪周期、五类股池与涨停表现”已完成
|
||||
> 当前状态：正式迁移，切片06“智能选股、自定义选股与策略持续跟踪”已完成
|
||||
|
||||
本账本是上下文恢复和人工审计的连续记录。任何迁移提交必须在同一提交中更新本文件及
|
||||
`保真迁移状态.json`。
|
||||
@@ -23,6 +23,9 @@
|
||||
| 2026-07-31 | `xiaobai-preservation-slice-01-20260731` | 启动、HTTP、账号、会员与系统管理原实现归位 | 自动差分通过，进入切片02 |
|
||||
| 2026-07-31 | `xiaobai-preservation-slice-02-20260731` | 公共行情、搜索、详情、图表与数据适配原实现归位 | 自动与浏览器差分通过，进入切片03 |
|
||||
| 2026-07-31 | `xiaobai-preservation-slice-03-20260731` | 情绪周期、五类股池与涨停表现原实现归位 | 自动、API与浏览器差分通过，进入切片04 |
|
||||
| 2026-07-31 | `xiaobai-preservation-slice-04-20260731` | 市场天梯与板块轮动原实现归位 | 自动、API与浏览器差分通过，进入切片05 |
|
||||
| 2026-07-31 | `xiaobai-preservation-slice-05-20260731` | 集合竞价、题材库、人气热榜与龙虎榜原实现归位 | 自动、API、数据库与浏览器差分通过，进入切片06 |
|
||||
| 2026-07-31 | `xiaobai-preservation-slice-06-20260731` | 智能选股、自定义选股与策略持续跟踪原实现归位 | 自动、API、数据库与浏览器差分通过，进入切片07 |
|
||||
|
||||
## 资产处置登记
|
||||
|
||||
@@ -40,6 +43,16 @@
|
||||
| `sentiment_engine.py` | 情绪周期计算 | 总览、轮动、选股 | 移动并保留兼容别名 | `app/backend/features/sentiment/engine.py` | 文件哈希与原版一致；248项Python与45项Playwright通过 | 已移动 |
|
||||
| `DashboardService`情绪及股池原因方法 | 业务服务 | 情绪页、五类股池、涨停表现 | 按职责机械移动 | `app/backend/features/sentiment/`、`app/backend/features/pools/` | 8个方法AST与原版一致；真实API完全一致 | 已移动 |
|
||||
| `ReviewDatabase`原因覆盖方法 | 持久化 | 股池原因人工覆盖 | 按职责机械移动 | `app/backend/features/pools/repository.py` | 2个方法AST与原版一致；数据库schema哈希一致 | 已移动 |
|
||||
| `DashboardService`板块轮动方法 | 业务服务 | 板块轮动页 | 按职责机械移动 | `app/backend/features/rotation/service.py` | 2个方法AST、真实API与原版一致 | 已移动 |
|
||||
| Tushare天梯与轮动构造函数 | 公共数据计算 | 市场天梯、板块轮动 | 原位置保持唯一实现 | `app/backend/data/providers/tushare_client.py` | 2个构造函数AST与原版一致 | 已归位 |
|
||||
| `MarketInsightsService` | 共享市场洞察服务 | 集合竞价、题材库、人气热榜 | 整体机械移动，保留唯一共享实现 | `app/backend/features/market/insights.py` | 22个方法AST与原版一致；根级模块为同一类对象别名 | 已移动 |
|
||||
| `DashboardService`竞价、题材、人气与龙虎榜方法 | 业务服务 | 切片05四类页面与API | 按职责机械移动 | `app/backend/features/auction/`、`themes/`、`popularity/`、`dragon_tiger/` | 8个方法AST、7个真实API与原版一致 | 已移动 |
|
||||
| `ReviewDatabase`竞价、人气与龙虎榜方法 | 持久化 | 市场洞察及后续智能选股 | 按职责机械移动并保持Mixin原接口 | `app/backend/features/auction/repository.py`、`popularity/repository.py`、`dragon_tiger/repository.py` | 7个方法AST一致；62个schema对象及5张关键表逐行一致 | 已移动 |
|
||||
| Tushare游资名录与龙虎榜实现 | 公共数据计算 | 龙虎榜与游资档案 | 原位置保持唯一实现 | `app/backend/data/providers/tushare_client.py` | 2个方法AST与原版一致 | 已归位 |
|
||||
| 选股引擎、策略库与公式编译器 | 业务计算 | 阶段、策略、自定义选股 | 整体机械移动并保留兼容别名 | `app/backend/features/screener/engine.py`、`strategies.py`、`compiler.py` | 原版AST/源码等价；根级模块为同一模块对象 | 已移动 |
|
||||
| `DashboardService`选股方法 | 业务服务 | 选股三个工作区 | 按职责机械移动 | `app/backend/features/screener/service.py` | 13个方法AST及3个真实API一致 | 已移动 |
|
||||
| `ReviewDatabase`选股方法 | 持久化 | 因子、策略运行、候选与跟踪 | 按职责机械移动并保持Mixin原接口 | `app/backend/features/screener/repository.py` | 25个方法AST一致；62个schema对象及13张关键表逐行一致 | 已移动 |
|
||||
| `StrategyTrackingService` | 业务服务 | 手动候选持续跟踪 | 保持唯一实现并调整容器导入 | `app/backend/features/screener/tracking.py` | 类定义AST、真实API与浏览器行为一致 | 已归位 |
|
||||
|
||||
处置只允许：`原样保留`、`移动`、`合并重复`、`待定`、`确认废弃`。
|
||||
|
||||
@@ -87,6 +100,36 @@
|
||||
- 回档：标签`xiaobai-preservation-slice-03-20260731`。
|
||||
- 完整证据：`docs/migration/evidence/slice-03/README.md`。
|
||||
|
||||
已完成切片：`slice-04-ladder-rotation`。
|
||||
|
||||
- 原版基线：提交`b3555d2`，即切片03回档点。
|
||||
- 迁移范围：2个板块轮动服务方法；市场天梯继续使用切片02已归位的原Tushare数据构造实现。
|
||||
- 兼容边界：`DashboardService`通过`RotationServiceMixin`保持所有原调用；天梯不制造空服务或第二套计算。
|
||||
- API与错误：天梯与9日轮动历史JSON完全一致；成分股两版均返回同一Tushare外部失败语义。
|
||||
- 验收：252项Python测试、4项切片源码等价测试、45项Playwright测试及两个真实页面流程通过。
|
||||
- 回档：标签`xiaobai-preservation-slice-04-20260731`。
|
||||
- 完整证据：`docs/migration/evidence/slice-04/README.md`。
|
||||
|
||||
已完成切片：`slice-05-auction-themes-popularity-dragon-tiger`。
|
||||
|
||||
- 原版基线：提交`814e757`，即切片04回档点。
|
||||
- 迁移范围：共享市场洞察服务、竞价/题材/人气入口、龙虎榜与游资档案服务、7个相关持久化方法。
|
||||
- 兼容边界：根级`market_insights.py`保留同一类对象别名；竞价与人气共用候选热度逻辑，不复制第二套实现。
|
||||
- API与数据库：7个真实API逐字段一致，仅排除每次请求必然变化的`request_id`；62个schema对象与5张关键表完全一致。
|
||||
- 验收：258项Python测试、6项切片源码等价测试、45项Playwright测试及四个真实页面流程通过。
|
||||
- 回档：标签`xiaobai-preservation-slice-05-20260731`。
|
||||
- 完整证据：`docs/migration/evidence/slice-05/README.md`。
|
||||
|
||||
已完成切片：`slice-06-screener-custom-tracking`。
|
||||
|
||||
- 原版基线：提交`cf2aad2`，即切片05回档点。
|
||||
- 迁移范围：完整选股引擎、29套高级策略、策略编译器、13个页面服务方法、25个持久化方法及持续跟踪。
|
||||
- 兼容边界：三个根级模块指向正式模块对象；选股引擎保留原版数据客户端依赖，不为通过边界测试改写算法。
|
||||
- API与数据库：3个真实API业务JSON一致；62个schema对象和13张关键表逐行一致。
|
||||
- 验收：原版231项、迁移版265项Python测试、7项切片源码等价测试、45项Playwright及四个真实工作区通过。
|
||||
- 回档：标签`xiaobai-preservation-slice-06-20260731`。
|
||||
- 完整证据：`docs/migration/evidence/slice-06/README.md`。
|
||||
|
||||
## 决策记录
|
||||
|
||||
| 日期 | 决策 | 原因 |
|
||||
|
||||
Reference in New Issue
Block a user