126 lines
4.4 KiB
Python
126 lines
4.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
连通性与表结构自检 (实机, 需 .env 真实连接串)
|
|
==============================================
|
|
运行: docker compose run --rm pms-web python scripts/check_db.py
|
|
|
|
检查项:
|
|
1. 三个库连通性 (153 代理 / 因子库 / 指数库)
|
|
2. pms_* 十张表是否存在与当前行数
|
|
3. 下游三表可读性 + 持仓数量列探测结果 (回填 QMT_INTERFACE_REQUIREMENTS A1/D1 用)
|
|
4. 行情 Redis (db13) 连通性与样本键
|
|
5. 运行参数表当前生效值
|
|
全部通过退出码 0; 任一 FAIL 退出码 1 (WARN 不影响退出码)。
|
|
"""
|
|
import os
|
|
import sys
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
PMS_TABLES = ["pms_command", "pms_plan", "pms_position", "pms_lot", "pms_instruction",
|
|
"pms_proposal", "pms_action_ledger", "pms_daily_report", "pms_industry_map",
|
|
"pms_runtime_param"]
|
|
DOWNSTREAM = ["trading_position", "trading_order", "trading_buy_plan"]
|
|
|
|
FAILED, WARNED = [], []
|
|
|
|
|
|
def line(tag, msg):
|
|
print(f" {tag:<6}{msg}")
|
|
|
|
|
|
def fail(msg):
|
|
FAILED.append(msg)
|
|
line("FAIL", msg)
|
|
|
|
|
|
def warn(msg):
|
|
WARNED.append(msg)
|
|
line("WARN", msg)
|
|
|
|
|
|
def main():
|
|
from app.db import session as dbs
|
|
|
|
print("\n[1] 数据库连通性")
|
|
for name, label in (("proxy", "153 代理 (pms_* / trading_*)"),
|
|
("factor", "因子分表库"), ("index", "大盘指数库")):
|
|
r = dbs.ping(name)
|
|
if r["ok"]:
|
|
line("OK", f"{label}")
|
|
elif name == "proxy":
|
|
fail(f"{label} 连接失败: {r['error']}")
|
|
else:
|
|
warn(f"{label} 连接失败 (非致命): {r['error']}")
|
|
|
|
print("\n[2] PMS 自有表")
|
|
for t in PMS_TABLES:
|
|
try:
|
|
r = dbs.fetch_one(f"SELECT COUNT(*) AS n FROM {t}")
|
|
line("OK", f"{t:<20} {r['n']} 行")
|
|
except Exception as e:
|
|
fail(f"{t} 不可用: {type(e).__name__}: {e}")
|
|
|
|
print("\n[3] 下游只读表")
|
|
from app.repo import downstream_repo
|
|
for t in DOWNSTREAM:
|
|
try:
|
|
cols = downstream_repo.describe(t)
|
|
line("OK", f"{t:<20} {len(cols)} 列: "
|
|
f"{', '.join(str(c.get('Field')) for c in cols[:10])}"
|
|
f"{' ...' if len(cols) > 10 else ''}")
|
|
except Exception as e:
|
|
warn(f"{t} 读取失败: {type(e).__name__}: {e}")
|
|
try:
|
|
ds = downstream_repo.fetch_positions()
|
|
line("OK" if ds["columns"].get("qty") else "WARN",
|
|
f"持仓列探测: {ds['columns']} (共 {ds['raw_count']} 行)")
|
|
if ds["raw_count"] and not ds["columns"].get("qty"):
|
|
warn("未识别出持仓数量列 —— 请按 QMT 需求清单 A1/D1 取得 DDL 后补 QTY_CANDIDATES")
|
|
except Exception as e:
|
|
warn(f"持仓快照读取失败: {e}")
|
|
|
|
print("\n[4] 行情 Redis (db13)")
|
|
try:
|
|
from app.services import market
|
|
pos = []
|
|
try:
|
|
from app.repo import pms_repo
|
|
pos = [p["ts_code"] for p in pms_repo.list_positions(only_open=True)][:3]
|
|
except Exception:
|
|
pass
|
|
probe = pos or ["600000.SH"]
|
|
got = {c: market.get_price(c) for c in probe}
|
|
if any(v for v in got.values()):
|
|
line("OK", f"实时价样本: {got}")
|
|
else:
|
|
warn(f"取不到实时价样本 {got} (非交易时段属正常)")
|
|
except Exception as e:
|
|
warn(f"Redis 行情不可用: {type(e).__name__}: {e}")
|
|
|
|
print("\n[5] 运行参数 (表值优先于 settings 初值)")
|
|
try:
|
|
from app.services import param_store
|
|
snap = param_store.snapshot()
|
|
from_table = [p for p in snap["params"] if p["source"] == "table"]
|
|
line("OK", f"可调参数 {len(snap['params'])} 项, 其中页面已改写 {len(from_table)} 项")
|
|
for k in ("PMS_TOTAL_SCALE", "PMS_PORTFOLIO_CAP", "PMS_STOCK_CAP", "PMS_AUTONOMY",
|
|
"PMS_SECTOR_SOURCE"):
|
|
line("", f"{k:<26} = {param_store.get(k)}")
|
|
if snap.get("source_error"):
|
|
warn(f"参数表读取异常: {snap['source_error']}")
|
|
except Exception as e:
|
|
fail(f"参数中心不可用: {type(e).__name__}: {e}")
|
|
|
|
print("\n" + "-" * 62)
|
|
if FAILED:
|
|
print(f"FAILED: {len(FAILED)} 项致命问题, {len(WARNED)} 项告警")
|
|
for m in FAILED:
|
|
print(" - " + m)
|
|
sys.exit(1)
|
|
print(f"ALL OK ({len(WARNED)} 项告警)" if WARNED else "ALL OK")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|