# -*- coding:utf-8 -*-
"""幂等补列：opp_mining_batches / opp_clusters 加 project_id（information_schema 守卫，绝不 DROP）。"""
import asyncio, os, sys
W = "/d/pipeline/pipeline-app"; os.chdir(W); sys.path.insert(0, W)
from appPublic.folderUtils import ProgramPath
from appPublic.jsonConfig import getConfig
from appPublic.event_dispatcher import EventDispatcher
from sqlor.dbpools import DBPools
from ahserver.serverenv import ServerEnv
p = ProgramPath(); c = getConfig(W, {"workdir": W, "ProgramPath": p}); DBPools(c.databases)
se = ServerEnv(); se.event_dispatcher = EventDispatcher(); se.get_module_dbname = lambda m: "pipeline"

TARGETS = [("opp_mining_batches", "project_id"), ("opp_clusters", "project_id")]

async def main():
    async with DBPools().sqlorContext("pipeline") as sor:
        for table, col in TARGETS:
            chk = await sor.sqlExe(
                "SELECT COLUMN_NAME FROM information_schema.COLUMNS "
                "WHERE table_schema=DATABASE() AND table_name=${t}$ AND column_name=${c}$",
                {"t": table, "c": col})
            await sor.sqlExe("COMMIT", {})
            if chk:
                print("SKIP(已存在): %s.%s" % (table, col))
                continue
            await sor.sqlExe(
                "ALTER TABLE %s ADD COLUMN `%s` VARCHAR(32) DEFAULT '' "
                "COMMENT '项目ID(空=平台级挖掘)'" % (table, col), {})
            await sor.sqlExe("COMMIT", {})
            print("ADDED: %s.%s" % (table, col))
        # 索引：批次按 org+project 查询
        try:
            await sor.sqlExe("CREATE INDEX idx_opp_batches_orgproj ON opp_mining_batches (org_id, project_id)", {})
            await sor.sqlExe("COMMIT", {})
            print("index added")
        except Exception as e:
            print("index skip:", str(e)[:60])
        # 验证
        for table, col in TARGETS:
            chk = await sor.sqlExe(
                "SELECT COLUMN_NAME FROM information_schema.COLUMNS "
                "WHERE table_schema=DATABASE() AND table_name=${t}$ AND column_name=${c}$",
                {"t": table, "c": col})
            await sor.sqlExe("COMMIT", {})
            print("VERIFY %s.%s:" % (table, col), "OK" if chk else "MISSING")

asyncio.run(main())