diff --git a/configs/utils/init_config.py b/configs/utils/init_config.py index 9f5f2bdf..523c426c 100644 --- a/configs/utils/init_config.py +++ b/configs/utils/init_config.py @@ -1,176 +1,43 @@ -# import nonebot +from configs.path_config import DATA_PATH from pathlib import Path try: import ujson as json except ModuleNotFoundError: import json -# driver: nonebot.Driver = nonebot.get_driver() - -base_config = Path() / 'config.json' -plugins_cmd_config = Path() / 'configs' / 'plugins2cmd_config.json' -plugins_setting = Path() / 'configs' / 'plugins_setting.json' - - -def init_config(): - if not base_config.exists(): - base_config.parent.mkdir(parents=True, exist_ok=True) - config_dict = { - 'apikey': { - 'LOLICON_KEY': '', - 'TL_KEY': [], - }, - 'sql': { - 'bind': '', - 'sql_name': '', - 'user': '', - 'password': '', - 'address': '', - 'port': '', - 'database': '', - }, - 'path': { - 'IMAGE_PATH': '', - 'VOICE_PATH': '', - 'TXT_PATH': '', - 'LOG_PATH': '', - 'DATA_PATH': '', - 'DRAW_PATH': '', - 'TEMP_PATH': '', - }, - 'proxy': { - 'system_proxy': '', - 'buff_proxy': '' - }, - 'level': { - 'DELETE_IMG_LEVEL': 7, - 'MOVE_IMG_LEVEL': 7, - 'UPLOAD_LEVEL': 6, - 'BAN_LEVEL': 5, - 'OC_LEVEL': 2, - 'MUTE_LEVEL': 5, - }, - 'auth': { - 'ADMIN_DEFAULT_AUTH': 5, - 'admin_plugins_auth': { - "admin_bot_manage": 2, - "ban": 5, - "delete_img": 7, - "move_img": 7, - "upload_img": 6, - "admin_help": 1, - "mute": 5 - } - } - } - with open(base_config, 'w', encoding='utf8') as f: - json.dump(config_dict, f, indent=4, ensure_ascii=False) - if not plugins_cmd_config.exists(): - plugins_cmd_config.parent.mkdir(parents=True, exist_ok=True) - config_dict = { - 'base_config': { - 'sign_in': ['签到'], - 'send_img': ['发送图片', '萝莉', '美图', '壁纸'], - 'send_setu': ['色图', '涩图', '瑟图', '查色图'], - 'white2black': ['黑白图', '黑白草图'], - 'coser': ['coser', 'cos'], - 'quotations': ['语录'], - 'jitang': ['鸡汤'], - 'send_dinggong_voice': ['骂我', '骂老子', '骂劳资'], - 'open_cases': ['开箱', '我的开箱', '群开箱统计', '我的金色'], - 'luxun': ['鲁迅说过', '鲁迅说'], - 'fake_msg': ['假消息'], - 'buy': ['购买', '购买道具'], - 'my_gold': ['我的金币'], - 'my_props': ['我的道具'], - 'shop_handle': ['商店'], - 'nonebot_plugin_cocdicer': ['骰子娘'], - 'update_pic': ['图片', '操作图片', '修改图片'], - 'search_buff_skin_price': ['查询皮肤'], - 'weather': ['天气', '查询天气', '天气查询'], - 'yiqing': ['疫情', '疫情查询', '查询疫情'], - 'what_anime': ['识番'], - 'search_anime': ['搜番'], - 'songpicker2': ['点歌'], - 'epic': ['epic'], - 'pixiv': ['pixiv', 'p站排行', '搜图'], - 'poke': ['戳一戳'], - 'draw_card': ['游戏抽卡', '原神一井', '原神来一井', '方舟一井', '方舟来一井'], - 'ai': ['ai', 'Ai', 'AI', 'aI'], - 'one_friend': ['我有一个朋友', '我有一个朋友想问问'], - 'translate': ['翻译', '英翻', '翻英', '日翻', '翻日', '韩翻', '翻韩'], - 'nonebot_plugin_picsearcher': ['识图'], - 'almanac': ['原神黄历', '黄历'], - 'material_remind': ['今日素材', '天赋材料'], - 'qiu_qiu_translation': ['丘丘翻译', '丘丘一下', '丘丘语翻译'], - 'query_resource_points': ['原神资源查询', '原神资源列表'], - } - } - with open(plugins_cmd_config, 'w', encoding='utf8') as f: - json.dump(config_dict, f, indent=4, ensure_ascii=False) - if not plugins_setting.exists(): - plugins_setting.parent.mkdir(parents=True, exist_ok=True) - config_dict = { - 'base': { - 'IMAGE_DIR_LIST': ["色图", "美图", "萝莉", "壁纸"], - 'BAN_RESULT': "才不会给你发消息.", - }, - 'draw_card': { - 'PRTS_FLAG': True, # 明日方舟 - 'GENSHIN_FLAG': True, # 原神 - 'PRETTY_FLAG': True, # 赛马娘 - 'GUARDIAN_FLAG': True, # 坎公骑冠剑 - 'PCR_FLAG': True, # 公主连结 - 'AZUR_FLAG': True, # 碧蓝航线 - 'FGO_FLAG': True, # 命运-冠位指定（FGO） - 'ONMYOJI_FLAG': True, # 阴阳师 - - 'PCR_TAI': False # pcr是否开启台服卡池 - }, - 'fudu': { - 'FUDU_PROBABILITY': 0.7, - }, - 'reimu': { - 'MAXINFO_REIMU': 7, - 'COUNT_PER_DAY_REIMU': 5, - }, - 'bt': { - 'MAXINFO_BT': 10, - }, - 'search_anime': { - 'MAXINFO_PRIVATE_ANIME': 20, - 'MAXINFO_GROUP_ANIME': 5, - }, - 'picsearcher': { - 'MAX_FIND_IMG_COUNT': 3, - }, - 'sign': { - 'MAX_SIGN_GOLD': 200, - }, - 'send_setu': { - 'INITIAL_SETU_PROBABILITY': 0.7, - 'MAX_SETU_R_COUNT': 5, - 'DOWNLOAD_SETU': True, - }, - 'malicious_ban': { - 'MALICIOUS_BAN_TIME': 30, - 'MALICIOUS_BAN_COUNT': 8, - 'MALICIOUS_CHECK_TIME': 5, - }, - 'open_case': { - 'INITIAL_OPEN_CASE_COUNT': 20, - }, - 'mute': { - 'MUTE_DEFAULT_COUNT': 10, - 'MUTE_DEFAULT_TIME': 7, - 'MUTE_DEFAULT_DURATION': 10, - }, - 'update_gocq': { - 'UPDATE_GOCQ_GROUP': [], - }, - } - with open(plugins_setting, 'w', encoding='utf8') as f: - json.dump(config_dict, f, indent=4, ensure_ascii=False) +def init_config(plugins2info_dict, plugins2cd_dict, plugins2exists_dict): + plugins2info_file = Path(DATA_PATH) / 'configs' / 'plugins2info.json' + plugins2info_file.parent.mkdir(exist_ok=True, parents=True) + + if plugins2info_file.exists(): + with open(plugins2info_file, 'r') as f: + _data = json.load(f) + for p in plugins2info_dict: + if not _data.get(p): + _data[p] = plugins2info_dict[p] + with open(plugins2info_file, 'w') as wf: + json.dump(_data, wf, ensure_ascii=False, indent=4) + plugins2info_dict = _data + else: + with open(plugins2info_file, 'w', encoding='utf8') as wf: + json.dump(plugins2info_dict, wf, ensure_ascii=False, indent=4) + + plugins2cd_file = Path(DATA_PATH) / 'configs' / 'plugins2cd.json' + if plugins2cd_file.exists(): + with open(plugins2cd_file, 'r', encoding='utf8') as f: + plugins2cd_dict = json.load(f) + else: + with open(plugins2cd_file, 'w', encoding='utf8') as wf: + json.dump(plugins2cd_dict, wf, ensure_ascii=False, indent=4) + + plugins2exists_file = Path(DATA_PATH) / 'configs' / 'plugins2exists.json' + if plugins2exists_file.exists(): + with open(plugins2exists_file, 'r', encoding='utf8') as f: + plugins2exists_dict = json.load(f) + else: + with open(plugins2exists_file, 'w', encoding='utf8') as wf: + json.dump(plugins2exists_dict, wf, ensure_ascii=False, indent=4) + return plugins2info_dict, plugins2cd_dict, plugins2exists_dict