66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
"""inject_xiaoguo_insight.py — 将小果情感分析结果注入 market.json
|
|
|
|
时序：market_watch(15:30) → market_insight(15:35) → xiaoguo_analyst(16:00) → 本脚本(16:05)
|
|
|
|
读取 xiaoguo_insights.json 的情感分析结果，注入到 market.json 的 insights 字段
|
|
"""
|
|
|
|
import json
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
|
|
DATA_DIR = Path(__file__).parent.parent / "data"
|
|
|
|
|
|
def inject():
|
|
market_path = DATA_DIR / "market.json"
|
|
xiaoguo_path = DATA_DIR / "xiaoguo_insights.json"
|
|
|
|
if not xiaoguo_path.exists():
|
|
print("xiaoguo_insights.json 不存在，跳过")
|
|
return
|
|
|
|
with open(xiaoguo_path, "r", encoding="utf-8") as f:
|
|
xiaoguo = json.load(f)
|
|
|
|
analyses = xiaoguo.get("analyses", [])
|
|
if not analyses:
|
|
print("小果分析为空，跳过")
|
|
return
|
|
|
|
with open(market_path, "r", encoding="utf-8") as f:
|
|
market = json.load(f)
|
|
|
|
insights = market.get("insights", [])
|
|
|
|
# 注入情感分析结果
|
|
for a in analyses:
|
|
name = a.get("name", "")
|
|
sentiment = a.get("sentiment", "neutral")
|
|
confidence = a.get("confidence", 0)
|
|
brief = a.get("brief", "")
|
|
sentiment_emoji = {"positive": "", "negative": "", "neutral": ""}.get(
|
|
sentiment, ""
|
|
)
|
|
label = {"positive": "利好", "negative": "利空", "neutral": "中性"}.get(
|
|
sentiment, "中性"
|
|
)
|
|
|
|
if brief:
|
|
insights.append(
|
|
f"[小果]{name}: {label}(置信度{confidence:.0%}) — {brief}"
|
|
)
|
|
|
|
market["insights"] = insights
|
|
market["xiaoguo_timestamp"] = xiaoguo.get("timestamp", "")
|
|
|
|
with open(market_path, "w", encoding="utf-8") as f:
|
|
json.dump(market, f, ensure_ascii=False, indent=2)
|
|
|
|
print(f"注入{len(analyses)}条小果分析到 market.json")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
inject()
|