From ac821fa4d0703c0ba2bfbf9890debcc052cda43a Mon Sep 17 00:00:00 2001
From: Kyomotoi <1172294279@qq.com>
Date: Thu, 13 Aug 2020 18:16:57 +0800
Subject: [Fix]

---
 ATRI/plugins/wordcloud.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

(limited to 'ATRI')

diff --git a/ATRI/plugins/wordcloud.py b/ATRI/plugins/wordcloud.py
index 4913727..cb63e04 100644
--- a/ATRI/plugins/wordcloud.py
+++ b/ATRI/plugins/wordcloud.py
@@ -1,10 +1,8 @@
 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
 
@@ -13,7 +11,7 @@ bot = nonebot.get_bot()
 master = config.MASTER()
 
 
-@on_command('add_word', aliases = ['增加词汇'], only_to_me = False)
+@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)
@@ -24,7 +22,7 @@ async def _(session: CommandSession):
         with open('ATRI/plugins/wordcloud/wordcloud.json', 'r') as f:
             data = json.load(f)
 
-        if w_tpye == '添加词云' or '上传词云':
+        if w_tpye == '增加词汇':
             if word in data.keys():
                 await session.send('该词已存在~！')
 
@@ -35,7 +33,7 @@ async def _(session: CommandSession):
                 f.close()
                 session.finish(f"学習しました！\nWord：[{word}]\nRepo：[{repo}]\nProbability：[{'%.2f%%' % (round(1 / prob , 1) * 100)}]")
         
-        elif w_tpye == '删除词云':
+        elif w_tpye == '删除词汇':
             if word in data.keys():
                 data.pop(word)
                 await session.send(f'已成功从ATRI记忆模块中抹除[{word}]')
-- 
cgit v1.2.3