1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
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))
|
