100 lines
3.5 KiB
Python
100 lines
3.5 KiB
Python
"""补充单株状态字典 tree_status：初选/重点/保存 三态，幂等。
|
||
|
||
背景：波次6 状态机要求 tree.status 覆盖完整选育流转
|
||
（存活→入选→初选→重点→保存→淘汰），而线上字典仅有
|
||
alive(存活)/selected(入选)/eliminated(淘汰) 三值。
|
||
本脚本补齐 primary(初选)/key(重点)/preserved(保存)，
|
||
使导入模板下拉与状态机流转齐备。
|
||
|
||
用法（backend/ 目录）：
|
||
ENVIRONMENT=dev "C:/ai/miniconda3/envs/dpb/python.exe" scripts/supplement_tree_status_dict.py --check
|
||
ENVIRONMENT=dev "C:/ai/miniconda3/envs/dpb/python.exe" scripts/supplement_tree_status_dict.py
|
||
|
||
幂等性：目标 dict_value 已存在则跳过；可重复执行。
|
||
"""
|
||
import argparse
|
||
import os
|
||
import sys
|
||
import uuid
|
||
|
||
BACKEND_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
if BACKEND_DIR not in sys.path:
|
||
sys.path.insert(0, BACKEND_DIR)
|
||
|
||
from sqlalchemy import create_engine, text # noqa: E402
|
||
|
||
from app.config.setting import settings # noqa: E402
|
||
|
||
# (dict_sort, dict_label, dict_value) —— 沿用 alive=1/eliminated=2/selected=3 的序号延续
|
||
NEW_ROWS = [
|
||
(4, "初选", "primary"),
|
||
(5, "重点", "key"),
|
||
(6, "保存", "preserved"),
|
||
]
|
||
DICT_TYPE = "tree_status"
|
||
|
||
|
||
def main() -> None:
|
||
parser = argparse.ArgumentParser(description="补充 tree_status 字典三态")
|
||
parser.add_argument("--check", action="store_true", help="仅预览将插入的行，不落库")
|
||
args = parser.parse_args()
|
||
|
||
engine = create_engine(settings.DB_URI)
|
||
print(f"# 连接: {settings.DB_URI} dict_type={DICT_TYPE} 模式: {'预览' if args.check else '执行'}")
|
||
|
||
with engine.connect() as conn:
|
||
type_row = conn.execute(
|
||
text("SELECT id FROM sys_dict_type WHERE dict_type = :dt"),
|
||
{"dt": DICT_TYPE},
|
||
).first()
|
||
if not type_row:
|
||
print(f"!! 字典类型 {DICT_TYPE} 不存在，中止")
|
||
sys.exit(1)
|
||
dict_type_id = type_row[0]
|
||
|
||
existing = {
|
||
row[0]
|
||
for row in conn.execute(
|
||
text("SELECT dict_value FROM sys_dict_data WHERE dict_type = :dt"),
|
||
{"dt": DICT_TYPE},
|
||
)
|
||
}
|
||
|
||
to_insert = [(sort, label, value) for sort, label, value in NEW_ROWS if value not in existing]
|
||
missing = [value for _, _, value in NEW_ROWS if value not in existing]
|
||
print(f"# 现有值: {sorted(existing)}")
|
||
print(f"# 待补: {missing}")
|
||
|
||
if args.check:
|
||
for sort, label, value in to_insert:
|
||
print(f" + (dict_sort={sort}) {label} -> {value}")
|
||
return
|
||
|
||
if not to_insert:
|
||
print("# 无新增，幂等退出 ✅")
|
||
return
|
||
|
||
for sort, label, value in to_insert:
|
||
conn.execute(
|
||
text(
|
||
"INSERT INTO sys_dict_data "
|
||
"(status, description, dict_sort, dict_label, dict_value, css_class, list_class, "
|
||
" is_default, dict_type, dict_type_id, uuid, is_deleted, created_time, updated_time) "
|
||
"VALUES (0, '', :sort, :label, :value, '', NULL, false, :dt, :dtid, :uuid, false, now(), now())"
|
||
),
|
||
{
|
||
"sort": sort,
|
||
"label": label,
|
||
"value": value,
|
||
"dt": DICT_TYPE,
|
||
"dtid": dict_type_id,
|
||
"uuid": str(uuid.uuid4()),
|
||
},
|
||
)
|
||
conn.commit()
|
||
print("# 已插入 ✅")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|