dekun
161 lines
5.6 KiB
Python
161 lines
5.6 KiB
Python
"""AI 日复盘 / 周复盘：附图收集与 journal 文本格式化（四所共用）。"""
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import uuid
|
|
from typing import Any, Callable, List, Mapping, Optional, Sequence
|
|
|
|
from journal_chart_lib import (
|
|
JOURNAL_CHART_ANCHOR_CLOSE,
|
|
JOURNAL_CHART_DEFAULT_LIMIT,
|
|
JOURNAL_CHART_DEFAULT_TF1,
|
|
JOURNAL_CHART_DEFAULT_TF2,
|
|
normalize_chart_timeframe,
|
|
)
|
|
|
|
|
|
def _journal_nz(v: Any, default: str = "无") -> str:
|
|
if v is None:
|
|
return default
|
|
s = str(v).strip()
|
|
return s if s else default
|
|
|
|
|
|
def journal_row_lines_for_ai(
|
|
idx: int,
|
|
row: Mapping[str, Any],
|
|
*,
|
|
include_hold_duration: bool = True,
|
|
) -> str:
|
|
"""把 journal 字段拼成给 AI 的文本；四所日复盘/周复盘共用。"""
|
|
lines = [
|
|
(
|
|
f"{idx}. {_journal_nz(row.get('coin'))} {_journal_nz(row.get('tf'))} "
|
|
f"| 盈亏:{_journal_nz(row.get('pnl'))}U "
|
|
f"| 实际RR:{_journal_nz(row.get('real_rr'))} "
|
|
f"| 预期RR:{_journal_nz(row.get('expect_rr'))}"
|
|
),
|
|
f" 开仓逻辑：{_journal_nz(row.get('entry_reason'))}",
|
|
f" 平仓/离场（交易员自述）：{_journal_nz(row.get('exit_reason'))}",
|
|
]
|
|
if include_hold_duration:
|
|
lines.append(f" 持仓时长：{_journal_nz(row.get('hold_duration'))}")
|
|
ee_bits = [
|
|
_journal_nz(row.get("early_exit")),
|
|
_journal_nz(row.get("early_exit_reason")),
|
|
_journal_nz(row.get("early_exit_trigger")),
|
|
_journal_nz(row.get("early_exit_note")),
|
|
]
|
|
if any(x != "无" for x in ee_bits):
|
|
lines.append(
|
|
" 提前离场记录："
|
|
f"{ee_bits[0]} | 原因:{ee_bits[1]} | 触发:{ee_bits[2]} | 备注:{ee_bits[3]}"
|
|
)
|
|
mood_bits = f"心态标签：{_journal_nz(row.get('mood_issues'))}"
|
|
if row.get("mood_score") is not None:
|
|
mood_bits += f" | 自评心态分：{row.get('mood_score')}"
|
|
lines.append(f" {mood_bits}")
|
|
if _journal_nz(row.get("post_breakeven_stare")) != "无":
|
|
lines.append(f" 保本后盯盘：{_journal_nz(row.get('post_breakeven_stare'))}")
|
|
if _journal_nz(row.get("new_trade_while_occupied")) != "无":
|
|
lines.append(f" 占用时新开仓：{_journal_nz(row.get('new_trade_while_occupied'))}")
|
|
if _journal_nz(row.get("note")) != "无":
|
|
lines.append(f" 备注：{_journal_nz(row.get('note'))}")
|
|
return "\n".join(lines) + "\n"
|
|
|
|
|
|
def collect_images_for_ai_review(
|
|
rows: Sequence,
|
|
upload_folder: str,
|
|
*,
|
|
build_chart_if_missing: Optional[Callable] = None,
|
|
) -> List[str]:
|
|
"""
|
|
收集传给视觉模型的本地图片路径。
|
|
- 优先 journal_entries.image 已存附图；
|
|
- 若无附图且提供 build_chart_if_missing，则临时生成 K 线图。
|
|
"""
|
|
paths: List[str] = []
|
|
seen = set()
|
|
upload_folder = os.path.abspath(upload_folder or "")
|
|
for row in rows or []:
|
|
candidate = None
|
|
try:
|
|
keys = row.keys() if hasattr(row, "keys") else []
|
|
except Exception:
|
|
keys = []
|
|
img = row["image"] if "image" in keys else None
|
|
if img:
|
|
candidate = os.path.join(upload_folder, str(img).strip())
|
|
elif build_chart_if_missing:
|
|
try:
|
|
candidate = build_chart_if_missing(row)
|
|
except Exception:
|
|
candidate = None
|
|
if not candidate:
|
|
continue
|
|
candidate = os.path.abspath(candidate)
|
|
if os.path.isfile(candidate) and candidate not in seen:
|
|
seen.add(candidate)
|
|
paths.append(candidate)
|
|
return paths
|
|
|
|
|
|
def build_journal_ai_chart_path(
|
|
row,
|
|
upload_folder: str,
|
|
*,
|
|
order_chart_enabled: bool,
|
|
normalize_exchange_symbol_fn: Callable[[str], str],
|
|
generate_chart_fn: Callable,
|
|
local_datetime_to_ms_fn: Callable[[str], Optional[int]],
|
|
now_ts_ms_fn: Callable[[], int],
|
|
) -> Optional[str]:
|
|
"""无已存附图时，按复盘记录开平仓时间临时生成 K 线图路径。"""
|
|
if not order_chart_enabled:
|
|
return None
|
|
try:
|
|
keys = row.keys() if hasattr(row, "keys") else []
|
|
except Exception:
|
|
return None
|
|
coin = (row["coin"] if "coin" in keys else "") or ""
|
|
coin = str(coin).strip()
|
|
if not coin:
|
|
return None
|
|
try:
|
|
symbol = normalize_exchange_symbol_fn(coin)
|
|
except Exception:
|
|
return None
|
|
open_dt = row["open_datetime"] if "open_datetime" in keys else ""
|
|
close_dt = row["close_datetime"] if "close_datetime" in keys else ""
|
|
entry_ms = local_datetime_to_ms_fn(open_dt)
|
|
exit_ms = local_datetime_to_ms_fn(close_dt)
|
|
if not entry_ms:
|
|
return None
|
|
row_tf = row["tf"] if "tf" in keys else ""
|
|
tf1 = normalize_chart_timeframe(row_tf) or JOURNAL_CHART_DEFAULT_TF1
|
|
tf2 = JOURNAL_CHART_DEFAULT_TF2 if tf1 != JOURNAL_CHART_DEFAULT_TF2 else "1h"
|
|
row_id = str(row["id"] if "id" in keys else "")[:8] or uuid.uuid4().hex[:8]
|
|
marker = {
|
|
"entry_ts_ms": entry_ms,
|
|
"exit_ts_ms": exit_ms,
|
|
"chart_anchor": JOURNAL_CHART_ANCHOR_CLOSE,
|
|
"now_ts_ms": int(now_ts_ms_fn()),
|
|
}
|
|
fname = f"ai_rev_{row_id}_{uuid.uuid4().hex[:6]}.png"
|
|
saved = generate_chart_fn(
|
|
symbol,
|
|
f"AI复盘 {coin}",
|
|
timeframes=[tf1, tf2],
|
|
limit=JOURNAL_CHART_DEFAULT_LIMIT,
|
|
out_dir=upload_folder,
|
|
filename=fname,
|
|
marker_payload=marker,
|
|
marker_timeframes={tf1, tf2},
|
|
layout="vertical",
|
|
)
|
|
if not saved:
|
|
return None
|
|
path = os.path.join(upload_folder, saved)
|
|
return path if os.path.isfile(path) else None
|