refactor: 修改引入风格

pkg/qqbot/banlist.py

@@ -1,18 +1,18 @@
-import pkg.utils.context
+from ..utils import context
 
 
 def is_banned(launcher_type: str, launcher_id: int, sender_id: int) -> bool:
-    if not pkg.utils.context.get_qqbot_manager().enable_banlist:
+    if not context.get_qqbot_manager().enable_banlist:
         return False
     
     result = False
     
     if launcher_type == 'group':
         # 检查是否显式声明发起人QQ要被person忽略
-        if sender_id in pkg.utils.context.get_qqbot_manager().ban_person:
+        if sender_id in context.get_qqbot_manager().ban_person:
             result = True
         else:
-            for group_rule in pkg.utils.context.get_qqbot_manager().ban_group:
+            for group_rule in context.get_qqbot_manager().ban_group:
                 if type(group_rule) == int:
                     if group_rule == launcher_id:  # 此群群号被禁用
                         result = True
@@ -32,7 +32,7 @@ def is_banned(launcher_type: str, launcher_id: int, sender_id: int) -> bool:
                     
     else:
         # ban_person, 与群规则相同
-        for person_rule in pkg.utils.context.get_qqbot_manager().ban_person:
+        for person_rule in context.get_qqbot_manager().ban_person:
             if type(person_rule) == int:
                 if person_rule == launcher_id:
                     result = True

pkg/qqbot/blob.py

@@ -2,21 +2,21 @@
 import os
 import time
 import base64
+import typing
 
-import config
 from mirai.models.message import MessageComponent, MessageChain, Image
 from mirai.models.message import ForwardMessageNode
 from mirai.models.base import MiraiBaseModel
-from typing import List
-import pkg.utils.context as context
-import pkg.utils.text2img as text2img
+
+from ..utils import text2img
+import config
 
 
 class ForwardMessageDiaplay(MiraiBaseModel):
     title: str = "群聊的聊天记录"
     brief: str = "[聊天记录]"
     source: str = "聊天记录"
-    preview: List[str] = []
+    preview: typing.List[str] = []
     summary: str = "查看x条转发消息"
 
 
@@ -26,7 +26,7 @@ class Forward(MessageComponent):
     """消息组件类型。"""
     display: ForwardMessageDiaplay
     """显示信息"""
-    node_list: List[ForwardMessageNode]
+    node_list: typing.List[ForwardMessageNode]
     """转发消息节点列表。"""
     def __init__(self, *args, **kwargs):
         if len(args) == 1:

pkg/qqbot/cmds/aamgr.py

@@ -1,10 +1,7 @@
-import importlib
-import inspect
 import logging
 import copy
 import pkgutil
 import traceback
-import types
 import json
 
 

pkg/qqbot/cmds/funcs/draw.py

@@ -1,11 +1,12 @@
-from ..aamgr import AbstractCommandNode, Context
 import logging
 
-from mirai import Image
+import mirai
+
+from .. import aamgr
 import config
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="draw",
     description="使用DALL·E生成图片",
@@ -13,9 +14,9 @@ import config
     aliases=[],
     privilege=1
 )
-class DrawCommand(AbstractCommandNode):
+class DrawCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import pkg.openai.session
 
         reply = []
@@ -28,7 +29,7 @@ class DrawCommand(AbstractCommandNode):
             res = session.draw_image(" ".join(ctx.params))
 
             logging.debug("draw_image result:{}".format(res))
-            reply = [Image(url=res['data'][0]['url'])]
+            reply = [mirai.Image(url=res['data'][0]['url'])]
             if not (hasattr(config, 'include_image_description')
                     and not config.include_image_description):
                 reply.append(" ".join(ctx.params))

pkg/qqbot/cmds/funcs/func.py

@@ -1,10 +1,9 @@
-from ..aamgr import AbstractCommandNode, Context
 import logging
-
 import json
 
+from .. import aamgr
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="func",
     description="管理内容函数",
@@ -12,9 +11,9 @@ import json
     aliases=[],
     privilege=1
 )
-class FuncCommand(AbstractCommandNode):
+class FuncCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         from pkg.plugin.models import host
 
         reply = []

pkg/qqbot/cmds/plugin/plugin.py

@@ -1,12 +1,9 @@
-from ..aamgr import AbstractCommandNode, Context
-
-import os
-
-import pkg.plugin.host as plugin_host
-import pkg.utils.updater as updater
+from ....plugin import host as plugin_host
+from ....utils import updater
+from .. import aamgr
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="plugin",
     description="插件管理",
@@ -14,9 +11,9 @@ import pkg.utils.updater as updater
     aliases=[],
     privilege=1
 )
-class PluginCommand(AbstractCommandNode):
+class PluginCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         reply = []
         plugin_list = plugin_host.__plugins__
         if len(ctx.params) == 0:
@@ -48,7 +45,7 @@ class PluginCommand(AbstractCommandNode):
             return False, []
 
     
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=PluginCommand,
     name="get",
     description="安装插件",
@@ -56,9 +53,9 @@ class PluginCommand(AbstractCommandNode):
     aliases=[],
     privilege=2
 )
-class PluginGetCommand(AbstractCommandNode):
+class PluginGetCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import threading
         import logging
         import pkg.utils.context
@@ -81,7 +78,7 @@ class PluginGetCommand(AbstractCommandNode):
         return True, reply
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=PluginCommand,
     name="update",
     description="更新指定插件或全部插件",
@@ -89,9 +86,9 @@ class PluginGetCommand(AbstractCommandNode):
     aliases=[],
     privilege=2
 )
-class PluginUpdateCommand(AbstractCommandNode):
+class PluginUpdateCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import threading
         import logging
         plugin_list = plugin_host.__plugins__
@@ -130,7 +127,7 @@ class PluginUpdateCommand(AbstractCommandNode):
         return True, reply
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=PluginCommand,
     name="del",
     description="删除插件",
@@ -138,9 +135,9 @@ class PluginUpdateCommand(AbstractCommandNode):
     aliases=[],
     privilege=2
 )
-class PluginDelCommand(AbstractCommandNode):
+class PluginDelCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         plugin_list = plugin_host.__plugins__
         reply = []
 
@@ -157,7 +154,7 @@ class PluginDelCommand(AbstractCommandNode):
         return True, reply
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=PluginCommand,
     name="on",
     description="启用指定插件",
@@ -165,7 +162,7 @@ class PluginDelCommand(AbstractCommandNode):
     aliases=[],
     privilege=2
 )
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=PluginCommand,
     name="off",
     description="禁用指定插件",
@@ -173,9 +170,9 @@ class PluginDelCommand(AbstractCommandNode):
     aliases=[],
     privilege=2
 )
-class PluginOnOffCommand(AbstractCommandNode):
+class PluginOnOffCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import pkg.plugin.switch as plugin_switch
 
         plugin_list = plugin_host.__plugins__

pkg/qqbot/cmds/session/default.py

@@ -1,7 +1,6 @@
-from ..aamgr import AbstractCommandNode, Context
+from .. import aamgr
 
-
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="default",
     description="操作情景预设",
@@ -9,9 +8,9 @@ from ..aamgr import AbstractCommandNode, Context
     aliases=[],
     privilege=1
 )
-class DefaultCommand(AbstractCommandNode):
+class DefaultCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import pkg.openai.session
         session_name = ctx.session_name
         params = ctx.params
@@ -45,7 +44,7 @@ class DefaultCommand(AbstractCommandNode):
         return True, reply
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=DefaultCommand,
     name="set",
     description="设置默认情景预设",
@@ -53,9 +52,9 @@ class DefaultCommand(AbstractCommandNode):
     aliases=[],
     privilege=2
 )
-class DefaultSetCommand(AbstractCommandNode):
+class DefaultSetCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         reply = []
 
         if len(ctx.crt_params) == 0:

pkg/qqbot/cmds/session/del.py

@@ -1,8 +1,7 @@
-from ..aamgr import AbstractCommandNode, Context
-import datetime
+from .. import aamgr
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="del",
     description="删除当前会话的历史记录",
@@ -10,9 +9,9 @@ import datetime
     aliases=[],
     privilege=1
 )
-class DelCommand(AbstractCommandNode):
+class DelCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import pkg.openai.session
         session_name = ctx.session_name
         params = ctx.params
@@ -33,7 +32,7 @@ class DelCommand(AbstractCommandNode):
         return True, reply
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=DelCommand,
     name="all",
     description="删除当前会话的全部历史记录",
@@ -41,9 +40,9 @@ class DelCommand(AbstractCommandNode):
     aliases=[],
     privilege=1
 )
-class DelAllCommand(AbstractCommandNode):
+class DelAllCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import pkg.openai.session
         session_name = ctx.session_name
         reply = []

pkg/qqbot/cmds/session/delhst.py

@@ -1,7 +1,7 @@
-from ..aamgr import AbstractCommandNode, Context
+from .. import aamgr
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="delhst",
     description="删除指定会话的所有历史记录",
@@ -9,9 +9,9 @@ from ..aamgr import AbstractCommandNode, Context
     aliases=[],
     privilege=2
 )
-class DelHistoryCommand(AbstractCommandNode):
+class DelHistoryCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import pkg.openai.session
         import pkg.utils.context
         params = ctx.params
@@ -31,7 +31,7 @@ class DelHistoryCommand(AbstractCommandNode):
         return True, reply
     
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=DelHistoryCommand,
     name="all",
     description="删除所有会话的全部历史记录",
@@ -39,9 +39,9 @@ class DelHistoryCommand(AbstractCommandNode):
     aliases=[],
     privilege=2
 )
-class DelAllHistoryCommand(AbstractCommandNode):
+class DelAllHistoryCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import pkg.utils.context
         reply = []
         pkg.utils.context.get_database_manager().delete_all_session_history()

pkg/qqbot/cmds/session/last.py

@@ -1,8 +1,9 @@
-from ..aamgr import AbstractCommandNode, Context
 import datetime
 
+from .. import aamgr
 
-@AbstractCommandNode.register(
+
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="last",
     description="切换前一次对话",
@@ -10,9 +11,9 @@ import datetime
     aliases=[],
     privilege=1
 )
-class LastCommand(AbstractCommandNode):
+class LastCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import pkg.openai.session
         session_name = ctx.session_name
 

pkg/qqbot/cmds/session/list.py

@@ -1,9 +1,10 @@
-from ..aamgr import AbstractCommandNode, Context
 import datetime
 import json
 
+from .. import aamgr
 
-@AbstractCommandNode.register(
+
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name='list',
     description='列出当前会话的所有历史记录',
@@ -11,9 +12,9 @@ import json
     aliases=[],
     privilege=1
 )
-class ListCommand(AbstractCommandNode):
+class ListCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import pkg.openai.session
         session_name = ctx.session_name
         params = ctx.params

pkg/qqbot/cmds/session/next.py

@@ -1,8 +1,9 @@
-from ..aamgr import AbstractCommandNode, Context
 import datetime
 
+from .. import aamgr
 
-@AbstractCommandNode.register(
+
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="next",
     description="切换后一次对话",
@@ -10,9 +11,9 @@ import datetime
     aliases=[],
     privilege=1
 )
-class NextCommand(AbstractCommandNode):
+class NextCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import pkg.openai.session
         session_name = ctx.session_name
         reply = []

pkg/qqbot/cmds/session/prompt.py

@@ -1,8 +1,7 @@
-from ..aamgr import AbstractCommandNode, Context
-import datetime
+from .. import aamgr
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="prompt",
     description="获取当前会话的前文",
@@ -10,9 +9,9 @@ import datetime
     aliases=[],
     privilege=1
 )
-class PromptCommand(AbstractCommandNode):
+class PromptCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import pkg.openai.session
         session_name = ctx.session_name
         params = ctx.params

pkg/qqbot/cmds/session/resend.py

@@ -1,8 +1,7 @@
-from ..aamgr import AbstractCommandNode, Context
-import datetime
+from .. import aamgr
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="resend",
     description="重新获取上一次问题的回复",
@@ -10,20 +9,22 @@ import datetime
     aliases=[],
     privilege=1
 )
-class ResendCommand(AbstractCommandNode):
+class ResendCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
-        import pkg.openai.session
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
+        from ....openai import session as openai_session
+        from ....utils import context
+        from ....qqbot import message
         import config
         session_name = ctx.session_name
         reply = []
 
-        session = pkg.openai.session.get_session(session_name)
+        session = openai_session.get_session(session_name)
         to_send = session.undo()
 
-        mgr = pkg.utils.context.get_qqbot_manager()
+        mgr = context.get_qqbot_manager()
 
-        reply = pkg.qqbot.message.process_normal_message(to_send, mgr, config,
+        reply = message.process_normal_message(to_send, mgr, config,
                                                         ctx.launcher_type, ctx.launcher_id,
                                                         ctx.sender_id)
 

pkg/qqbot/cmds/session/reset.py

@@ -1,11 +1,11 @@
-from ..aamgr import AbstractCommandNode, Context
 import tips as tips_custom
 
-import pkg.openai.session
-import pkg.utils.context
+from .. import aamgr
+from ....openai import session
+from ....utils import context
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name='reset',
     description='重置当前会话',
@@ -13,21 +13,21 @@ import pkg.utils.context
     aliases=[],
     privilege=1
 )
-class ResetCommand(AbstractCommandNode):
+class ResetCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         params = ctx.params
         session_name = ctx.session_name
         
         reply = ""
 
         if len(params) == 0:
-            pkg.openai.session.get_session(session_name).reset(explicit=True)
+            session.get_session(session_name).reset(explicit=True)
             reply = [tips_custom.command_reset_message]
         else:
             try:
                 import pkg.openai.dprompt as dprompt
-                pkg.openai.session.get_session(session_name).reset(explicit=True, use_prompt=params[0])
+                session.get_session(session_name).reset(explicit=True, use_prompt=params[0])
                 reply = [tips_custom.command_reset_name_message+"{}".format(dprompt.mode_inst().get_full_name(params[0]))]
             except Exception as e:
                 reply = ["[bot]会话重置失败：{}".format(e)]

pkg/qqbot/cmds/system/cconfig.py

@@ -1,6 +1,7 @@
-from ..aamgr import AbstractCommandNode, Context
 import json
 
+from .. import aamgr
+
 
 def config_operation(cmd, params):
     reply = []
@@ -85,7 +86,7 @@ def config_operation(cmd, params):
     return reply
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="cfg",
     description="配置项管理",
@@ -93,8 +94,8 @@ def config_operation(cmd, params):
     aliases=[],
     privilege=2
 )
-class CfgCommand(AbstractCommandNode):
+class CfgCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         return True, config_operation(ctx.command, ctx.params)
     

pkg/qqbot/cmds/system/cmd.py

@@ -1,7 +1,7 @@
-from ..aamgr import AbstractCommandNode, Context, __command_list__
+from .. import aamgr
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="cmd",
     description="显示指令列表",
@@ -9,10 +9,10 @@ from ..aamgr import AbstractCommandNode, Context, __command_list__
     aliases=[],
     privilege=1
 )
-class CmdCommand(AbstractCommandNode):
+class CmdCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
-        command_list = __command_list__
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
+        command_list = aamgr.__command_list__
 
         reply = []
 

pkg/qqbot/cmds/system/help.py

@@ -1,7 +1,7 @@
-from ..aamgr import AbstractCommandNode, Context
+from .. import aamgr
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="help",
     description="显示自定义的帮助信息",
@@ -9,9 +9,9 @@ from ..aamgr import AbstractCommandNode, Context
     aliases=[],
     privilege=1
 )
-class HelpCommand(AbstractCommandNode):
+class HelpCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import tips
         reply = ["[bot] "+tips.help_message + "\n请输入 !cmd 查看指令列表"]
 

pkg/qqbot/cmds/system/reload.py

@@ -1,7 +1,9 @@
-from ..aamgr import AbstractCommandNode, Context
 import threading
 
-@AbstractCommandNode.register(
+from .. import aamgr
+
+
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="reload",
     description="执行热重载",
@@ -9,9 +11,9 @@ import threading
     aliases=[],
     privilege=2
 )
-class ReloadCommand(AbstractCommandNode):
+class ReloadCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         reply = []
 
         import pkg.utils.reloader

pkg/qqbot/cmds/system/update.py

@@ -1,9 +1,10 @@
-from ..aamgr import AbstractCommandNode, Context
 import threading
 import traceback
 
+from .. import aamgr
 
-@AbstractCommandNode.register(
+
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="update",
     description="更新程序",
@@ -11,9 +12,9 @@ import traceback
     aliases=[],
     privilege=2
 )
-class UpdateCommand(AbstractCommandNode):
+class UpdateCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         reply = []
         import pkg.utils.updater
         import pkg.utils.reloader

pkg/qqbot/cmds/system/usage.py

@@ -1,8 +1,7 @@
-from ..aamgr import AbstractCommandNode, Context
-import logging
+from .. import aamgr
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="usage",
     description="获取使用情况",
@@ -10,9 +9,9 @@ import logging
     aliases=[],
     privilege=1
 )
-class UsageCommand(AbstractCommandNode):
+class UsageCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         import config
         import pkg.utils.credit as credit
         import pkg.utils.context

pkg/qqbot/cmds/system/version.py

@@ -1,7 +1,7 @@
-from ..aamgr import AbstractCommandNode, Context
+from .. import aamgr
 
 
-@AbstractCommandNode.register(
+@aamgr.AbstractCommandNode.register(
     parent=None,
     name="version",
     description="查看版本信息",
@@ -9,9 +9,9 @@ from ..aamgr import AbstractCommandNode, Context
     aliases=[],
     privilege=1
 )
-class VersionCommand(AbstractCommandNode):
+class VersionCommand(aamgr.AbstractCommandNode):
     @classmethod
-    def process(cls, ctx: Context) -> tuple[bool, list]:
+    def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
         reply = []
         import pkg.utils.updater
     

pkg/qqbot/command.py

@@ -1,23 +1,7 @@
 # 指令处理模块
 import logging
-import json
-import datetime
-import os
-import threading
-import traceback
-
-import pkg.openai.session
-import pkg.openai.manager
-import pkg.utils.reloader
-import pkg.utils.updater
-import pkg.utils.context
-import pkg.qqbot.message
-import pkg.utils.credit as credit
-# import pkg.qqbot.cmds.model as cmdmodel
-import pkg.qqbot.cmds.aamgr as cmdmgr
-
-from mirai import Image
 
+from ..qqbot.cmds import aamgr as cmdmgr
 
 
 def process_command(session_name: str, text_message: str, mgr, config,

pkg/qqbot/manager.py

@@ -1,32 +1,25 @@
-import asyncio
 import json
 import os
-import threading
-
+import logging
 
 from mirai import At, GroupMessage, MessageEvent, Mirai, StrangerMessage, WebSocketAdapter, HTTPAdapter, \
     FriendMessage, Image, MessageChain, Plain
-from func_timeout import func_set_timeout
+import func_timeout
 
-import pkg.openai.session
-import pkg.openai.manager
-from func_timeout import FunctionTimedOut
-import logging
+from ..openai import session as openai_session
 
-import pkg.qqbot.filter
-import pkg.qqbot.process as processor
-import pkg.utils.context
-
-import pkg.plugin.host as plugin_host
-import pkg.plugin.models as plugin_models
+from ..qqbot import filter as qqbot_filter
+from ..qqbot import process as processor
+from ..utils import context
+from ..plugin import host as plugin_host
+from ..plugin import models as plugin_models
 import tips as tips_custom
-
-import pkg.qqbot.adapter as msadapter
+from ..qqbot import adapter as msadapter
 
 
 # 检查消息是否符合泛响应匹配机制
 def check_response_rule(group_id:int, text: str):
-    config = pkg.utils.context.get_config()
+    config = context.get_config()
 
     rules = config.response_rules
 
@@ -55,7 +48,7 @@ def check_response_rule(group_id:int, text: str):
 
 
 def response_at(group_id: int):
-    config = pkg.utils.context.get_config()
+    config = context.get_config()
 
     use_response_rule = config.response_rules
 
@@ -73,7 +66,7 @@ def response_at(group_id: int):
 
 
 def random_responding(group_id):
-    config = pkg.utils.context.get_config()
+    config = context.get_config()
 
     use_response_rule = config.response_rules
 
@@ -130,10 +123,10 @@ class QQBotManager:
                 self.adapter = NakuruProjectAdapter(config.nakuru_config)
                 self.bot_account_id = self.adapter.bot_account_id
         else:
-            self.adapter = pkg.utils.context.get_qqbot_manager().adapter
-            self.bot_account_id = pkg.utils.context.get_qqbot_manager().bot_account_id
+            self.adapter = context.get_qqbot_manager().adapter
+            self.bot_account_id = context.get_qqbot_manager().bot_account_id
 
-        pkg.utils.context.set_qqbot_manager(self)
+        context.set_qqbot_manager(self)
 
         # 注册诸事件
         # Caution: 注册新的事件处理器之后，请务必在unsubscribe_all中编写相应的取消订阅代码
@@ -154,7 +147,7 @@ class QQBotManager:
 
                 self.on_person_message(event)
 
-            pkg.utils.context.get_thread_ctl().submit_user_task(
+            context.get_thread_ctl().submit_user_task(
                 friend_message_handler,
             )
         self.adapter.register_listener(
@@ -179,7 +172,7 @@ class QQBotManager:
 
                 self.on_person_message(event)
 
-            pkg.utils.context.get_thread_ctl().submit_user_task(
+            context.get_thread_ctl().submit_user_task(
                 stranger_message_handler,
             )
         # nakuru不区分好友和陌生人，故仅为yirimirai注册陌生人事件
@@ -206,7 +199,7 @@ class QQBotManager:
 
                 self.on_group_message(event)
 
-            pkg.utils.context.get_thread_ctl().submit_user_task(
+            context.get_thread_ctl().submit_user_task(
                 group_message_handler,
                 event
             )
@@ -250,22 +243,22 @@ class QQBotManager:
             if hasattr(banlist, "enable_group"):
                 self.enable_group = banlist.enable_group
 
-        config = pkg.utils.context.get_config()
+        config = context.get_config()
         if os.path.exists("sensitive.json") \
                 and config.sensitive_word_filter is not None \
                 and config.sensitive_word_filter:
             with open("sensitive.json", "r", encoding="utf-8") as f:
                 sensitive_json = json.load(f)
-                self.reply_filter = pkg.qqbot.filter.ReplyFilter(
+                self.reply_filter = qqbot_filter.ReplyFilter(
                     sensitive_words=sensitive_json['words'],
                     mask=sensitive_json['mask'] if 'mask' in sensitive_json else '*',
                     mask_word=sensitive_json['mask_word'] if 'mask_word' in sensitive_json else ''
                 )
         else:
-            self.reply_filter = pkg.qqbot.filter.ReplyFilter([])
+            self.reply_filter = qqbot_filter.ReplyFilter([])
 
     def send(self, event, msg, check_quote=True, check_at_sender=True):
-        config = pkg.utils.context.get_config()
+        config = context.get_config()
         
         if check_at_sender and config.at_sender:
             msg.insert(
@@ -306,7 +299,7 @@ class QQBotManager:
                 for i in range(self.retry):
                     try:
                         
-                        @func_set_timeout(config.process_message_timeout)
+                        @func_timeout.func_set_timeout(config.process_message_timeout)
                         def time_ctrl_wrapper():
                             reply = processor.process_message('person', event.sender.id, str(event.message_chain),
                                                             event.message_chain,
@@ -315,16 +308,16 @@ class QQBotManager:
                         
                         reply = time_ctrl_wrapper()
                         break
-                    except FunctionTimedOut:
+                    except func_timeout.FunctionTimedOut:
                         logging.warning("person_{}: 超时，重试中({})".format(event.sender.id, i))
-                        pkg.openai.session.get_session('person_{}'.format(event.sender.id)).release_response_lock()
-                        if "person_{}".format(event.sender.id) in pkg.qqbot.process.processing:
-                            pkg.qqbot.process.processing.remove('person_{}'.format(event.sender.id))
+                        openai_session.get_session('person_{}'.format(event.sender.id)).release_response_lock()
+                        if "person_{}".format(event.sender.id) in processor.processing:
+                            processor.processing.remove('person_{}'.format(event.sender.id))
                         failed += 1
                         continue
 
                 if failed == self.retry:
-                    pkg.openai.session.get_session('person_{}'.format(event.sender.id)).release_response_lock()
+                    openai_session.get_session('person_{}'.format(event.sender.id)).release_response_lock()
                     self.notify_admin("{} 请求超时".format("person_{}".format(event.sender.id)))
                     reply = [tips_custom.reply_message]
 
@@ -344,7 +337,7 @@ class QQBotManager:
             failed = 0
             for i in range(self.retry):
                 try:
-                    @func_set_timeout(config.process_message_timeout)
+                    @func_timeout.func_set_timeout(config.process_message_timeout)
                     def time_ctrl_wrapper():
                         replys = processor.process_message('group', event.group.id,
                                                         str(event.message_chain).strip() if text is None else text,
@@ -354,16 +347,16 @@ class QQBotManager:
                     
                     replys = time_ctrl_wrapper()
                     break
-                except FunctionTimedOut:
+                except func_timeout.FunctionTimedOut:
                     logging.warning("group_{}: 超时，重试中({})".format(event.group.id, i))
-                    pkg.openai.session.get_session('group_{}'.format(event.group.id)).release_response_lock()
-                    if "group_{}".format(event.group.id) in pkg.qqbot.process.processing:
-                        pkg.qqbot.process.processing.remove('group_{}'.format(event.group.id))
+                    openai_session.get_session('group_{}'.format(event.group.id)).release_response_lock()
+                    if "group_{}".format(event.group.id) in processor.processing:
+                        processor.processing.remove('group_{}'.format(event.group.id))
                     failed += 1
                     continue
 
             if failed == self.retry:
-                pkg.openai.session.get_session('group_{}'.format(event.group.id)).release_response_lock()
+                openai_session.get_session('group_{}'.format(event.group.id)).release_response_lock()
                 self.notify_admin("{} 请求超时".format("group_{}".format(event.group.id)))
                 replys = [tips_custom.replys_message]
 
@@ -392,7 +385,7 @@ class QQBotManager:
 
     # 通知系统管理员
     def notify_admin(self, message: str):
-        config = pkg.utils.context.get_config()
+        config = context.get_config()
         if config.admin_qq != 0 and config.admin_qq != []:
             logging.info("通知管理员:{}".format(message))
             if type(config.admin_qq) == int:
@@ -410,7 +403,7 @@ class QQBotManager:
                     )
 
     def notify_admin_message_chain(self, message):
-        config = pkg.utils.context.get_config()
+        config = context.get_config()
         if config.admin_qq != 0 and config.admin_qq != []:
             logging.info("通知管理员:{}".format(message))
             if type(config.admin_qq) == int:

pkg/qqbot/message.py

@@ -1,19 +1,20 @@
 # 普通消息处理模块
 import logging
-import openai
-import pkg.utils.context
-import pkg.openai.session
 
-import pkg.plugin.host as plugin_host
-import pkg.plugin.models as plugin_models
-import pkg.qqbot.blob as blob
+import openai
+
+from ..utils import context
+from ..openai import session as openai_session
+
+from ..plugin import host as plugin_host
+from ..plugin import models as plugin_models
 import tips as tips_custom
 
 
 def handle_exception(notify_admin: str = "", set_reply: str = "") -> list:
     """处理异常，当notify_admin不为空时，会通知管理员，返回通知用户的消息"""
     import config
-    pkg.utils.context.get_qqbot_manager().notify_admin(notify_admin)
+    context.get_qqbot_manager().notify_admin(notify_admin)
     if config.hide_exce_info_to_user:
         return [tips_custom.alter_tip_message] if tips_custom.alter_tip_message else []
     else:
@@ -26,7 +27,7 @@ def process_normal_message(text_message: str, mgr, config, launcher_type: str,
     logging.info("[{}]发送消息:{}".format(session_name, text_message[:min(20, len(text_message))] + (
         "..." if len(text_message) > 20 else "")))
 
-    session = pkg.openai.session.get_session(session_name)
+    session = openai_session.get_session(session_name)
 
     unexpected_exception_times = 0
 
@@ -54,7 +55,7 @@ def process_normal_message(text_message: str, mgr, config, launcher_type: str,
                 "funcs_called": funcs,
             }
 
-            event = pkg.plugin.host.emit(plugin_models.NormalMessageResponded, **args)
+            event = plugin_host.emit(plugin_models.NormalMessageResponded, **args)
 
             if event.get_return_value("prefix") is not None:
                 prefix = event.get_return_value("prefix")
@@ -78,29 +79,29 @@ def process_normal_message(text_message: str, mgr, config, launcher_type: str,
 
             if 'message' in e.error and e.error['message'].__contains__('You exceeded your current quota'):
                 # 尝试切换api-key
-                current_key_name = pkg.utils.context.get_openai_manager().key_mgr.get_key_name(
-                    pkg.utils.context.get_openai_manager().key_mgr.using_key
+                current_key_name = context.get_openai_manager().key_mgr.get_key_name(
+                    context.get_openai_manager().key_mgr.using_key
                 )
-                pkg.utils.context.get_openai_manager().key_mgr.set_current_exceeded()
+                context.get_openai_manager().key_mgr.set_current_exceeded()
 
                 # 触发插件事件
                 args = {
                     'key_name': current_key_name,
-                    'usage': pkg.utils.context.get_openai_manager().audit_mgr
-                            .get_usage(pkg.utils.context.get_openai_manager().key_mgr.get_using_key_md5()),
-                    'exceeded_keys': pkg.utils.context.get_openai_manager().key_mgr.exceeded,
+                    'usage': context.get_openai_manager().audit_mgr
+                            .get_usage(context.get_openai_manager().key_mgr.get_using_key_md5()),
+                    'exceeded_keys': context.get_openai_manager().key_mgr.exceeded,
                 }
                 event = plugin_host.emit(plugin_models.KeyExceeded, **args)
 
                 if not event.is_prevented_default():
-                    switched, name = pkg.utils.context.get_openai_manager().key_mgr.auto_switch()
+                    switched, name = context.get_openai_manager().key_mgr.auto_switch()
 
                     if not switched:
                         reply = handle_exception(
                             "api-key调用额度超限({}),无可用api_key,请向OpenAI账户充值或在config.py中更换api_key；如果你认为这是误判，请尝试重启程序。".format(
                                 current_key_name), "[bot]err:API调用额度超额，请联系管理员，或等待修复")
                     else:
-                        openai.api_key = pkg.utils.context.get_openai_manager().key_mgr.get_using_key()
+                        openai.api_key = context.get_openai_manager().key_mgr.get_using_key()
                         mgr.notify_admin("api-key调用额度超限({}),接口报错,已切换到{}".format(current_key_name, name))
                         reply = ["[bot]err:API调用额度超额，已自动切换，请重新发送消息"]
                         continue

pkg/qqbot/process.py

@@ -5,28 +5,22 @@ import time
 import mirai
 import logging
 
-from mirai import MessageChain, Plain
-
 # 这里不使用动态引入config
 # 因为在这里动态引入会卡死程序
 # 而此模块静态引用config与动态引入的表现一致
 # 已弃用，由于超时时间现已动态使用
 # import config as config_init_import
 
-import pkg.openai.session
-import pkg.openai.manager
-import pkg.utils.reloader
-import pkg.utils.updater
-import pkg.utils.context
-import pkg.qqbot.message
-import pkg.qqbot.command
-import pkg.qqbot.ratelimit as ratelimit
+from ..qqbot import ratelimit
+from ..qqbot import command, message
+from ..openai import session as openai_session
+from ..utils import context
 
-import pkg.plugin.host as plugin_host
-import pkg.plugin.models as plugin_models
-import pkg.qqbot.ignore as ignore
-import pkg.qqbot.banlist as banlist
-import pkg.qqbot.blob as blob
+from ..plugin import host as plugin_host
+from ..plugin import models as plugin_models
+from ..qqbot import ignore
+from ..qqbot import banlist
+from ..qqbot import blob
 import tips as tips_custom
 
 processing = []
@@ -41,11 +35,11 @@ def is_admin(qq: int) -> bool:
         return qq == config.admin_qq
 
 
-def process_message(launcher_type: str, launcher_id: int, text_message: str, message_chain: MessageChain,
-                    sender_id: int) -> MessageChain:
+def process_message(launcher_type: str, launcher_id: int, text_message: str, message_chain: mirai.MessageChain,
+                    sender_id: int) -> mirai.MessageChain:
     global processing
 
-    mgr = pkg.utils.context.get_qqbot_manager()
+    mgr = context.get_qqbot_manager()
 
     reply = []
     session_name = "{}_{}".format(launcher_type, launcher_id)
@@ -62,7 +56,7 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
     import config
 
     if not config.wait_last_done and session_name in processing:
-        return MessageChain([Plain(tips_custom.message_drop_tip)])
+        return mirai.MessageChain([mirai.Plain(tips_custom.message_drop_tip)])
 
     # 检查是否被禁言
     if launcher_type == 'group':
@@ -74,9 +68,9 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
     import config
     if config.income_msg_check:
         if mgr.reply_filter.is_illegal(text_message):
-            return MessageChain(Plain("[bot] 消息中存在不合适的内容, 请更换措辞"))
+            return mirai.MessageChain(mirai.Plain("[bot] 消息中存在不合适的内容, 请更换措辞"))
 
-    pkg.openai.session.get_session(session_name).acquire_response_lock()
+    openai_session.get_session(session_name).acquire_response_lock()
 
     text_message = text_message.strip()
 
@@ -87,7 +81,7 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
     # 处理消息
     try:
 
-        config = pkg.utils.context.get_config()
+        config = context.get_config()
 
         processing.append(session_name)
         try:
@@ -114,7 +108,7 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
                     reply = event.get_return_value("reply")
 
                 if not event.is_prevented_default():
-                    reply = pkg.qqbot.command.process_command(session_name, text_message,
+                    reply = command.process_command(session_name, text_message,
                                                               mgr, config, launcher_type, launcher_id, sender_id, is_admin(sender_id))
 
             else:  # 消息
@@ -124,7 +118,7 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
                     if ratelimit.is_reach_limit(session_name):
                         logging.info("根据限速策略丢弃[{}]消息: {}".format(session_name, text_message))
 
-                        return MessageChain(["[bot]"+tips_custom.rate_limit_drop_tip]) if tips_custom.rate_limit_drop_tip != "" else []
+                        return mirai.MessageChain(["[bot]"+tips_custom.rate_limit_drop_tip]) if tips_custom.rate_limit_drop_tip != "" else []
 
                 before = time.time()
                 # 触发插件事件
@@ -146,7 +140,7 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
                     reply = event.get_return_value("reply")
 
                 if not event.is_prevented_default():
-                    reply = pkg.qqbot.message.process_normal_message(text_message,
+                    reply = message.process_normal_message(text_message,
                                                                      mgr, config, launcher_type, launcher_id, sender_id)
 
                 # 限速等待时间
@@ -170,7 +164,7 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
         finally:
             processing.remove(session_name)
     finally:
-        pkg.openai.session.get_session(session_name).release_response_lock()
+        openai_session.get_session(session_name).release_response_lock()
 
     # 检查延迟时间
     if config.force_delay_range[1] == 0:
@@ -191,4 +185,4 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
         logging.info("[风控] 强制延迟{:.2f}秒(如需关闭，请到config.py修改force_delay_range字段)".format(delay_time))
         time.sleep(delay_time)
 
-    return MessageChain(reply)
+    return mirai.MessageChain(reply)

pkg/qqbot/sources/nakuru.py

@@ -1,19 +1,18 @@
-import mirai
-
-from ..adapter import MessageSourceAdapter, MessageConverter, EventConverter
-import nakuru
-import nakuru.entities.components as nkc
-
 import asyncio
 import typing
 import traceback
 import logging
-import json
 
-from pkg.qqbot.blob import Forward, ForwardMessageNode, ForwardMessageDiaplay
+import mirai
+
+import nakuru
+import nakuru.entities.components as nkc
+
+from .. import adapter as adapter_model
+from ...qqbot import blob
 
 
-class NakuruProjectMessageConverter(MessageConverter):
+class NakuruProjectMessageConverter(adapter_model.MessageConverter):
     """消息转换器"""
     @staticmethod
     def yiri2target(message_chain: mirai.MessageChain) -> list:
@@ -49,7 +48,7 @@ class NakuruProjectMessageConverter(MessageConverter):
                     nakuru_msg_list.append(nkc.Record.fromURL(component.url))
                 elif component.path is not None:
                     nakuru_msg_list.append(nkc.Record.fromFileSystem(component.path))
-            elif type(component) is Forward:
+            elif type(component) is blob.Forward:
                 # 转发消息
                 yiri_forward_node_list = component.node_list
                 nakuru_forward_node_list = []
@@ -102,7 +101,7 @@ class NakuruProjectMessageConverter(MessageConverter):
         return chain
 
 
-class NakuruProjectEventConverter(EventConverter):
+class NakuruProjectEventConverter(adapter_model.EventConverter):
     """事件转换器"""
     @staticmethod
     def yiri2target(event: typing.Type[mirai.Event]):
@@ -157,7 +156,7 @@ class NakuruProjectEventConverter(EventConverter):
             raise Exception("未支持转换的事件类型: " + str(event))
 
 
-class NakuruProjectAdapter(MessageSourceAdapter):
+class NakuruProjectAdapter(adapter_model.MessageSourceAdapter):
     """nakuru-project适配器"""
     bot: nakuru.CQHTTP
     bot_account_id: int

pkg/qqbot/sources/yirimirai.py

@@ -1,13 +1,14 @@
-from ..adapter import MessageSourceAdapter
+import asyncio
+import typing
+
 import mirai
 import mirai.models.bus
 from mirai.bot import MiraiRunner
 
-import asyncio
-import typing
+from .. import adapter as adapter_model
 
 
-class YiriMiraiAdapter(MessageSourceAdapter):
+class YiriMiraiAdapter(adapter_model.MessageSourceAdapter):
     """YiriMirai适配器"""
     bot: mirai.Mirai