1 files changed, 72 insertions, 0 deletions
diff --git a/ATRI/plugins/wordcloud.py b/ATRI/plugins/wordcloud.py
new file mode 100644
index 0000000..4913727
--- /dev/null
+++ b/ATRI/plugins/wordcloud.py
@@ -0,0 +1,72 @@
+import json
+from pathlib import Path
+from random import randint
+import nonebot
+from nonebot import on_command
+from nonebot import CommandSession
+from nonebot import MessageSegment
+
+import config # type: ignore
+
+
+bot = nonebot.get_bot()
+master = config.MASTER()
+
+
+@on_command('add_word', aliases = ['增加词汇'], only_to_me = False)
+async def _(session: CommandSession):
+    if session.event.user_id == master:
+        msg = session.event.raw_message.split(' ', 3)
+        w_tpye = msg[0]
+        word = msg[1]
+        repo = msg[2]
+        prob = int(msg[3])
+        with open('ATRI/plugins/wordcloud/wordcloud.json', 'r') as f:
+            data = json.load(f)
+
+        if w_tpye == '添加词云' or '上传词云':
+            if word in data.keys():
+                await session.send('该词已存在~！')
+
+            else:
+                data[f"{word}"] = [f"{repo}",prob]
+                f = open('ATRI/plugins/wordcloud/wordcloud.json', 'w')
+                f.write(json.dumps(data))
+                f.close()
+                session.finish(f"学習しました！\nWord：[{word}]\nRepo：[{repo}]\nProbability：[{'%.2f%%' % (round(1 / prob , 1) * 100)}]")
+        
+        elif w_tpye == '删除词云':
+            if word in data.keys():
+                data.pop(word)
+                await session.send(f'已成功从ATRI记忆模块中抹除[{word}]')
+            
+            else:
+                 session.finish(f'ATRI貌似没法从记忆中找到关键词[{word}]呢...')
+
+
+[email protected]_message("group")
+async def repo(context):
+    user = context["user_id"]
+    group = context["group_id"]
+    word = context["message"]
+    print(word)
+    with open('ATRI/plugins/noobList/noobList.json', 'r') as f:
+        nL = json.load(f)
+
+    if str(user) in nL.keys():
+        pass
+    else:
+        with open('ATRI/plugins/wordcloud/wordcloud.json', 'r') as f:
+            data = json.load(f)
+
+        if str(word) in data.keys():
+            lt = data[f"{word}"]
+            print(lt)
+            msg = lt[0]
+            prob = int(lt[1])
+            res = randint(1,prob)
+            if res == 1:
+                await bot.send_msg(
+                    group_id = group,
+                    message = msg
+                ) # type: ignore
+\ No newline at end of file