[Update]

新增：
   * 舆情检测
   * 对涩图加以调用限制

修复：
  * Pixiv插件全体
  * 储存群聊信息时无法创建文件

优化：
  * 部分代码重构，效率up
  * 调整插件结构，使其看起来更舒服

1 files changed, 151 insertions, 0 deletions

diff --git a/ATRI/utils/utils_rule/__init__.py b/ATRI/utils/utils_rule/__init__.py
new file mode 100644
index 0000000..dcaeedc
--- /dev/null
+++ b/ATRI/utils/utils_rule/__init__.py
@@ -0,0 +1,151 @@
+#!/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 typing import Optional
+from nonebot.rule import Rule
+from nonebot.typing import Bot, Event
+
+
+def check_banlist(for_group: Optional[bool] = False) -> 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(
+            '.') / 'ATRI' / 'utils' / 'utils_rule' / 'ban_list_user.json'
+        BAN_LIST_GROUP_PATH = Path(
+            '.') / 'ATRI' / '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 not for_group:
+            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
+        else:
+            if group not in data_group:
+                return True
+            else:
+                return False
+
+    return Rule(_chech_banlist)
+
+
+def check_switch(func_name: str, notice: bool) -> Rule:
+    '''
+    检查目标功能是否开启
+
+    :return: bool
+    '''
+    async def _check_switch(bot: Bot, event: Event, state: dict) -> bool:
+        # 获取目标信息
+        group = str(event.group_id)
+
+        # 文件目录
+        SWITCH_ALL_PATH = Path('.') / 'ATRI' / '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:
+            if notice:
+                await bot.send(event, f"Service-{func_name} has been closed.")
+            return False
+
+    return Rule(_check_switch)