dekun
919ac70bc2
Saving SIM_DEFAULT_MODE also switches runtime trading.mode immediately. Co-authored-by: Cursor <cursoragent@cursor.com>
81 lines
2.1 KiB
Python
81 lines
2.1 KiB
Python
"""交易模式: sim | live, 持久化到 app_runtime_settings."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
from typing import Callable
|
|
|
|
from lib.instance.runtime_settings_lib import runtime_get, runtime_set, with_db
|
|
|
|
TRADING_MODE_KEY = "trading.mode"
|
|
MODE_SIM = "sim"
|
|
MODE_LIVE = "live"
|
|
VALID_MODES = (MODE_SIM, MODE_LIVE)
|
|
|
|
|
|
def default_trading_mode() -> str:
|
|
raw = (os.getenv("SIM_DEFAULT_MODE") or "").strip().lower()
|
|
if raw in VALID_MODES:
|
|
return raw
|
|
return MODE_SIM
|
|
|
|
|
|
def peek_persisted_trading_mode(db_path: str | None = None) -> str | None:
|
|
"""只读 SQLite 中的 trading.mode(供 env 配置页展示当前生效值)."""
|
|
import sqlite3
|
|
|
|
path = (db_path or os.getenv("DB_PATH") or "crypto.db").strip()
|
|
if not path:
|
|
return None
|
|
if not os.path.isabs(path):
|
|
path = os.path.abspath(path)
|
|
if not os.path.isfile(path):
|
|
return None
|
|
try:
|
|
conn = sqlite3.connect(path, timeout=2)
|
|
try:
|
|
row = conn.execute(
|
|
"SELECT value FROM app_runtime_settings WHERE key=?",
|
|
(TRADING_MODE_KEY,),
|
|
).fetchone()
|
|
finally:
|
|
conn.close()
|
|
except Exception:
|
|
return None
|
|
if not row:
|
|
return None
|
|
m = str(row[0] or "").strip().lower()
|
|
return m if m in VALID_MODES else None
|
|
|
|
|
|
def normalize_mode(mode: str | None) -> str:
|
|
m = (mode or "").strip().lower()
|
|
if m in VALID_MODES:
|
|
return m
|
|
raise ValueError("mode 须为 sim 或 live")
|
|
|
|
|
|
def get_trading_mode(get_db: Callable) -> str:
|
|
def _read(conn):
|
|
v = runtime_get(conn, TRADING_MODE_KEY)
|
|
if v is None or str(v).strip() == "":
|
|
return default_trading_mode()
|
|
m = str(v).strip().lower()
|
|
return m if m in VALID_MODES else default_trading_mode()
|
|
|
|
return with_db(get_db, _read)
|
|
|
|
|
|
def set_trading_mode(get_db: Callable, mode: str) -> str:
|
|
m = normalize_mode(mode)
|
|
|
|
def _write(conn):
|
|
runtime_set(conn, TRADING_MODE_KEY, m)
|
|
return m
|
|
|
|
return with_db(get_db, _write)
|
|
|
|
|
|
def is_sim_mode(get_db: Callable) -> bool:
|
|
return get_trading_mode(get_db) == MODE_SIM
|