From 64a991e035e52e0a17e73d4e671a22ea9a7489da Mon Sep 17 00:00:00 2001
From: Kyomotoi <1172294279@qq.com>
Date: Sun, 14 Mar 2021 16:22:34 +0800
Subject: =?UTF-8?q?=E2=9C=A8=F0=9F=90=9B=20=E6=9B=B4=E6=96=B0=E6=8F=92?=
=?UTF-8?q?=E4=BB=B6=EF=BC=8C=E4=BF=AE=E5=A4=8D=E4=B8=80=E4=BA=9Bbug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
ATRI/plugins/key-repo/__init__.py | 171 ++++++++++++++++++++++++++++++++++++++
1 file changed, 171 insertions(+)
create mode 100644 ATRI/plugins/key-repo/__init__.py
(limited to 'ATRI/plugins/key-repo')
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))
--
cgit v1.2.3