suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2025-06-08 12:53:00 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add custom_post

Browse Source
This commit is contained in:
felinae98 2022-05-25 21:26:00 +08:00
parent 8720ff4d2b
commit 1bec4f9933
No known key found for this signature in database
GPG Key ID: 00C8B010587FF610
2 changed files with 30 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/plugins/nonebot_bison/post/abstract_post.py
View File

@ -42,9 +42,14 @@ class AbstractPost(OptionalMixin, BasePost):
msg_segments = await self.generate_pic_messages() msg_segments = await self.generate_pic_messages()
else: else:
msg_segments = await self.generate_text_messages() msg_segments = await self.generate_text_messages()
if self.compress: if msg_segments:
msgs = [reduce(lambda x, y: x.append(y), msg_segments, Message())] if self.compress:
msgs = [reduce(lambda x, y: x.append(y), msg_segments, Message())]
else:
msgs = list(
map(lambda msg_segment: Message([msg_segment]), msg_segments)
)
else: else:
msgs = list(map(lambda msg_segment: Message([msg_segment]), msg_segments)) msgs = []
msgs.extend(self.extra_msg) msgs.extend(self.extra_msg)
return msgs return msgs

22
src/plugins/nonebot_bison/post/custom_post.py Normal file
View File

@ -0,0 +1,22 @@
from dataclasses import dataclass
from nonebot.adapters.onebot.v11.message import Message, MessageSegment
from .abstract_post import AbstractPost, BasePost
@dataclass
class _CustomPost(BasePost):
message_segments: list[MessageSegment]
async def generate_text_messages(self) -> list[MessageSegment]:
return self.message_segments
async def generate_pic_messages(self) -> list[MessageSegment]:
... # TODO
@dataclass
class CustomPost(_CustomPost, AbstractPost):
...