xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-11 16:26:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: 请求处理控制流基础架构

Browse Source
This commit is contained in:
RockChinQ
2024-01-26 15:51:49 +08:00
parent a064c24f60
commit 8d084427d2
55 changed files with 1430 additions and 146 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/qqbot/bansess/bansess.py
View File

@@ -3,7 +3,7 @@
from __future__ import annotations
import re
from ...boot import app
from ...core import app
from ...config import manager as cfg_mgr

2
pkg/qqbot/cntfilter/cntfilter.py
View File

@@ -1,6 +1,6 @@
from __future__ import annotations
from ...boot import app
from ...core import app
from . import entities
from . import filter
from .filters import cntignore, banwords, baiduexamine

2
pkg/qqbot/cntfilter/filter.py
View File

@@ -2,7 +2,7 @@
from __future__ import annotations
import abc
from ...boot import app
from ...core import app
from . import entities

2
pkg/qqbot/longtext/longtext.py
View File

@@ -5,7 +5,7 @@ import traceback
from PIL import Image, ImageDraw, ImageFont
from mirai.models.message import MessageComponent, Plain
from ...boot import app
from ...core import app
from . import strategy
from .strategies import image, forward

2
pkg/qqbot/longtext/strategy.py
View File

@@ -5,7 +5,7 @@ import typing
import mirai
from mirai.models.message import MessageComponent
from ...boot import app
from ...core import app
class LongTextStrategy(metaclass=abc.ABCMeta):

92
pkg/qqbot/manager.py
View File

@@ -24,7 +24,7 @@ from .cntfilter import cntfilter
from .longtext import longtext
from .ratelim import ratelim
from ..boot import app
from ..core import app, entities as core_entities
# 控制QQ消息输入输出的类
@@ -91,45 +91,29 @@ class QQBotManager:
# Caution: 注册新的事件处理器之后，请务必在unsubscribe_all中编写相应的取消订阅代码
async def on_friend_message(event: FriendMessage):
async def friend_message_handler():
# 触发事件
args = {
"launcher_type": "person",
"launcher_id": event.sender.id,
"sender_id": event.sender.id,
"message_chain": event.message_chain,
}
plugin_event = plugin_host.emit(plugin_models.PersonMessageReceived, **args)
await self.ap.query_pool.add_query(
launcher_type=core_entities.LauncherTypes.PERSON,
launcher_id=event.sender.id,
sender_id=event.sender.id,
message_event=event,
message_chain=event.message_chain
)
if plugin_event.is_prevented_default():
return
await self.on_person_message(event)
asyncio.create_task(friend_message_handler())
self.adapter.register_listener(
FriendMessage,
on_friend_message
)
async def on_stranger_message(event: StrangerMessage):
await self.ap.query_pool.add_query(
launcher_type=core_entities.LauncherTypes.PERSON,
launcher_id=event.sender.id,
sender_id=event.sender.id,
message_event=event,
message_chain=event.message_chain
)
async def stranger_message_handler():
# 触发事件
args = {
"launcher_type": "person",
"launcher_id": event.sender.id,
"sender_id": event.sender.id,
"message_chain": event.message_chain,
}
plugin_event = plugin_host.emit(plugin_models.PersonMessageReceived, **args)
if plugin_event.is_prevented_default():
return
await self.on_person_message(event)
asyncio.create_task(stranger_message_handler())
# nakuru不区分好友和陌生人，故仅为yirimirai注册陌生人事件
if config['msg_source_adapter'] == 'yirimirai':
self.adapter.register_listener(
@@ -139,49 +123,19 @@ class QQBotManager:
async def on_group_message(event: GroupMessage):
async def group_message_handler(event: GroupMessage):
# 触发事件
args = {
"launcher_type": "group",
"launcher_id": event.group.id,
"sender_id": event.sender.id,
"message_chain": event.message_chain,
}
plugin_event = plugin_host.emit(plugin_models.GroupMessageReceived, **args)
if plugin_event.is_prevented_default():
return
await self.on_group_message(event)
asyncio.create_task(group_message_handler(event))
await self.ap.query_pool.add_query(
launcher_type=core_entities.LauncherTypes.GROUP,
launcher_id=event.group.id,
sender_id=event.sender.id,
message_event=event,
message_chain=event.message_chain
)
self.adapter.register_listener(
GroupMessage,
on_group_message
)
def unsubscribe_all():
"""取消所有订阅
用于在热重载流程中卸载所有事件处理器
"""
self.adapter.unregister_listener(
FriendMessage,
on_friend_message
)
if config['msg_source_adapter'] == 'yirimirai':
self.adapter.unregister_listener(
StrangerMessage,
on_stranger_message
)
self.adapter.unregister_listener(
GroupMessage,
on_group_message
)
self.unsubscribe_all = unsubscribe_all
async def send(self, event, msg, check_quote=True, check_at_sender=True):
config = context.get_config_manager().data

2
pkg/qqbot/process.py
View File

@@ -14,7 +14,7 @@ from ..utils import context
from ..plugin import host as plugin_host
from ..plugin import models as plugin_models
import tips as tips_custom
from ..boot import app
from ..core import app
from .cntfilter import entities
processing = []

2
pkg/qqbot/ratelim/algo.py
View File

@@ -1,7 +1,7 @@
from __future__ import annotations
import abc
from ...boot import app
from ...core import app
class ReteLimitAlgo(metaclass=abc.ABCMeta):

2
pkg/qqbot/ratelim/ratelim.py
View File

@@ -2,7 +2,7 @@ from __future__ import annotations
from . import algo
from .algos import fixedwin
from ...boot import app
from ...core import app
class RateLimiter:

2
pkg/qqbot/resprule/resprule.py
View File

@@ -2,7 +2,7 @@ from __future__ import annotations
import mirai
from ...boot import app
from ...core import app
from . import entities, rule
from .rules import atbot, prefix, regexp, random

2
pkg/qqbot/resprule/rule.py
View File

@@ -3,7 +3,7 @@ import abc
import mirai
from ...boot import app
from ...core import app
from . import entities