import os

from module.plot_entity import generate_er_diagram, parse_sql_file_without_comments

if __name__ == '__main__':

    #====================================================================#
    #------------------------Version: 1.0.0 -----------------------------#
    #====================================================================#

    # 你的SQL脚本文件相对路径(必须是Navicat导出的，若不写注释则为表名或字段名)
    sql_file_path = 'db_struct.sql'
    # 中文er图(True)还是表名er图(False)
    er_chinese = True

    #--------------------------------------------------------------------#
    #====================================================================#

    parse_data = {}
    if os.path.exists(sql_file_path):
        # 解析SQL文件
        parse_data = parse_sql_file_without_comments(sql_file_path)
    else:
        print(f"文件 {sql_file_path} 不存在。")

    # 打印解析到的数据
    # print("解析结果：" + json.dumps(parse_data, indent=4, ensure_ascii=False))

    if not parse_data:
        print("未解析到任何表结构信息，请检查 SQL 文件路径和格式！")
    else:
        # 生成ER图
        generate_er_diagram(parse_data, './export', er_chinese)