diff options
Diffstat (limited to 'utils/utils_rule')
| -rw-r--r-- | utils/utils_rule/__init__.py | 143 | ||||
| -rw-r--r-- | utils/utils_rule/ban_list_group.json | 1 | ||||
| -rw-r--r-- | utils/utils_rule/ban_list_user.json | 1 | ||||
| -rw-r--r-- | utils/utils_rule/switch.json | 1 |
4 files changed, 0 insertions, 146 deletions
diff --git a/utils/utils_rule/__init__.py b/utils/utils_rule/__init__.py deleted file mode 100644 index 1ec09db..0000000 --- a/utils/utils_rule/__init__.py +++ /dev/null @@ -1,143 +0,0 @@ -#!/usr/bin/env python3 -# -*- encoding: utf-8 -*- -''' -@File : __init__.py -@Time : 2020/11/06 19:27:00 -@Author : Kyomotoi -@Contact : [email protected] -@Github : https://github.com/Kyomotoi -@License : Copyright © 2018-2020 Kyomotoi, All Rights Reserved. -''' -__author__ = 'kyomotoi' - -import os -import json -from pathlib import Path -from nonebot.rule import Rule -from nonebot.typing import Bot, Event - - -def check_banlist() -> Rule: - ''' - 检查目标是否存在于封禁名单 - - :return: bool - ''' - async def _chech_banlist(bot: Bot, event: Event, state: dict) -> bool: - # 获取目标信息 - user = str(event.user_id) - group = str(event.group_id) - - # 名单目录 - BAN_LIST_USER_PATH = Path( - '.') / 'utils' / 'utils_rule' / 'ban_list_user.json' - BAN_LIST_GROUP_PATH = Path( - '.') / 'utils' / 'utils_rule' / 'ban_list_group.json' - - # 检查文件是否存在，如不存在，自动创建并写入默认值 - if not BAN_LIST_USER_PATH.is_file(): - with open(BAN_LIST_USER_PATH, 'w') as f: - f.write(json.dumps({})) - - if not BAN_LIST_GROUP_PATH.is_file(): - with open(BAN_LIST_GROUP_PATH, 'w') as f: - f.write(json.dumps({})) - - # 读取文件 - with open(BAN_LIST_USER_PATH, 'r') as f: - data_user = json.load(f) - - with open(BAN_LIST_GROUP_PATH, 'r') as f: - data_group = json.load(f) - - # 判断目标 - if user: - if user not in data_user: - if group: - if group not in data_group: - return True - else: - return False - else: - return True - else: - return False - - elif group: - if group not in data_group: - if user: - if user not in data_user: - return True - else: - return False - else: - return True - else: - return False - else: - return False - - return Rule(_chech_banlist) - - -def check_switch(func_name: str) -> Rule: - ''' - 检查目标功能是否开启 - - :return: bool - ''' - async def _check_switch(bot: Bot, event: Event, state: dict) -> bool: - # 获取目标信息 - group = str(event.group_id) - - # 文件目录 - SWITCH_ALL_PATH = Path('.') / 'utils' / 'utils_rule' / 'switch.json' - SWITCH_ALONE_PATH = Path( - '.') / 'ATRI' / 'data' / 'data_Group' / f'{group}' / 'switch.json' - - # 检查文件是否存在，如不存在，自动创建并写入默认值 - if not SWITCH_ALL_PATH.is_file(): - with open(SWITCH_ALL_PATH, 'ws') as f: - f.write(json.dumps({})) - - if not SWITCH_ALONE_PATH.is_file(): - try: - os.mkdir( - Path('.') / 'ATRI' / 'data' / 'data_Group' / f'{group}') - except: - pass - - with open(SWITCH_ALONE_PATH, 'w') as f: - f.write(json.dumps({})) - - # 读取文件 - with open(SWITCH_ALL_PATH, 'r') as f: - data_all = json.load(f) - - with open(SWITCH_ALONE_PATH, 'r') as f: - data_alone = json.load(f) - - # 判断目标是否存在于将要读取的文件，如不存在，写入 - # 此项举措是为了适应以后版本更新出现新插件的状况 - # 不至于每次都需要修改 - if func_name not in data_all: - data_all[func_name] = "True" - with open(SWITCH_ALL_PATH, 'w') as f: - f.write(json.dumps(data_all)) - - if func_name not in data_alone: - data_alone[func_name] = "True" - with open(SWITCH_ALONE_PATH, 'w') as f: - f.write(json.dumps(data_alone)) - - # 判断目标 - if data_all[func_name] == "True": - if data_alone[func_name] == "True": - return True - else: - return False - else: - await bot.send(event, f"Service-{func_name} has been closed.") - return False - - return Rule(_check_switch)
\ No newline at end of file diff --git a/utils/utils_rule/ban_list_group.json b/utils/utils_rule/ban_list_group.json deleted file mode 100644 index 9e26dfe..0000000 --- a/utils/utils_rule/ban_list_group.json +++ /dev/null @@ -1 +0,0 @@ -{}
\ No newline at end of file diff --git a/utils/utils_rule/ban_list_user.json b/utils/utils_rule/ban_list_user.json deleted file mode 100644 index 9e26dfe..0000000 --- a/utils/utils_rule/ban_list_user.json +++ /dev/null @@ -1 +0,0 @@ -{}
\ No newline at end of file diff --git a/utils/utils_rule/switch.json b/utils/utils_rule/switch.json deleted file mode 100644 index c19bf91..0000000 --- a/utils/utils_rule/switch.json +++ /dev/null @@ -1 +0,0 @@ -{"anime-setu": "True", "anime-pic-search": "True", "anime-vid-search": "True", "all-off-anime-setu": "True", "ai-face": "True", "pixiv-pic-search": "True", "pixiv-author-search": "True", "pixiv-rank": "True", "one-key-adult": "True", "genshin-search": "True"}
\ No newline at end of file |