import os
import json
from sqlalchemy import create_engine, MetaData, Table, func
from sqlalchemy.orm import sessionmaker
from dotenv import load_dotenv

load_dotenv()

def generate_questions_for_table(table_name, file_path, num_questions=50):
    engine = create_engine(os.getenv("SQL_DATABASE_URL", ""))
    metadata = MetaData()
    metadata.reflect(bind=engine)

    # 定义每张表的列索引、值列和问题模板
    tables_info = {
        "ProjectProperties": (0, "Value", "{name_value}的属性值是多少?", "属性值是{answer_value}"),
        "OtherFee": (3, "Amount", "{name_value}的金额是多少?", "金额是{answer_value}"),
        "FeeCollectionTable": (1, "Rate", "{name_value}的费率是多少?", "费率是{answer_value}"),
        "ProjectDivision": (5, "Sum_Price", "{name_value}的合价是多少?", "合价是{answer_value}"),
        "ProjectDivisions_CostPreview": (4, "Direct_Cost", "{name_value}的直接费是多少?", "直接费是{answer_value}"),
        "TotalCalculateTable": (3, "Amount", "{name_value}的金额是多少?", "金额是{answer_value}"),
        "ProjectQuantities": (6, "Encoding", "{name_value}的编码是多少?", "编码是{answer_value}")
    }

    if table_name not in tables_info:
        print(f"未找到表 {table_name} 的配置信息")
        return

    # 获取表信息
    name_index, value_column, question_template, answer_template = tables_info[table_name]

    # 加载表
    table = Table(table_name, metadata, autoload_with=engine)

    # 创建会话
    Session = sessionmaker(bind=engine)
    session = Session()

    # 获取列名
    name_column = table.columns.keys()[name_index]

    questions_and_answers = []

    # 生成指定数量的问题
    for _ in range(num_questions):
        # 查询表中的随机一行，并获取名称列和值列的值
        row = session.query(table).order_by(func.random()).first()
        name_value = getattr(row, name_column)
        answer_value = getattr(row, value_column)

        # 构造问题和答案
        question = question_template.format(name_value=name_value)
        answer = answer_template.format(answer_value=answer_value)

        # 添加到列表中
        questions_and_answers.append({
            "question": question,
            "answer": answer
        })

    # 将问题和答案以 JSON 格式写入文件
    with open(file_path, 'w', encoding='utf-8') as file:
        json.dump(questions_and_answers, file, ensure_ascii=False, indent=4)

    print(f"已生成表 {table_name} 的问题到文件: {file_path}")

def main():
    engine = create_engine(os.getenv("SQL_DATABASE_URL", ""))
    metadata = MetaData()
    metadata.reflect(bind=engine)

    # 获取脚本所在目录
    script_dir = os.path.dirname(os.path.abspath(__file__))

    # 遍历每张表并生成对应的 JSON 文件
    for table_name in metadata.tables.keys():
        # 文件路径为：脚本目录 + 表名 + .json
        file_path = os.path.join(script_dir, f"{table_name}.json")
        generate_questions_for_table(table_name, file_path)

if __name__ == "__main__":
    main()