"""补齐 breeding 全部建表（幂等，逐句隔离）。

用正则按顶层 SQL 关键字（CREATE TABLE / CREATE INDEX / COMMENT ON / ALTER TABLE ...）
切分，避免部分 CREATE TABLE 不以分号结尾导致多表被合并成一条语句的问题。
每张 CREATE TABLE 成功/失败均打印表名与真实异常（repr 保证 ascii 安全）。
"""
import re
from pathlib import Path

from sqlalchemy import text

from app.core.database import engine

SQL_PATH = Path(r"d:\dpb\dpb\backend\sql\bre_tables.sql")

CHECK_TABLES = [
    "bre_trait", "bre_tree", "bre_germplasm",
    "bre_prediction", "bre_prediction_value",
    "bre_combining_ability", "bre_statistics_job",
]


def classify(s: str) -> str:
    u = s.upper()
    if u.startswith("CREATE TABLE"):
        return "CREATE"
    if "CREATE INDEX" in u:
        return "IDX"
    if u.startswith("COMMENT"):
        return "CMT"
    return "OTHER"


def main() -> None:
    sql = SQL_PATH.read_text(encoding="utf-8")
    parts = re.split(
        r"(?=(?:CREATE TABLE|CREATE INDEX|COMMENT ON|ALTER TABLE|DROP TABLE|DROP INDEX))",
        sql,
        flags=re.IGNORECASE,
    )
    stmts: list[str] = []
    for p in parts:
        s = p.strip().rstrip(";").strip()
        if s and not s.startswith("--"):
            stmts.append(s)

    ok = fail = 0
    with engine.connect() as conn:
        for s in stmts:
            kind = classify(s)
            try:
                conn.execute(text(s))
                conn.commit()
                ok += 1
                if kind == "CREATE":
                    m = re.search(r"breed_\w+", s)
                    print("CREATE OK", m.group(0) if m else s[:30])
            except Exception as e:  # noqa: BLE001
                conn.rollback()
                fail += 1
                if kind in ("CREATE", "IDX", "CMT"):
                    print(kind, "FAIL", s[:38].replace("\n", " "), "|", repr(e)[:140])
    with engine.connect() as conn:
        for t in CHECK_TABLES:
            try:
                conn.execute(text(f"select 1 from {t} limit 1"))
                print("EXISTS", t)
            except Exception:  # noqa: BLE001
                print("MISSING", t)
    print(f"DDL done: ok={ok} fail={fail}")


if __name__ == "__main__":
    main()