"""补充单株状态字典 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()