feat: 持久保存bot对象以成功重启

pkg/openai/manager.py

@@ -30,7 +30,7 @@ class OpenAIInteract:
 
     # 请求OpenAI Completion
     def request_completion(self, prompt, stop):
-        print("request")
+        # print("request")
         response = openai.Completion.create(
             prompt=prompt,
             stop=stop,

pkg/qqbot/manager.py

@@ -57,7 +57,7 @@ class QQBotManager:
 
     reply_filter = None
 
-    def __init__(self, mirai_http_api_config: dict, timeout: int = 60, retry: int = 3):
+    def __init__(self, mirai_http_api_config: dict, timeout: int = 60, retry: int = 3, first_time_init=True):
 
         self.timeout = timeout
         self.retry = retry
@@ -70,6 +70,28 @@ class QQBotManager:
         else:
             self.reply_filter = pkg.qqbot.filter.ReplyFilter([])
 
+
+        if first_time_init:
+            self.first_time_init(mirai_http_api_config)
+        else:
+            self.bot = pkg.utils.context.get_qqbot_manager().bot
+
+        pkg.utils.context.set_qqbot_manager(self)
+
+        @self.bot.on(FriendMessage)
+        async def on_friend_message(event: FriendMessage):
+            go(self.on_person_message, (event,))
+
+        @self.bot.on(StrangerMessage)
+        async def on_stranger_message(event: StrangerMessage):
+            go(self.on_person_message, (event,))
+
+        @self.bot.on(GroupMessage)
+        async def on_group_message(event: GroupMessage):
+            go(self.on_group_message, (event,))
+
+    def first_time_init(self, mirai_http_api_config: dict):
+
         if 'adapter' not in mirai_http_api_config or mirai_http_api_config['adapter'] == "WebSocketAdapter":
             bot = Mirai(
                 qq=mirai_http_api_config['qq'],
@@ -92,22 +114,9 @@ class QQBotManager:
         else:
             raise Exception("未知的适配器类型")
 
-        @bot.on(FriendMessage)
-        async def on_friend_message(event: FriendMessage):
-            go(self.on_person_message, (event,))
-
-        @bot.on(StrangerMessage)
-        async def on_stranger_message(event: StrangerMessage):
-            go(self.on_person_message, (event,))
-
-        @bot.on(GroupMessage)
-        async def on_group_message(event: GroupMessage):
-            go(self.on_group_message, (event,))
 
         self.bot = bot
 
-        pkg.utils.context.set_qqbot_manager(self)
-
     def send(self, event, msg, check_quote=True):
         asyncio.run(
             self.bot.send(event, msg, quote=True if hasattr(config,

pkg/qqbot/process.py

@@ -1,6 +1,7 @@
 # 此模块提供了消息处理的具体逻辑的接口
 import asyncio
 import datetime
+import threading
 
 import pkg.qqbot.manager as manager
 from func_timeout import func_set_timeout
@@ -8,7 +9,6 @@ import logging
 import openai
 
 from mirai import Image, MessageChain
-from mirai.models.message import Quote
 
 import config
 
@@ -162,8 +162,9 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
                                 reply.append(" ".join(params))
                     elif cmd == 'reload' and launcher_type == 'person' and launcher_id == config.admin_qq:
                         try:
-                            pkg.utils.reloader.reload_all()
-                            reply = ["[bot]已重新加载所有模块"]
+                            # pkg.utils.reloader.reload_all()
+                            threading.Thread(target=pkg.utils.reloader.reload_all, daemon=True).start()
+                            # reply = ["[bot]已重新加载所有模块"]
                         except Exception as e:
                             logging.error("reload failed:{}".format(e))
                             reply = ["[bot]重载失败:{}".format(e)]

pkg/utils/context.py

@@ -3,7 +3,8 @@ context = {
         'database.manager.DatabaseManager': None,
         'openai.manager.OpenAIInteract': None,
         'qqbot.manager.QQBotManager': None,
-    }
+    },
+    'logger_handler': None,
 }
 
 

pkg/utils/reloader.py

@@ -1,9 +1,14 @@
 import logging
+import os
+import threading
+
+import colorlog
 
 import pkg
 import importlib
 import pkgutil
 import pkg.utils.context
+from main import log_colors_config
 
 
 def walk(module, prefix=''):
@@ -16,7 +21,20 @@ def walk(module, prefix=''):
 
 
 def reload_all():
+    # 执行关闭流程
+    logging.info("执行程序关闭流程")
+    import main
+    main.stop()
+    import pkg
+
     context = pkg.utils.context.context
     walk(pkg)
     importlib.reload(__import__('config'))
+    importlib.reload(__import__('main'))
     pkg.utils.context.context = context
+
+    # 执行启动流程
+    logging.info("执行程序启动流程")
+    main.main()
+
+    logging.info('程序启动完成')