"""诊断单个问题的 SQL 生成与执行（用法：python scripts/dev/diag_question.py "问题" 角色）。"""
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[2]))

from app.service.analyst_agent import AnalystAgent
from app.utils.auth import AuthContext


def main():
    question = sys.argv[1] if len(sys.argv) > 1 else "近30天申购金额总额"
    role = sys.argv[2] if len(sys.argv) > 2 else "ops"
    agent = AnalystAgent()
    ctx = AuthContext(subject_id=f"STAFF-{role}", token_type="staff", roles=[role], staff_type=role)
    domain = {"analyst": "full", "advisor": "assigned", "risk_officer": "risk", "ops": "aggregate"}[role]
    scope = []
    if domain == "assigned":
        scope = agent.repo.resolve_advisor_scope(ctx.subject_id)
    sql, _ = agent._generate_sql(question, domain, scope)
    print("SQL:", sql)
    try:
        res = agent.repo.execute_readonly(sql)
        print("OK cols=", res["columns"], "rows[:3]=", res["rows"][:3])
    except Exception as exc:  # noqa: BLE001
        print("EXEC ERROR:", type(exc).__name__, exc)


if __name__ == "__main__":
    main()