93 lines
4.0 KiB
Python
93 lines
4.0 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""promote_reassess_backfill.py — 新提拔自选的重评补推（老莫：未就绪的要补推）
|
||
promote 结束后 detach 启动：
|
||
1. 读 promote 记录的新提拔候选（holding_strategies 里 full_analysis 为空/短的 active 自选）
|
||
2. 对未就绪的补跑 per_stock_reassess（300s 超时）
|
||
3. 收集就绪的完整重评，通过 XMPP 补推给老莫
|
||
"""
|
||
import sys, os, json, time, subprocess, sqlite3, urllib.request
|
||
from datetime import datetime
|
||
|
||
DB = "/home/hmo/MoFin/data/mofin.db"
|
||
PY = "/home/hmo/MoFin/venv/bin/python"
|
||
REASSESS = "/home/hmo/MoFin/deploy/profile-scripts/per_stock_reassess.py"
|
||
XMPP = "http://127.0.0.1:5805/"
|
||
TIMEOUT = 300
|
||
MAX_RETRY = 2
|
||
|
||
def xmpp_push(body):
|
||
try:
|
||
req = urllib.request.Request(
|
||
XMPP, data=json.dumps({"to": "hmo@yoin.fun", "body": body, "type": "chat"}).encode(),
|
||
headers={"Content-Type": "application/json"})
|
||
urllib.request.urlopen(req, timeout=10)
|
||
return True
|
||
except Exception as e:
|
||
print(f" ⚠️ XMPP 补推失败: {e}", flush=True)
|
||
return False
|
||
|
||
def main(wait_sec=90):
|
||
# 给首次重评完成时间（promote 触发的 REASSESS 可能还在跑）
|
||
print(f"[BACKFILL] 等待 {wait_sec}s 后检查未就绪重评...", flush=True)
|
||
time.sleep(wait_sec)
|
||
conn = sqlite3.connect(DB, timeout=30)
|
||
conn.execute("PRAGMA busy_timeout=30000")
|
||
# 最近 30 分钟新提拔的自选，full_analysis 为空/短（未就绪）
|
||
rows = conn.execute("""
|
||
SELECT code, name, strategy_name, rr_ratio, LENGTH(full_analysis) as fa_len, created_at
|
||
FROM holding_strategies
|
||
WHERE status='active' AND decision_type='自选策略'
|
||
AND created_at >= datetime('now', 'localtime', '-30 minutes')
|
||
AND (full_analysis IS NULL OR LENGTH(full_analysis) < 50)
|
||
ORDER BY created_at DESC
|
||
""").fetchall()
|
||
conn.close()
|
||
if not rows:
|
||
print("[BACKFILL] 无未就绪重评", flush=True)
|
||
return
|
||
print(f"[BACKFILL] 发现 {len(rows)} 只未就绪重评，逐一补跑...", flush=True)
|
||
backfilled = []
|
||
for code, name, strat, rr, fa_len, created in rows:
|
||
ok = False
|
||
for attempt in range(MAX_RETRY):
|
||
try:
|
||
r = subprocess.run([PY, REASSESS, code], capture_output=True, text=True, timeout=TIMEOUT)
|
||
if r.returncode == 0:
|
||
ok = True
|
||
break
|
||
else:
|
||
print(f" ⚠️ {code} 重评失败(尝试{attempt+1}): {r.stderr.strip()[:100]}", flush=True)
|
||
except Exception as e:
|
||
print(f" ⚠️ {code} 重评异常(尝试{attempt+1}): {e}", flush=True)
|
||
time.sleep(5)
|
||
if ok:
|
||
backfilled.append((code, name, strat, rr))
|
||
if not backfilled:
|
||
print("[BACKFILL] 补跑后仍无就绪报告，跳过补推", flush=True)
|
||
return
|
||
# 收集就绪的完整报告
|
||
conn = sqlite3.connect(DB, timeout=30)
|
||
codes = ",".join("?" * len(backfilled))
|
||
ready = conn.execute(
|
||
f"SELECT code, name, strategy_name, rr_ratio, full_analysis FROM holding_strategies "
|
||
f"WHERE status='active' AND code IN ({codes})",
|
||
tuple(c[0] for c in backfilled)).fetchall()
|
||
conn.close()
|
||
body = f"📤 重评报告补推（{len(ready)}只新入自选的重评已就绪）\n"
|
||
for i, (code, name, strat, rr, fa) in enumerate(ready, 1):
|
||
nm = name or code
|
||
body += f"\n━━━ [{i}] {code} {nm} ━━━\n"
|
||
body += f"策略: {strat or 'unknown'} | RR: {rr or 'N/A'}\n"
|
||
if fa and len(str(fa)) > 50:
|
||
body += f"【12维重评】\n{fa}\n"
|
||
else:
|
||
body += "（仍未能生成，已放弃补推）\n"
|
||
if len(body) > 6000:
|
||
body = body[:6000] + "\n…（内容过长已截断）"
|
||
ok = xmpp_push(body)
|
||
print(f"[BACKFILL] 补推{'成功' if ok else '失败'}: {len(ready)} 只", flush=True)
|
||
|
||
if __name__ == "__main__":
|
||
wait = int(sys.argv[1]) if len(sys.argv) > 1 else 90
|
||
main(wait) |