import json
import pandas as pd


def getQueries_From_json(json_name):
    # 打开并读取JSON文件
    try:
        with open(json_name + ".json", "r", encoding="utf-8-sig") as file:
            data = json.load(file)
        # 从JSON数据中提取所有名为"Execute DAX Query"的事件中的QueryText
        queries = [
            event["metrics"]["QueryText"]
            for event in data["events"]
            if event["name"] == "Execute DAX Query"
        ]
        return queries
    except Exception as e:
        print(f"读取JSON文件错误: {e}，文件路径: {json_name}.json")
        raise


def getQueries_From_excel(excel_name):
    # 使用pandas读取Excel文件
    df = pd.read_excel(excel_name, engine="openpyxl")
    # 在DataFrame中添加一个新的列"NAME"，其值为"EVALUATE '" + 原始的"NAME"列值 + "'"
    df["NAME"] = "EVALUATE '" + df["NAME"] + "'"
    # 将新的"NAME"列转换为列表
    queries = df["NAME"].to_list()
    return queries


if __name__ == "__main__":
    import os
    from config import CURRENT_DIR_PATH, EXPORT_PAGE_JSON_NAME_1

    # 获取当前脚本所在的目录路径
    EXPORT_PAGE_JSON_PATH_1 = os.path.join(
        CURRENT_DIR_PATH, "Export_Json", EXPORT_PAGE_JSON_NAME_1
    )
    queries = getQueries_From_json(EXPORT_PAGE_JSON_PATH_1)
    print(queries)