import json

from Log import logger
from app.models import MenuCapacityModel, WebMenuModel, GroupModel


async def dialog_menu_sync(db):
    menu_list = []
    with open("env_conf/menu_conf.json", 'r', encoding='utf-8') as file:
        # 加载JSON数据
        data = json.load(file)
        menu_list = data.get("data", [])

    db.query(WebMenuModel).delete()
    db.query(MenuCapacityModel).delete()
    db.commit()

    for menu in menu_list:
        # print(menu)
        agent = menu.pop("agent", [])
        for i in agent:
            capacity = MenuCapacityModel(menu_id=menu["id"], capacity_id=i, capacity_type=2)
            db.add(capacity)
        dialog = menu.pop("dialog", [])
        for i in dialog:
            capacity = MenuCapacityModel(menu_id=menu["id"], capacity_id=i, capacity_type=1)
            db.add(capacity)
        menu_obj = WebMenuModel(**menu)
        db.add(menu_obj)
    db.commit()


async def default_group_sync(db):
    group = db.query(GroupModel).filter_by(group_type=2).first()
    if not group:
        logger.error("未初始化")