678 lines
29 KiB
Python
678 lines
29 KiB
Python
"""今日计划页面视图。"""
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
from collections import Counter
|
||
from datetime import date, datetime, timedelta
|
||
from typing import Dict, List, Optional
|
||
|
||
import numpy as np
|
||
import pandas as pd
|
||
import streamlit as st
|
||
|
||
from app.backtest.engine import BacktestEngine, PortfolioState, BtConfig
|
||
from app.utils.portfolio import list_investment_pool
|
||
from app.utils.db import db_session
|
||
|
||
from app.ui.shared import (
|
||
LOGGER,
|
||
LOG_EXTRA,
|
||
get_latest_trade_date,
|
||
get_query_params,
|
||
set_query_params,
|
||
)
|
||
|
||
|
||
def render_today_plan() -> None:
|
||
LOGGER.info("渲染今日计划页面", extra=LOG_EXTRA)
|
||
st.header("今日计划")
|
||
latest_trade_date = get_latest_trade_date()
|
||
if latest_trade_date:
|
||
st.caption(f"最新交易日：{latest_trade_date.isoformat()}（统计数据请见左侧系统监控）")
|
||
else:
|
||
st.caption("统计与决策概览现已移至左侧'系统监控'侧栏。")
|
||
try:
|
||
with db_session(read_only=True) as conn:
|
||
date_rows = conn.execute(
|
||
"""
|
||
SELECT DISTINCT trade_date
|
||
FROM agent_utils
|
||
ORDER BY trade_date DESC
|
||
LIMIT 30
|
||
"""
|
||
).fetchall()
|
||
except Exception: # noqa: BLE001
|
||
LOGGER.exception("加载 agent_utils 失败", extra=LOG_EXTRA)
|
||
st.warning("暂未写入部门/代理决策，请先运行回测或策略评估流程。")
|
||
return
|
||
|
||
trade_dates = [row["trade_date"] for row in date_rows]
|
||
if not trade_dates:
|
||
st.info("暂无决策记录，完成一次回测后即可在此查看部门意见与投票结果。")
|
||
return
|
||
|
||
query = get_query_params()
|
||
default_trade_date = query.get("date", [trade_dates[0]])[0]
|
||
try:
|
||
default_idx = trade_dates.index(default_trade_date)
|
||
except ValueError:
|
||
default_idx = 0
|
||
trade_date = st.selectbox("交易日", trade_dates, index=default_idx)
|
||
|
||
with db_session(read_only=True) as conn:
|
||
code_rows = conn.execute(
|
||
"""
|
||
SELECT DISTINCT ts_code
|
||
FROM agent_utils
|
||
WHERE trade_date = ?
|
||
ORDER BY ts_code
|
||
""",
|
||
(trade_date,),
|
||
).fetchall()
|
||
symbols = [row["ts_code"] for row in code_rows]
|
||
|
||
detail_tab, assistant_tab = st.tabs(["标的详情", "投资助理模式"])
|
||
with assistant_tab:
|
||
_render_today_plan_assistant_view(trade_date)
|
||
|
||
with detail_tab:
|
||
if not symbols:
|
||
st.info("所选交易日暂无 agent_utils 记录。")
|
||
else:
|
||
_render_today_plan_symbol_view(trade_date, symbols, query)
|
||
|
||
|
||
def _render_today_plan_assistant_view(trade_date: str | int | date) -> None:
|
||
st.info("已开启投资助理模式：以下内容为组合级（去标的）建议，不包含任何具体标的代码。")
|
||
try:
|
||
candidates = list_investment_pool(trade_date=trade_date)
|
||
if candidates:
|
||
scores = [float(item.score or 0.0) for item in candidates]
|
||
statuses = [item.status or "UNKNOWN" for item in candidates]
|
||
tags: List[str] = []
|
||
rationales: List[str] = []
|
||
for item in candidates:
|
||
if getattr(item, "tags", None):
|
||
tags.extend(item.tags)
|
||
if getattr(item, "rationale", None):
|
||
rationales.append(str(item.rationale))
|
||
cnt = Counter(statuses)
|
||
tag_cnt = Counter(tags)
|
||
st.subheader("候选池聚合概览（已匿名化）")
|
||
col_a, col_b, col_c = st.columns(3)
|
||
col_a.metric("候选数", f"{len(candidates)}")
|
||
col_b.metric("平均评分", f"{np.mean(scores):.3f}" if scores else "-")
|
||
col_c.metric("中位评分", f"{np.median(scores):.3f}" if scores else "-")
|
||
|
||
st.write("状态分布：")
|
||
st.json(dict(cnt))
|
||
|
||
if tag_cnt:
|
||
st.write("常见标签（示例）：")
|
||
st.json(dict(tag_cnt.most_common(10)))
|
||
|
||
if rationales:
|
||
st.write("汇总理由（节选，不含代码）：")
|
||
seen = set()
|
||
excerpts = []
|
||
for rationale in rationales:
|
||
text = rationale.strip()
|
||
if text and text not in seen:
|
||
seen.add(text)
|
||
excerpts.append(text)
|
||
if len(excerpts) >= 3:
|
||
break
|
||
for idx, excerpt in enumerate(excerpts, start=1):
|
||
st.markdown(f"**理由 {idx}:** {excerpt}")
|
||
|
||
avg_score = float(np.mean(scores)) if scores else 0.0
|
||
suggest_pct = max(0.0, min(0.3, 0.10 + (avg_score - 0.5) * 0.2))
|
||
st.subheader("组合级建议（不指定标的）")
|
||
st.write(
|
||
f"基于候选池平均评分 {avg_score:.3f}，建议今日用于新增买入的现金比例约为 {suggest_pct:.0%}。"
|
||
)
|
||
st.write(
|
||
"建议分配思路：在候选池中挑选若干得分较高的标的按目标权重等比例分配，或以分批买入的方式分摊入场时点。"
|
||
)
|
||
if st.button("生成组合级操作建议（仅输出，不执行）"):
|
||
st.success("已生成组合级建议（仅供参考）。")
|
||
st.write({
|
||
"候选数": len(candidates),
|
||
"平均评分": avg_score,
|
||
"建议新增买入比例": f"{suggest_pct:.0%}",
|
||
})
|
||
else:
|
||
st.info("所选交易日暂无候选投资池数据。")
|
||
except Exception: # noqa: BLE001
|
||
LOGGER.exception("加载候选池聚合信息失败", extra=LOG_EXTRA)
|
||
st.error("加载候选池数据时发生错误。")
|
||
|
||
|
||
def _render_today_plan_symbol_view(
|
||
trade_date: str | int | date,
|
||
symbols: List[str],
|
||
query_params: Dict[str, List[str]],
|
||
) -> None:
|
||
default_ts = query_params.get("code", [symbols[0]])[0]
|
||
try:
|
||
default_ts_idx = symbols.index(default_ts)
|
||
except ValueError:
|
||
default_ts_idx = 0
|
||
ts_code = st.selectbox("标的", symbols, index=default_ts_idx)
|
||
batch_symbols = st.multiselect("批量重评估（可多选）", symbols, default=[])
|
||
|
||
if st.button("一键重评估所有标的", type="primary", width='stretch'):
|
||
with st.spinner("正在对所有标的进行重评估，请稍候..."):
|
||
try:
|
||
trade_date_obj: Optional[date] = None
|
||
try:
|
||
trade_date_obj = date.fromisoformat(str(trade_date))
|
||
except Exception:
|
||
try:
|
||
trade_date_obj = datetime.strptime(str(trade_date), "%Y%m%d").date()
|
||
except Exception:
|
||
pass
|
||
if trade_date_obj is None:
|
||
raise ValueError(f"无法解析交易日：{trade_date}")
|
||
|
||
progress = st.progress(0.0)
|
||
changes_all: List[Dict[str, object]] = []
|
||
success_count = 0
|
||
error_count = 0
|
||
|
||
for idx, code in enumerate(symbols, start=1):
|
||
try:
|
||
with db_session(read_only=True) as conn:
|
||
before_rows = conn.execute(
|
||
"SELECT agent, action FROM agent_utils WHERE trade_date = ? AND ts_code = ?",
|
||
(trade_date, code),
|
||
).fetchall()
|
||
before_map = {row["agent"]: row["action"] for row in before_rows}
|
||
|
||
cfg = BtConfig(
|
||
id="reeval_ui_all",
|
||
name="UI All Re-eval",
|
||
start_date=trade_date_obj,
|
||
end_date=trade_date_obj,
|
||
universe=[code],
|
||
params={},
|
||
)
|
||
engine = BacktestEngine(cfg)
|
||
state = PortfolioState()
|
||
engine.simulate_day(trade_date_obj, state)
|
||
|
||
with db_session(read_only=True) as conn:
|
||
after_rows = conn.execute(
|
||
"SELECT agent, action FROM agent_utils WHERE trade_date = ? AND ts_code = ?",
|
||
(trade_date, code),
|
||
).fetchall()
|
||
for row in after_rows:
|
||
agent = row["agent"]
|
||
new_action = row["action"]
|
||
old_action = before_map.get(agent)
|
||
if new_action != old_action:
|
||
changes_all.append(
|
||
{"代码": code, "代理": agent, "原动作": old_action, "新动作": new_action}
|
||
)
|
||
success_count += 1
|
||
except Exception: # noqa: BLE001
|
||
LOGGER.exception("重评估 %s 失败", code, extra=LOG_EXTRA)
|
||
error_count += 1
|
||
|
||
progress.progress(idx / len(symbols))
|
||
|
||
if error_count > 0:
|
||
st.error(f"一键重评估完成：成功 {success_count} 个，失败 {error_count} 个")
|
||
else:
|
||
st.success(f"一键重评估完成：所有 {success_count} 个标的重评估成功")
|
||
|
||
if changes_all:
|
||
st.write("检测到以下动作变更：")
|
||
st.dataframe(pd.DataFrame(changes_all), hide_index=True, width='stretch')
|
||
|
||
st.rerun()
|
||
except Exception as exc: # noqa: BLE001
|
||
LOGGER.exception("一键重评估失败", extra=LOG_EXTRA)
|
||
st.error(f"一键重评估执行过程中发生错误：{exc}")
|
||
|
||
set_query_params(date=str(trade_date), code=str(ts_code))
|
||
|
||
with db_session(read_only=True) as conn:
|
||
rows = conn.execute(
|
||
"""
|
||
SELECT agent, action, utils, feasible, weight
|
||
FROM agent_utils
|
||
WHERE trade_date = ? AND ts_code = ?
|
||
ORDER BY CASE WHEN agent = 'global' THEN 1 ELSE 0 END, agent
|
||
""",
|
||
(trade_date, ts_code),
|
||
).fetchall()
|
||
|
||
if not rows:
|
||
st.info("未查询到详细决策记录，稍后再试。")
|
||
return
|
||
|
||
try:
|
||
feasible_actions = json.loads(rows[0]["feasible"] or "[]")
|
||
except (KeyError, TypeError, json.JSONDecodeError):
|
||
feasible_actions = []
|
||
|
||
global_info = None
|
||
dept_records: List[Dict[str, object]] = []
|
||
dept_details: Dict[str, Dict[str, object]] = {}
|
||
agent_records: List[Dict[str, object]] = []
|
||
|
||
for item in rows:
|
||
agent_name = item["agent"]
|
||
action = item["action"]
|
||
weight = float(item["weight"] or 0.0)
|
||
try:
|
||
utils = json.loads(item["utils"] or "{}")
|
||
except json.JSONDecodeError:
|
||
utils = {}
|
||
|
||
if agent_name == "global":
|
||
global_info = {
|
||
"action": action,
|
||
"confidence": float(utils.get("_confidence", 0.0)),
|
||
"target_weight": float(utils.get("_target_weight", 0.0)),
|
||
"department_votes": utils.get("_department_votes", {}),
|
||
"requires_review": bool(utils.get("_requires_review", False)),
|
||
"scope_values": utils.get("_scope_values", {}),
|
||
"close_series": utils.get("_close_series", []),
|
||
"turnover_series": utils.get("_turnover_series", []),
|
||
"department_supplements": utils.get("_department_supplements", {}),
|
||
"department_dialogue": utils.get("_department_dialogue", {}),
|
||
"department_telemetry": utils.get("_department_telemetry", {}),
|
||
}
|
||
continue
|
||
|
||
if agent_name.startswith("dept_"):
|
||
code = agent_name.split("dept_", 1)[-1]
|
||
signals = utils.get("_signals", [])
|
||
risks = utils.get("_risks", [])
|
||
supplements = utils.get("_supplements", [])
|
||
dialogue = utils.get("_dialogue", [])
|
||
telemetry = utils.get("_telemetry", {})
|
||
dept_records.append(
|
||
{
|
||
"部门": code,
|
||
"行动": action,
|
||
"信心": float(utils.get("_confidence", 0.0)),
|
||
"权重": weight,
|
||
"摘要": utils.get("_summary", ""),
|
||
"核心信号": "；".join(signals) if isinstance(signals, list) else signals,
|
||
"风险提示": "；".join(risks) if isinstance(risks, list) else risks,
|
||
"补充次数": len(supplements) if isinstance(supplements, list) else 0,
|
||
}
|
||
)
|
||
dept_details[code] = {
|
||
"supplements": supplements if isinstance(supplements, list) else [],
|
||
"dialogue": dialogue if isinstance(dialogue, list) else [],
|
||
"summary": utils.get("_summary", ""),
|
||
"signals": signals,
|
||
"risks": risks,
|
||
"telemetry": telemetry if isinstance(telemetry, dict) else {},
|
||
}
|
||
else:
|
||
score_map = {
|
||
key: float(val)
|
||
for key, val in utils.items()
|
||
if not str(key).startswith("_")
|
||
}
|
||
agent_records.append(
|
||
{
|
||
"代理": agent_name,
|
||
"建议动作": action,
|
||
"权重": weight,
|
||
"SELL": score_map.get("SELL", 0.0),
|
||
"HOLD": score_map.get("HOLD", 0.0),
|
||
"BUY_S": score_map.get("BUY_S", 0.0),
|
||
"BUY_M": score_map.get("BUY_M", 0.0),
|
||
"BUY_L": score_map.get("BUY_L", 0.0),
|
||
}
|
||
)
|
||
|
||
if feasible_actions:
|
||
st.caption(f"可行操作集合：{', '.join(feasible_actions)}")
|
||
|
||
st.subheader("全局策略")
|
||
if global_info:
|
||
col1, col2, col3 = st.columns(3)
|
||
col1.metric("最终行动", global_info["action"])
|
||
col2.metric("信心", f"{global_info['confidence']:.2f}")
|
||
col3.metric("目标权重", f"{global_info['target_weight']:+.2%}")
|
||
if global_info["department_votes"]:
|
||
st.json(global_info["department_votes"])
|
||
if global_info["requires_review"]:
|
||
st.warning("部门分歧较大，已标记为需人工复核。")
|
||
with st.expander("基础上下文数据", expanded=False):
|
||
scope = global_info.get("scope_values") or {}
|
||
close_series = global_info.get("close_series") or []
|
||
turnover_series = global_info.get("turnover_series") or []
|
||
st.write("最新字段：")
|
||
if scope:
|
||
st.json(scope)
|
||
st.download_button(
|
||
"下载字段(JSON)",
|
||
data=json.dumps(scope, ensure_ascii=False, indent=2),
|
||
file_name=f"{ts_code}_{trade_date}_scope.json",
|
||
mime="application/json",
|
||
key="dl_scope_json",
|
||
)
|
||
if close_series:
|
||
st.write("收盘价时间序列 (最近窗口)：")
|
||
st.json(close_series)
|
||
try:
|
||
import csv
|
||
import io
|
||
|
||
buf = io.StringIO()
|
||
writer = csv.writer(buf)
|
||
writer.writerow(["trade_date", "close"])
|
||
for dt_val, val in close_series:
|
||
writer.writerow([dt_val, val])
|
||
st.download_button(
|
||
"下载收盘价(CSV)",
|
||
data=buf.getvalue(),
|
||
file_name=f"{ts_code}_{trade_date}_close_series.csv",
|
||
mime="text/csv",
|
||
key="dl_close_csv",
|
||
)
|
||
except Exception: # noqa: BLE001
|
||
pass
|
||
if turnover_series:
|
||
st.write("换手率时间序列 (最近窗口)：")
|
||
st.json(turnover_series)
|
||
dept_sup = global_info.get("department_supplements") or {}
|
||
dept_dialogue = global_info.get("department_dialogue") or {}
|
||
dept_telemetry = global_info.get("department_telemetry") or {}
|
||
if dept_sup or dept_dialogue:
|
||
with st.expander("部门补数与对话记录", expanded=False):
|
||
if dept_sup:
|
||
st.write("补充数据：")
|
||
st.json(dept_sup)
|
||
if dept_dialogue:
|
||
st.write("对话片段：")
|
||
st.json(dept_dialogue)
|
||
if dept_telemetry:
|
||
with st.expander("部门 LLM 元数据", expanded=False):
|
||
st.json(dept_telemetry)
|
||
else:
|
||
st.info("暂未写入全局策略摘要。")
|
||
|
||
st.subheader("部门意见")
|
||
if dept_records:
|
||
keyword = st.text_input("筛选摘要/信号关键词", value="")
|
||
filtered = dept_records
|
||
if keyword.strip():
|
||
kw = keyword.strip()
|
||
filtered = [
|
||
item
|
||
for item in dept_records
|
||
if kw in str(item.get("摘要", "")) or kw in str(item.get("核心信号", ""))
|
||
]
|
||
min_conf = st.slider("最低信心过滤", 0.0, 1.0, 0.0, 0.05)
|
||
sort_col = st.selectbox("排序列", ["信心", "权重"], index=0)
|
||
filtered = [row for row in filtered if float(row.get("信心", 0.0)) >= min_conf]
|
||
filtered = sorted(filtered, key=lambda r: float(r.get(sort_col, 0.0)), reverse=True)
|
||
dept_df = pd.DataFrame(filtered)
|
||
st.dataframe(dept_df, width='stretch', hide_index=True)
|
||
try:
|
||
st.download_button(
|
||
"下载部门意见(CSV)",
|
||
data=dept_df.to_csv(index=False),
|
||
file_name=f"{trade_date}_{ts_code}_departments.csv",
|
||
mime="text/csv",
|
||
key="dl_dept_csv",
|
||
)
|
||
except Exception: # noqa: BLE001
|
||
pass
|
||
for code, details in dept_details.items():
|
||
with st.expander(f"{code} 补充详情", expanded=False):
|
||
supplements = details.get("supplements", [])
|
||
dialogue = details.get("dialogue", [])
|
||
if supplements:
|
||
st.write("补充数据：")
|
||
st.json(supplements)
|
||
else:
|
||
st.caption("无补充数据请求。")
|
||
if dialogue:
|
||
st.write("对话记录：")
|
||
for idx, line in enumerate(dialogue, start=1):
|
||
st.markdown(f"**回合 {idx}:** {line}")
|
||
else:
|
||
st.caption("无额外对话。")
|
||
telemetry = details.get("telemetry") or {}
|
||
if telemetry:
|
||
st.write("LLM 元数据：")
|
||
st.json(telemetry)
|
||
else:
|
||
st.info("暂无部门记录。")
|
||
|
||
st.subheader("代理评分")
|
||
if agent_records:
|
||
sort_agent_by = st.selectbox(
|
||
"代理排序",
|
||
["权重", "SELL", "HOLD", "BUY_S", "BUY_M", "BUY_L"],
|
||
index=1,
|
||
)
|
||
agent_df = pd.DataFrame(agent_records)
|
||
if sort_agent_by in agent_df.columns:
|
||
agent_df = agent_df.sort_values(sort_agent_by, ascending=False)
|
||
st.dataframe(agent_df, width='stretch', hide_index=True)
|
||
try:
|
||
st.download_button(
|
||
"下载代理评分(CSV)",
|
||
data=agent_df.to_csv(index=False),
|
||
file_name=f"{trade_date}_{ts_code}_agents.csv",
|
||
mime="text/csv",
|
||
key="dl_agent_csv",
|
||
)
|
||
except Exception: # noqa: BLE001
|
||
pass
|
||
else:
|
||
st.info("暂无基础代理评分。")
|
||
|
||
st.divider()
|
||
st.subheader("相关新闻")
|
||
try:
|
||
with db_session(read_only=True) as conn:
|
||
try:
|
||
trade_date_obj = date.fromisoformat(str(trade_date))
|
||
except Exception:
|
||
try:
|
||
trade_date_obj = datetime.strptime(str(trade_date), "%Y%m%d").date()
|
||
except Exception:
|
||
trade_date_obj = date.today() - timedelta(days=7)
|
||
|
||
news_query = """
|
||
SELECT id, title, source, pub_time, sentiment, heat, entities
|
||
FROM news
|
||
WHERE ts_code = ? AND pub_time >= ?
|
||
ORDER BY pub_time DESC
|
||
LIMIT 10
|
||
"""
|
||
seven_days_ago = (trade_date_obj - timedelta(days=7)).strftime("%Y-%m-%d")
|
||
news_rows = conn.execute(news_query, (ts_code, seven_days_ago)).fetchall()
|
||
|
||
if news_rows:
|
||
news_data = []
|
||
for row in news_rows:
|
||
entities_info = {}
|
||
try:
|
||
if row["entities"]:
|
||
entities_info = json.loads(row["entities"])
|
||
except (json.JSONDecodeError, TypeError):
|
||
pass
|
||
|
||
news_item = {
|
||
"标题": row["title"],
|
||
"来源": row["source"],
|
||
"发布时间": row["pub_time"],
|
||
"情感指数": f"{row['sentiment']:.2f}" if row["sentiment"] is not None else "-",
|
||
"热度评分": f"{row['heat']:.2f}" if row["heat"] is not None else "-",
|
||
}
|
||
|
||
industries = entities_info.get("industries", [])
|
||
if industries:
|
||
news_item["相关行业"] = "、".join(industries[:3])
|
||
|
||
news_data.append(news_item)
|
||
|
||
news_df = pd.DataFrame(news_data)
|
||
for col in news_df.columns:
|
||
news_df[col] = news_df[col].astype(str)
|
||
st.dataframe(news_df, width='stretch', hide_index=True)
|
||
|
||
st.write("详细新闻内容：")
|
||
for idx, row in enumerate(news_rows):
|
||
with st.expander(f"{idx+1}. {row['title']}", expanded=False):
|
||
st.write(f"**来源：** {row['source']}")
|
||
st.write(f"**发布时间：** {row['pub_time']}")
|
||
|
||
entities_info = {}
|
||
try:
|
||
if row["entities"]:
|
||
entities_info = json.loads(row["entities"])
|
||
except (json.JSONDecodeError, TypeError):
|
||
pass
|
||
|
||
sentiment_display = f"{row['sentiment']:.2f}" if row["sentiment"] is not None else "-"
|
||
heat_display = f"{row['heat']:.2f}" if row["heat"] is not None else "-"
|
||
st.write(f"**情感指数：** {sentiment_display} | **热度评分：** {heat_display}")
|
||
|
||
industries = entities_info.get("industries", [])
|
||
if industries:
|
||
st.write(f"**相关行业：** {'、'.join(industries)}")
|
||
|
||
important_keywords = entities_info.get("important_keywords", [])
|
||
if important_keywords:
|
||
st.write(f"**重要关键词：** {'、'.join(important_keywords)}")
|
||
|
||
url = entities_info.get("source_url", "")
|
||
if url:
|
||
st.markdown(f"[查看原文]({url})", unsafe_allow_html=True)
|
||
else:
|
||
st.info(f"近7天内暂无关于 {ts_code} 的新闻。")
|
||
except Exception as exc: # noqa: BLE001
|
||
LOGGER.exception("获取新闻数据失败", extra=LOG_EXTRA)
|
||
st.error(f"获取新闻数据时发生错误：{exc}")
|
||
|
||
st.divider()
|
||
st.info("投资池与仓位概览已移至单独页面。请在侧边或页面导航中选择“投资池/仓位”以查看详细信息。")
|
||
|
||
st.divider()
|
||
st.subheader("策略重评估")
|
||
st.caption("对当前选中的交易日与标的，立即触发一次策略评估并回写 agent_utils。")
|
||
cols_re = st.columns([1, 1])
|
||
if cols_re[0].button("对该标的重评估", key="reevaluate_current_symbol"):
|
||
with st.spinner("正在重评估..."):
|
||
try:
|
||
trade_date_obj: Optional[date] = None
|
||
try:
|
||
trade_date_obj = date.fromisoformat(str(trade_date))
|
||
except Exception:
|
||
try:
|
||
trade_date_obj = datetime.strptime(str(trade_date), "%Y%m%d").date()
|
||
except Exception:
|
||
pass
|
||
if trade_date_obj is None:
|
||
raise ValueError(f"无法解析交易日：{trade_date}")
|
||
with db_session(read_only=True) as conn:
|
||
before_rows = conn.execute(
|
||
"""
|
||
SELECT agent, action, utils FROM agent_utils
|
||
WHERE trade_date = ? AND ts_code = ?
|
||
""",
|
||
(trade_date, ts_code),
|
||
).fetchall()
|
||
before_map = {row["agent"]: (row["action"], row["utils"]) for row in before_rows}
|
||
cfg = BtConfig(
|
||
id="reeval_ui",
|
||
name="UI Re-evaluation",
|
||
start_date=trade_date_obj,
|
||
end_date=trade_date_obj,
|
||
universe=[ts_code],
|
||
params={},
|
||
)
|
||
engine = BacktestEngine(cfg)
|
||
state = PortfolioState()
|
||
engine.simulate_day(trade_date_obj, state)
|
||
with db_session(read_only=True) as conn:
|
||
after_rows = conn.execute(
|
||
"""
|
||
SELECT agent, action, utils FROM agent_utils
|
||
WHERE trade_date = ? AND ts_code = ?
|
||
""",
|
||
(trade_date, ts_code),
|
||
).fetchall()
|
||
changes = []
|
||
for row in after_rows:
|
||
agent = row["agent"]
|
||
new_action = row["action"]
|
||
old_action, _old_utils = before_map.get(agent, (None, None))
|
||
if new_action != old_action:
|
||
changes.append({"代理": agent, "原动作": old_action, "新动作": new_action})
|
||
if changes:
|
||
st.success("重评估完成，检测到动作变更：")
|
||
st.dataframe(pd.DataFrame(changes), hide_index=True, width='stretch')
|
||
else:
|
||
st.success("重评估完成，无动作变更。")
|
||
st.rerun()
|
||
except Exception as exc: # noqa: BLE001
|
||
LOGGER.exception("重评估失败", extra=LOG_EXTRA)
|
||
st.error(f"重评估失败：{exc}")
|
||
if cols_re[1].button("批量重评估（所选）", key="reevaluate_batch", disabled=not batch_symbols):
|
||
with st.spinner("批量重评估执行中..."):
|
||
try:
|
||
trade_date_obj: Optional[date] = None
|
||
try:
|
||
trade_date_obj = date.fromisoformat(str(trade_date))
|
||
except Exception:
|
||
try:
|
||
trade_date_obj = datetime.strptime(str(trade_date), "%Y%m%d").date()
|
||
except Exception:
|
||
pass
|
||
if trade_date_obj is None:
|
||
raise ValueError(f"无法解析交易日：{trade_date}")
|
||
progress = st.progress(0.0)
|
||
changes_all: List[Dict[str, object]] = []
|
||
for idx, code in enumerate(batch_symbols, start=1):
|
||
with db_session(read_only=True) as conn:
|
||
before_rows = conn.execute(
|
||
"SELECT agent, action FROM agent_utils WHERE trade_date = ? AND ts_code = ?",
|
||
(trade_date, code),
|
||
).fetchall()
|
||
before_map = {row["agent"]: row["action"] for row in before_rows}
|
||
cfg = BtConfig(
|
||
id="reeval_ui_batch",
|
||
name="UI Batch Re-eval",
|
||
start_date=trade_date_obj,
|
||
end_date=trade_date_obj,
|
||
universe=[code],
|
||
params={},
|
||
)
|
||
engine = BacktestEngine(cfg)
|
||
state = PortfolioState()
|
||
engine.simulate_day(trade_date_obj, state)
|
||
with db_session(read_only=True) as conn:
|
||
after_rows = conn.execute(
|
||
"SELECT agent, action FROM agent_utils WHERE trade_date = ? AND ts_code = ?",
|
||
(trade_date, code),
|
||
).fetchall()
|
||
for row in after_rows:
|
||
agent = row["agent"]
|
||
new_action = row["action"]
|
||
old_action = before_map.get(agent)
|
||
if new_action != old_action:
|
||
changes_all.append({"代码": code, "代理": agent, "原动作": old_action, "新动作": new_action})
|
||
progress.progress(idx / max(1, len(batch_symbols)))
|
||
st.success("批量重评估完成。")
|
||
if changes_all:
|
||
st.dataframe(pd.DataFrame(changes_all), hide_index=True, width='stretch')
|
||
st.rerun()
|
||
except Exception as exc: # noqa: BLE001
|
||
LOGGER.exception("批量重评估失败", extra=LOG_EXTRA)
|
||
st.error(f"批量重评估失败：{exc}")
|