72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
"""策略配置持久化 — 读写用户覆盖值。
|
|
|
|
职责: 将每个策略的用户定制设置（基础参数、策略参数、评分、买卖信号）持久化到 JSON。
|
|
不知道: 引擎、AI、前端、回测。
|
|
存储: data/user_data/strategy_overrides/{strategy_id}.json
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import logging
|
|
from pathlib import Path
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _overrides_dir(data_dir: Path) -> Path:
|
|
d = data_dir / "user_data" / "strategy_overrides"
|
|
d.mkdir(parents=True, exist_ok=True)
|
|
return d
|
|
|
|
|
|
def _path(data_dir: Path, strategy_id: str) -> Path:
|
|
return _overrides_dir(data_dir) / f"{strategy_id}.json"
|
|
|
|
|
|
def load_override(data_dir: Path, strategy_id: str) -> dict:
|
|
"""读取策略的用户覆盖配置，不存在返回空 dict"""
|
|
p = _path(data_dir, strategy_id)
|
|
if not p.exists():
|
|
return {}
|
|
try:
|
|
data = json.loads(p.read_text(encoding="utf-8"))
|
|
# 清理 basic_filter 中值为 None/空的键（避免固化无意义的空值）
|
|
bf = data.get("basic_filter")
|
|
if isinstance(bf, dict):
|
|
cleaned = {k: v for k, v in bf.items() if v is not None}
|
|
if cleaned:
|
|
data["basic_filter"] = cleaned
|
|
else:
|
|
del data["basic_filter"]
|
|
return data
|
|
except Exception as e:
|
|
logger.warning("load override %s failed: %s", strategy_id, e)
|
|
return {}
|
|
|
|
|
|
def save_override(data_dir: Path, strategy_id: str, overrides: dict) -> None:
|
|
"""保存策略的用户覆盖配置（全量覆盖写）"""
|
|
p = _path(data_dir, strategy_id)
|
|
p.parent.mkdir(parents=True, exist_ok=True)
|
|
p.write_text(json.dumps(overrides, ensure_ascii=False, indent=2), encoding="utf-8")
|
|
|
|
|
|
def delete_override(data_dir: Path, strategy_id: str) -> None:
|
|
"""删除策略的用户覆盖配置（重置为默认值）"""
|
|
p = _path(data_dir, strategy_id)
|
|
if p.exists():
|
|
p.unlink()
|
|
|
|
|
|
def list_overrides(data_dir: Path) -> dict[str, dict]:
|
|
"""返回所有策略的覆盖配置 {strategy_id: overrides}"""
|
|
d = _overrides_dir(data_dir)
|
|
result: dict[str, dict] = {}
|
|
for f in d.glob("*.json"):
|
|
try:
|
|
sid = f.stem
|
|
result[sid] = json.loads(f.read_text(encoding="utf-8"))
|
|
except Exception:
|
|
continue
|
|
return result
|