From 37ad77a6f7966d0394a43134ab1339533dc16ecf Mon Sep 17 00:00:00 2001
From: zzj1804 <mehjz@foxmail.com>
Date: Thu, 2 Dec 2021 14:38:48 +0800
Subject: =?UTF-8?q?=E6=96=B0=E5=A2=9Echatbot=E6=8F=92=E4=BB=B6,=E4=BB=8E?=
 =?UTF-8?q?=E7=BE=A4=E8=81=8A=E5=A4=A9=E5=AD=A6=E4=B9=A0,=E5=B9=B6?=
 =?UTF-8?q?=E9=9A=8F=E6=9C=BA=E5=9B=9E=E8=AF=9D=20=E4=BF=AE=E6=94=B9chat?=
 =?UTF-8?q?=E6=8F=92=E4=BB=B6,=E6=97=A0=E6=B3=95=E5=9B=9E=E8=AF=9D?=
 =?UTF-8?q?=E6=97=B6=E5=B0=9D=E8=AF=95chatterbot?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/plugins/chatbot/__init__.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 ATRI/plugins/chatbot/__init__.py

(limited to 'ATRI/plugins/chatbot/__init__.py')

diff --git a/ATRI/plugins/chatbot/__init__.py b/ATRI/plugins/chatbot/__init__.py
new file mode 100644
index 0000000..53b3772
--- /dev/null
+++ b/ATRI/plugins/chatbot/__init__.py
@@ -0,0 +1,31 @@
+import random
+from ATRI.config import ChatterBot
+from ATRI.plugins.atri_chat_bot import ATRIChatBot
+from nonebot import on_message
+from nonebot import on_command
+from nonebot.adapters.cqhttp import (
+    Bot,
+    GroupMessageEvent,
+    MessageEvent,
+)
+from nonebot.permission import SUPERUSER
+
+chatbot = on_message(priority=114514)
+
+@chatbot.handle()
+async def _learn_from_group(bot: Bot, event: MessageEvent):
+    text = event.get_plaintext().strip()
+    if not text:
+        return
+    if isinstance(event, GroupMessageEvent):  # 从群友那学习说话
+        ATRIChatBot.learn(event.get_session_id(), text)
+        if random.random() <= ChatterBot.group_random_response_rate:   # 随机回话
+            await chatbot.finish(await ATRIChatBot.get_response(text))
+
+
+chatbot_learn = on_command("/learn_corpus", permission=SUPERUSER)
+
+@chatbot_learn.handle()
+async def _learn_from_corpus(bot: Bot, event: MessageEvent):
+    ATRIChatBot.learn_from_corpus()
+    await chatbot.finish("咱从corpus那学习完了!")
\ No newline at end of file
-- 
cgit v1.2.3