mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 20:14:36 +00:00
73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
# 长消息处理相关
|
|
import config
|
|
from mirai.models.message import MessageComponent, MessageChain
|
|
from mirai.models.message import ForwardMessageNode
|
|
from mirai.models.base import MiraiBaseModel
|
|
from typing import List, Optional
|
|
import pkg.utils.context as context
|
|
|
|
|
|
class ForwardMessageDiaplay(MiraiBaseModel):
|
|
title: str = "群聊的聊天记录"
|
|
brief: str = "[聊天记录]"
|
|
source: str = "聊天记录"
|
|
preview: List[str] = []
|
|
summary: str = "查看x条转发消息"
|
|
|
|
|
|
class Forward(MessageComponent):
|
|
"""合并转发。"""
|
|
type: str = "Forward"
|
|
"""消息组件类型。"""
|
|
display: ForwardMessageDiaplay
|
|
"""显示信息"""
|
|
node_list: List[ForwardMessageNode]
|
|
"""转发消息节点列表。"""
|
|
def __init__(self, *args, **kwargs):
|
|
if len(args) == 1:
|
|
self.node_list = args[0]
|
|
super().__init__(**kwargs)
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def __str__(self):
|
|
return '[聊天记录]'
|
|
|
|
|
|
|
|
def check_text(text: str) -> list:
|
|
"""检查文本是否为长消息，并转换成该使用的消息链组件"""
|
|
if not hasattr(config, 'blob_message_threshold'):
|
|
return [text]
|
|
|
|
if len(text) > config.blob_message_threshold:
|
|
if not hasattr(config, 'blob_message_strategy'):
|
|
raise AttributeError('未定义长消息处理策略')
|
|
|
|
if config.blob_message_strategy == 'image':
|
|
# 转换成图片
|
|
pass
|
|
elif config.blob_message_strategy == 'forward':
|
|
# 敏感词屏蔽
|
|
text = context.get_qqbot_manager().reply_filter.process(text)
|
|
|
|
# 包装转发消息
|
|
display = ForwardMessageDiaplay(
|
|
title='群聊的聊天记录',
|
|
brief='[聊天记录]',
|
|
source='聊天记录',
|
|
preview=["bot: "+text],
|
|
summary="查看1条转发消息"
|
|
)
|
|
|
|
node = ForwardMessageNode(
|
|
sender_id=config.mirai_http_api_config['qq'],
|
|
sender_name='bot',
|
|
message_chain=MessageChain([text])
|
|
)
|
|
|
|
forward = Forward(
|
|
display=display,
|
|
node_list=[node]
|
|
)
|
|
|
|
return [forward] |