217 lines
11 KiB
Python
217 lines
11 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""阈值单位口径修正 + 量纲守卫装配（迁移脚本·准备态）
|
||
|
||
依据：财务口径裁定 finance-20260916-cma-caliber-ruling.md v1.0 §四（裁定③）+ §3.3/§3.4
|
||
· 4.1 单位规范：偿债能力比率=**倍**（无量纲）；资产负债率等=%；返利率=%
|
||
· 4.3 具体阈值裁定（贸易型：白酒经销 / IT 经销）
|
||
· 3.3 09-09 假红灯 586/587 按「口径错误」作废并重算
|
||
· 3.4 切换顺序不可颠倒：先单位 → 再期间 → 后调度（本脚本只做第 1 步）
|
||
|
||
默认 dry-run（只读：列改动 + 影响面复算，不写库）。`--apply` 才写（**需审批**）。
|
||
|
||
用法：
|
||
python3 scripts/migrate_kpi_threshold_unit_20260916.py # dry-run
|
||
python3 scripts/migrate_kpi_threshold_unit_20260916.py --apply # 写阈值/单位（需放行）
|
||
python3 scripts/migrate_kpi_threshold_unit_20260916.py --apply --resolve-false-reds
|
||
# 同时作废假红灯 586/587
|
||
说明：
|
||
· 影响面口径 = 用阈值复算「最近可用期间(<= 最新已入库月)」的命中条数，
|
||
分别给「修正前 / 修正后」两组数字（裁定 §4.5 要求：影响面以修正后复算值为准）。
|
||
· 返利率（F_REBATE_RATE）**不改阈值**（量纲一致、判定有效），仅标注口径待确认（裁定 §4.4）。
|
||
"""
|
||
import argparse
|
||
import sys
|
||
import os
|
||
|
||
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
|
||
from sqlalchemy import text # noqa: E402
|
||
from app.database import get_engine # noqa: E402
|
||
from app.utils.dimension_guard import check_dimension # noqa: E402
|
||
from app.utils.threshold_parser import eval_threshold, LEVEL_PRIORITY # noqa: E402
|
||
|
||
ENTITY_ID = 1 # 酣客（白酒经销）；博海（IT）无阈值 KPI 待核
|
||
|
||
#: 迁移计划：kpi_code → 目标 unit / 三档阈值（None = 不改该字段）
|
||
PLAN = {
|
||
"F_CURRENT_RATIO": {"name": "流动比率", "unit": "倍",
|
||
"green": ">=2.0", "yellow": ">=1.0", "red": "<1.0",
|
||
"why": "裁定 §4.3：经验值 2:1；<1 才够紧急（旧 <150 为百分数写法）"},
|
||
"F_QUICK_RATIO": {"name": "速动比率", "unit": "倍",
|
||
"green": ">=1.0", "yellow": ">=0.5", "red": "<0.5",
|
||
"why": "裁定 §4.3：经验值 1:1（旧 <80 为百分数写法）"},
|
||
"F_DEBT_RATIO": {"name": "资产负债率", "unit": "%",
|
||
"green": "<=60", "yellow": ">60", "red": ">70",
|
||
"why": "裁定 §4.3：红 >70 / 黄 60–70 / 绿 ≤60"},
|
||
}
|
||
#: 明确不改、但需标注的项
|
||
UNCHANGED_NOTES = {
|
||
"F_REBATE_RATE": ("返利率", "%", "裁定 §4.4：量纲一致、判定有效，不改阈值；"
|
||
"48.44% 口径本身待确认 → 预警保留并标注「口径待确认」"),
|
||
}
|
||
|
||
#: 假红灯（09-09 单位错配产生）—— 裁定 §3.3：按「口径错误」作废
|
||
FALSE_RED_ALERT_IDS = (586, 587)
|
||
VOID_RESOLUTION = "[口径错误·作废] 阈值与实测值跨量级比较（流动/速动比率单位错配），" \
|
||
"非财务事实；由 finance-20260916-cma-caliber-ruling §3.3 裁定作废，待单位修正后重算。"
|
||
|
||
|
||
def q(sql, params=None):
|
||
with get_engine().connect() as c:
|
||
return [dict(r._mapping) for r in c.execute(text(sql), params or {}).fetchall()]
|
||
|
||
|
||
def exec_sql(sql, params=None):
|
||
with get_engine().begin() as c:
|
||
return c.execute(text(sql), params or {})
|
||
|
||
|
||
def fetch_plan_rows():
|
||
codes = list(PLAN.keys()) + list(UNCHANGED_NOTES.keys())
|
||
ph = ",".join(":%s" % ("c%d" % i) for i in range(len(codes)))
|
||
return q("SELECT id, kpi_code, kpi_name, unit, threshold_green, threshold_yellow, threshold_red, status "
|
||
"FROM kpi_definitions WHERE entity_id=:e AND kpi_code IN (%s) ORDER BY id" % ph,
|
||
dict({"e": ENTITY_ID}, **{("c%d" % i): c for i, c in enumerate(codes)}))
|
||
|
||
|
||
def latest_period():
|
||
r = q("SELECT MAX(period) AS p FROM kpi_values WHERE period REGEXP '^[0-9]{4}-[0-9]{2}$'")
|
||
return r[0]["p"] if r else None
|
||
|
||
|
||
def impact():
|
||
"""影响面复算：最近可用期间上，各 KPI 在「修正前 / 修正后」的命中条数。"""
|
||
p = latest_period()
|
||
rows = q(
|
||
"SELECT d.id, d.kpi_code, d.kpi_name, d.unit, d.threshold_green AS g, "
|
||
"d.threshold_yellow AS y, d.threshold_red AS r, v.actual_value AS actual, v.period "
|
||
"FROM kpi_definitions d JOIN kpi_values v ON v.kpi_id = d.id "
|
||
"WHERE d.entity_id=:e AND d.status='active' AND v.period = :p AND v.actual_value IS NOT NULL "
|
||
"AND (d.threshold_red IS NOT NULL OR d.threshold_yellow IS NOT NULL OR d.threshold_green IS NOT NULL) "
|
||
"ORDER BY d.id", {"e": ENTITY_ID, "p": p})
|
||
|
||
def judge(actual, unit, g, y, r):
|
||
exprs = {"green": g, "yellow": y, "red": r}
|
||
guard_hit = None
|
||
for lv in LEVEL_PRIORITY:
|
||
if exprs[lv]:
|
||
gd = check_dimension(actual, exprs[lv], unit)
|
||
if not gd["ok"]:
|
||
guard_hit = gd
|
||
break
|
||
if guard_hit:
|
||
return "口径错误"
|
||
for lv in LEVEL_PRIORITY:
|
||
if exprs[lv] and eval_threshold(actual, exprs[lv]):
|
||
return lv
|
||
return "info"
|
||
|
||
before, after, detail = {"red": 0, "yellow": 0, "green": 0, "info": 0, "口径错误": 0}, \
|
||
{"red": 0, "yellow": 0, "green": 0, "info": 0, "口径错误": 0}, []
|
||
for x in rows:
|
||
plan = PLAN.get(x["kpi_code"])
|
||
g2 = plan["green"] if plan else x["g"]
|
||
y2 = plan["yellow"] if plan else x["y"]
|
||
r2 = plan["red"] if plan else x["r"]
|
||
u2 = plan["unit"] if plan else x["unit"]
|
||
b = judge(x["actual"], x["unit"], x["g"], x["y"], x["r"])
|
||
a = judge(x["actual"], u2, g2, y2, r2)
|
||
before[b] = before.get(b, 0) + 1
|
||
after[a] = after.get(a, 0) + 1
|
||
if b != a:
|
||
detail.append({"kpi": "%s/%s" % (x["kpi_code"], x["kpi_name"]), "period": x["period"],
|
||
"actual": x["actual"], "unit": x["unit"],
|
||
"before": b, "after": a,
|
||
"thresholds": "%s|%s|%s → %s|%s|%s" % (x["g"], x["y"], x["r"], g2, y2, r2)})
|
||
return p, before, after, detail
|
||
|
||
|
||
def main():
|
||
ap = argparse.ArgumentParser()
|
||
ap.add_argument("--apply", action="store_true", help="真正写库（缺省=dry-run）")
|
||
ap.add_argument("--resolve-false-reds", action="store_true",
|
||
help="同时把假红灯 %s 按口径错误作废（status=resolved）" % (FALSE_RED_ALERT_IDS,))
|
||
args = ap.parse_args()
|
||
|
||
print("=== 阈值单位口径修正 %s | entity_id=%d ===" % ("APPLY" if args.apply else "DRY-RUN", ENTITY_ID))
|
||
rows = fetch_plan_rows()
|
||
if not rows:
|
||
print("!! 未取到目标 KPI 行（kpi_code 不匹配？）")
|
||
return 2
|
||
|
||
print("\n-- 计划改动 --")
|
||
todo = []
|
||
for x in rows:
|
||
plan = PLAN.get(x["kpi_code"])
|
||
if not plan:
|
||
note = UNCHANGED_NOTES.get(x["kpi_code"])
|
||
print(" [不改] id=%-4s %-6s %-8s unit=%s 阈值=%s|%s|%s ← %s"
|
||
% (x["id"], x["kpi_code"], x["kpi_name"], x["unit"],
|
||
x["threshold_green"], x["threshold_yellow"], x["threshold_red"],
|
||
note[2] if note else ""))
|
||
continue
|
||
changed = (x["unit"] != plan["unit"] or x["threshold_green"] != plan["green"]
|
||
or x["threshold_yellow"] != plan["yellow"] or x["threshold_red"] != plan["red"])
|
||
print(" [%s] id=%-4s %-6s %-8s unit %s→%s 阈值 %s|%s|%s → %s|%s|%s"
|
||
% ("改" if changed else "已符", x["id"], x["kpi_code"], x["kpi_name"],
|
||
x["unit"], plan["unit"], x["threshold_green"], x["threshold_yellow"], x["threshold_red"],
|
||
plan["green"], plan["yellow"], plan["red"]))
|
||
print(" 理由: %s" % plan["why"])
|
||
if changed:
|
||
todo.append((x["id"], plan))
|
||
|
||
print("\n-- 影响面复算（最近可用期间） --")
|
||
p, before, after, detail = impact()
|
||
print(" 期间 P* = %s" % p)
|
||
print(" 修正前: 红 %d / 黄 %d / 绿 %d / 无判定 %d / 口径错误 %d"
|
||
% (before["red"], before["yellow"], before["green"], before["info"], before["口径错误"]))
|
||
print(" 修正后: 红 %d / 黄 %d / 绿 %d / 无判定 %d / 口径错误 %d"
|
||
% (after["red"], after["yellow"], after["green"], after["info"], after["口径错误"]))
|
||
print(" 条数变化（仅列变化项）:")
|
||
for d in detail:
|
||
print(" %-22s %s actual=%s unit=%s %s → %s [%s]"
|
||
% (d["kpi"], d["period"], d["actual"], d["unit"], d["before"], d["after"], d["thresholds"]))
|
||
print(" ※ 影响面以「修正后」为准（裁定 §4.5）；此处红/黄为静态阈值复算上界，"
|
||
"实际落库条数还受期间口径与幂等约束。")
|
||
|
||
if args.resolve_false_reds:
|
||
print("\n-- 假红灯作废 --")
|
||
ph = ",".join(":a%d" % i for i in range(len(FALSE_RED_ALERT_IDS)))
|
||
al = q("SELECT id, kpi_id, alert_level, status, alert_message FROM kpi_alerts "
|
||
"WHERE id IN (%s)" % ph,
|
||
{("a%d" % i): v for i, v in enumerate(FALSE_RED_ALERT_IDS)})
|
||
for a in al:
|
||
print(" id=%s level=%s status=%s msg=%s" % (a["id"], a["alert_level"], a["status"],
|
||
(a["alert_message"] or "")[:70]))
|
||
if not al:
|
||
print(" （未取到该两条预警，可能已清理）")
|
||
|
||
if not args.apply:
|
||
print("\n=== dry-run 结束：未写库（%d 项待改，加 --apply 落库 / 需审批）===" % len(todo))
|
||
return 0
|
||
|
||
for kid, plan in todo:
|
||
exec_sql("UPDATE kpi_definitions SET unit=:u, threshold_green=:g, threshold_yellow=:y, "
|
||
"threshold_red=:r WHERE id=:i AND entity_id=:e",
|
||
{"u": plan["unit"], "g": plan["green"], "y": plan["yellow"],
|
||
"r": plan["red"], "i": kid, "e": ENTITY_ID})
|
||
if args.resolve_false_reds:
|
||
exec_sql("UPDATE kpi_alerts SET status='resolved', resolution=:res, resolved_at=NOW() "
|
||
"WHERE id IN (%s)" % ",".join(":a%d" % i for i in range(len(FALSE_RED_ALERT_IDS))),
|
||
dict({("a%d" % i): v for i, v in enumerate(FALSE_RED_ALERT_IDS)},
|
||
res=VOID_RESOLUTION))
|
||
print("\n=== 已提交：%d 项阈值/单位%s ===" % (len(todo),
|
||
"；已作废假红灯 %s" % (FALSE_RED_ALERT_IDS,) if args.resolve_false_reds else ""))
|
||
print("-- 写入后回查 --")
|
||
for x in fetch_plan_rows():
|
||
if x["kpi_code"] in PLAN:
|
||
print(" id=%-4s %-6s unit=%s 阈值=%s|%s|%s"
|
||
% (x["id"], x["kpi_code"], x["unit"], x["threshold_green"],
|
||
x["threshold_yellow"], x["threshold_red"]))
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|