✨🐛 更新插件，修复一些bug

1 files changed, 171 insertions, 0 deletions

diff --git a/ATRI/plugins/key-repo/__init__.py b/ATRI/plugins/key-repo/__init__.py
new file mode 100644
index 0000000..4b2241c
--- /dev/null
+++ b/ATRI/plugins/key-repo/__init__.py
@@ -0,0 +1,171 @@
+import os
+import json
+import time
+from random import choice
+from pathlib import Path
+
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import Bot, MessageEvent
+
+from ATRI.config import nonebot_config
+from ATRI.service import Service as sv
+from ATRI.rule import is_block, is_in_dormant, is_in_service, to_bot
+
+
+# 此功能未完善
+
+
+KEYREPO_DIV = Path('.') / 'ATRI' / 'data' / 'database' / 'KeyRepo'
+os.makedirs(KEYREPO_DIV, exist_ok=True)
+
+
+__plugin_name__ = "KeyRepo"
+
+keyrepo = sv.on_message(rule=is_block()
+                        & is_in_dormant()
+                        & is_in_service(__plugin_name__)
+                        & to_bot())
+
+@keyrepo.handle()
+async def _keyrepo(bot: Bot, event: MessageEvent) -> None:
+    msg = str(event.get_message())
+
+    file_name = "data.json"
+    path = KEYREPO_DIV / file_name
+    try:
+        data = json.loads(path.read_bytes())
+    except:
+        with open(path, 'w') as r:
+            r.write(json.dumps({}))
+        data = {}
+
+    for key in data.keys():
+        if key in msg:
+            await keyrepo.finish(choice(data[key]))
+
+
+train = sv.on_command(
+    name="调教",
+    cmd="/train",
+    rule=is_block()
+)
+
+@train.got("key", prompt="哦哦哦要开始学习了！请告诉咱知识点")
+async def _train(bot: Bot, event: MessageEvent, state: T_State) -> None:
+    if "[CQ" in state["key"]:
+        await train.reject("仅支持纯文本呢...")
+
+@train.got("repo", prompt="咱该如何回答呢？")
+async def _trainR(bot: Bot, event: MessageEvent, state: T_State) -> None:
+    if "[CQ" in state["repo"]:
+        await train.reject("仅支持纯文本呢...")
+        
+    if state["key"] == "-d":
+        file_name = "review.json"
+        path = KEYREPO_DIV / file_name
+        try:
+            data = json.loads(path.read_bytes())
+        except:
+            data = {}
+
+        key = state["repo"]
+        if key not in data:
+            await train.finish("未发现该待审核的知识点呢...")
+        else:
+            msg = (
+                f"Key: {key}\n"
+                f"Repo: {data[key]['repo']}\n"
+                "已经从咱的审核列表移除！"
+            )
+            del data[key]
+            with open(path, 'w') as r:
+                r.write(json.dumps(data))
+            await train.finish(msg)
+    elif state["key"] == "-i":
+        file_name = "review.json"
+        path = KEYREPO_DIV / file_name
+        try:
+            data = json.loads(path.read_bytes())
+        except:
+            data = {}
+        if state["repo"] not in data:
+            await train.finish("未发现该知识点呢")
+        key = data[state["repo"]]
+        
+        msg = (
+            f"用户: {key['user']}\n"
+            f"知识点: {state['repo']}"
+            f"回复: {key['repo']}"
+            f"时间: {key['time']}"
+            "/train -r 知识点 y/n"
+        )
+        await train.finish(msg)
+    elif state["key"] == "-ls":
+        file_name = "review.json"
+        path = KEYREPO_DIV / file_name
+        try:
+            data = json.loads(path.read_bytes())
+        except:
+            data = {}
+        keys = "，".join(data.keys())
+        msg = f"目前等待审核的有如下：\n{keys}"
+        await train.finish(msg)
+    elif state["key"] == "-r":
+        file_name = "review.json"
+        path = KEYREPO_DIV / file_name
+        try:
+            data = json.loads(path.read_bytes())
+        except:
+            data = {}
+        
+    
+    key = state["key"]
+    repo = state["repo"]
+    user = event.get_user_id()
+    if user not in nonebot_config["superusers"]:
+        file_name = "review.json"
+        path = KEYREPO_DIV / file_name
+        try:
+            data = json.loads(path.read_bytes())
+        except:
+            data = {}
+        
+        if key in data:
+            msg = "欸欸欸，该词还在等待咱的审核，请先等先来的审核完再提交吧..."
+            await train.finish(msg)
+        else:
+            data[key] = {
+                "user": user,
+                "repo": repo,
+                "time": time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())
+            }
+            with open(path, 'r') as r:
+                r.write(json.dumps(data, indent=4))
+
+            msg = (
+            "欸欸欸这不错欸！不过，还是先等待咱审核审核，"
+            "如想撤销本次学习，请发送 /train -d 知识点"
+            )
+            await train.finish(msg)
+        
+    else:
+        file_name = "data.json"
+        path = KEYREPO_DIV / file_name
+        try:
+            data = json.loads(path.read_bytes())
+        except:
+            data = {}
+        
+        if key in data:
+            repo_list: list = data[key]
+            repo_list.append(repo)
+            data[key] = repo_list
+            msg = f"哦哦哦，{key}原来还有这样的回复，学到了~！"
+            await bot.send(event, msg)
+        else:
+            data[key] = [repo]
+            msg = "好欸，咱学到了新的知识点！"
+            await bot.send(event, msg)
+        
+        with open(path, 'w') as r:
+            r.write(json.dumps(data))