refactor: establish standalone application boundary
This commit is contained in:
leefer
@@ -0,0 +1,220 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import secrets
|
||||
from datetime import date
|
||||
from typing import Any
|
||||
|
||||
from backend.bootstrap.config import normalize_date
|
||||
from backend.features.heaven.agent import HeavenAgentError, interpret_heaven
|
||||
from backend.features.heaven.engine import (
|
||||
build_five_phase_field,
|
||||
hexagram_from_lines,
|
||||
)
|
||||
from backend.features.market import MarketServiceMixin
|
||||
|
||||
|
||||
class HeavenReadingMixin:
|
||||
def heaven_personal(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||
trade_date = normalize_date(str(payload.get("trade_date") or date.today().isoformat()))
|
||||
field = build_five_phase_field(
|
||||
trade_date,
|
||||
self.database.list_sector_phase_overrides(),
|
||||
)
|
||||
personal = self.account_personal_field(trade_date, field, public=True)
|
||||
if not personal:
|
||||
raise ValueError("请先在账号设置中保存个人命理资料。")
|
||||
return personal
|
||||
|
||||
def heaven_hexagram(self, raw_lines: Any) -> dict[str, Any]:
|
||||
if not isinstance(raw_lines, list):
|
||||
raise ValueError("六爻起卦结果格式不正确。")
|
||||
try:
|
||||
lines = [int(value) for value in raw_lines]
|
||||
except (TypeError, ValueError) as exc:
|
||||
raise ValueError("六爻必须由六、七、八、九组成。") from exc
|
||||
return hexagram_from_lines(lines)
|
||||
|
||||
def heaven_readings(
|
||||
self, mode: str, context_date: str = "", limit: int = 100
|
||||
) -> dict[str, Any]:
|
||||
mode = str(mode or "").strip()
|
||||
if mode not in {"trend", "fortune", "heart"}:
|
||||
raise ValueError("解读记录类型不正确。")
|
||||
normalized_date = normalize_date(context_date) if context_date else ""
|
||||
return {
|
||||
"mode": mode,
|
||||
"items": self.database.list_heaven_readings(
|
||||
self.current_user_id, mode, normalized_date, limit
|
||||
),
|
||||
}
|
||||
|
||||
@staticmethod
|
||||
def _heaven_reading_identity(
|
||||
mode: str, context_date: str, context: dict[str, Any]
|
||||
) -> tuple[str, str]:
|
||||
display_date = MarketServiceMixin._display_compact_date(context_date)
|
||||
if mode == "trend":
|
||||
stock = (context.get("selected_focus") or {}).get("stock") or {}
|
||||
code = str(stock.get("code") or "").strip()
|
||||
name = str(stock.get("name") or "").strip()
|
||||
hexagram = context.get("hexagram") or {}
|
||||
transformed = hexagram.get("transformed") or {}
|
||||
subject = " ".join(item for item in (code, name) if item) or "观势"
|
||||
detail = f"{display_date} · {hexagram.get('name') or '--'} → {transformed.get('name') or '--'}"
|
||||
return subject, detail
|
||||
if mode == "fortune":
|
||||
field = context.get("five_phase_field") or {}
|
||||
pillars = field.get("pillars") or {}
|
||||
dominant = (field.get("balance") or [{}])[0]
|
||||
subject = f"{display_date} 观气"
|
||||
detail = (
|
||||
f"{pillars.get('year') or '--'}年 · {pillars.get('month') or '--'}月 · "
|
||||
f"{pillars.get('day') or '--'}日 · {dominant.get('element') or '--'}气偏显"
|
||||
)
|
||||
return subject, detail
|
||||
hexagram = context.get("hexagram") or {}
|
||||
transformed = hexagram.get("transformed") or {}
|
||||
return (
|
||||
f"{display_date} 观心",
|
||||
f"{hexagram.get('name') or '--'} → {transformed.get('name') or '--'}",
|
||||
)
|
||||
|
||||
def heaven_interpret(self, payload: dict[str, Any]) -> dict[str, Any]:
|
||||
mode = str(payload.get("mode") or "").strip()
|
||||
if mode not in {"trend", "fortune", "heart"}:
|
||||
raise ValueError("问天解读模式不正确。")
|
||||
trade_date = normalize_date(str(payload.get("trade_date") or date.today().isoformat()))
|
||||
if mode == "fortune":
|
||||
existing = self.database.latest_heaven_reading(
|
||||
self.current_user_id, "fortune", trade_date
|
||||
)
|
||||
if self._legacy_truncated_heaven_reading(existing):
|
||||
self.database.delete_heaven_reading(
|
||||
self.current_user_id, int(existing["id"])
|
||||
)
|
||||
existing = None
|
||||
if existing:
|
||||
return {
|
||||
"answer": existing["answer"],
|
||||
"mode": mode,
|
||||
"compiler": "stored",
|
||||
"notice": "",
|
||||
"reading": existing,
|
||||
"reused": True,
|
||||
}
|
||||
if mode in {"trend", "fortune"}:
|
||||
setup = self.heaven_setup(
|
||||
trade_date,
|
||||
str(payload.get("sector") or ""),
|
||||
str(payload.get("stock_code") or ""),
|
||||
payload.get("manual_data"),
|
||||
)
|
||||
if mode == "trend":
|
||||
chart = setup["chart"]
|
||||
if not chart.get("available"):
|
||||
issues = "；".join((chart.get("quality") or {}).get("issues") or [])
|
||||
raise ValueError(f"观势数据未通过六爻校验，暂不解势：{issues}")
|
||||
hexagram_context = json.loads(json.dumps(chart["hexagram"], ensure_ascii=False))
|
||||
for line in hexagram_context.get("lines", []):
|
||||
line.pop("evidence", None)
|
||||
line.pop("score", None)
|
||||
line.pop("talent", None)
|
||||
line.pop("layer", None)
|
||||
line.pop("role", None)
|
||||
if not line.get("moving"):
|
||||
line.pop("text", None)
|
||||
line.pop("image", None)
|
||||
line.pop("line_name", None)
|
||||
context = {
|
||||
"data_trade_date": setup["trade_date"],
|
||||
"selected_focus": {
|
||||
"sector": chart.get("sector") or "",
|
||||
"stock": chart.get("stock") or {},
|
||||
},
|
||||
"hexagram": hexagram_context,
|
||||
"movement": chart.get("movement") or {},
|
||||
}
|
||||
else:
|
||||
personal_profile = self.account_personal_field(
|
||||
setup["calendar_date"],
|
||||
setup["field"],
|
||||
public=False,
|
||||
)
|
||||
fortune_field = json.loads(json.dumps(setup["field"], ensure_ascii=False))
|
||||
catalog = fortune_field.pop("sector_catalog", [])
|
||||
dominant_elements = {
|
||||
item.get("element") for item in fortune_field.get("balance", [])[:2]
|
||||
}
|
||||
fortune_field["industry_affinity"] = [
|
||||
{
|
||||
"element": group.get("element"),
|
||||
"examples": [
|
||||
item.get("name")
|
||||
for item in group.get("industries", [])[:8]
|
||||
if item.get("name")
|
||||
],
|
||||
}
|
||||
for group in catalog
|
||||
if group.get("element") in dominant_elements
|
||||
]
|
||||
context = {
|
||||
"calendar_date": setup["calendar_date"],
|
||||
"five_phase_field": fortune_field,
|
||||
"personal_profile": personal_profile,
|
||||
}
|
||||
context_date = setup["calendar_date"]
|
||||
if mode == "trend":
|
||||
context_date = setup["trade_date"]
|
||||
else:
|
||||
context = {
|
||||
"hexagram": self.heaven_hexagram(payload.get("lines")),
|
||||
"ritual": "用户已完成30秒静心、六次三枚铜钱起卦，并在心中察看第一念。问题未输入。",
|
||||
}
|
||||
context_date = trade_date
|
||||
result, compiler = self._call_heaven_agent(mode, context)
|
||||
subject, subject_detail = self._heaven_reading_identity(
|
||||
mode, context_date, context
|
||||
)
|
||||
dedupe_key = (
|
||||
f"fortune:{context_date}"
|
||||
if mode == "fortune"
|
||||
else f"{mode}:{context_date}:{secrets.token_urlsafe(12)}"
|
||||
)
|
||||
reading = self.database.save_heaven_reading(
|
||||
self.current_user_id,
|
||||
mode,
|
||||
context_date,
|
||||
subject,
|
||||
subject_detail,
|
||||
str(result.get("answer") or ""),
|
||||
context,
|
||||
dedupe_key,
|
||||
)
|
||||
return {
|
||||
**result,
|
||||
"mode": mode,
|
||||
"compiler": compiler,
|
||||
"notice": "智能解读已自动切换可用服务。" if compiler == "fallback" else "",
|
||||
"reading": reading,
|
||||
"reused": False,
|
||||
}
|
||||
|
||||
@staticmethod
|
||||
def _legacy_truncated_heaven_reading(reading: dict[str, Any] | None) -> bool:
|
||||
return bool(reading and str(reading.get("answer") or "").rstrip().endswith("……"))
|
||||
|
||||
def _call_heaven_agent(self, mode: str, context: dict[str, Any]) -> tuple[dict[str, Any], str]:
|
||||
result = self.llm_gateway.call(
|
||||
f"heaven_{mode}",
|
||||
f"heaven-{mode}-v1",
|
||||
lambda profile: interpret_heaven(
|
||||
mode,
|
||||
context,
|
||||
profile.api_key,
|
||||
profile.base_url,
|
||||
profile.model,
|
||||
),
|
||||
(HeavenAgentError,),
|
||||
)
|
||||
return result.value, result.role
|
||||
Reference in New Issue
Block a user