ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2024-11-28 08:12:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✨ implement rich text serializer

Browse Source
This commit is contained in:
StarHeartHunt 2021-07-08 16:12:43 +08:00
parent f597303698
commit 7c7994e2e2
1 changed files with 26 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
packages/nonebot-adapter-feishu/nonebot/adapters/feishu/message.py
View File

@ -58,6 +58,15 @@ class MessageSegment(BaseMessageSegment["Message"]):
def is_text(self) -> bool:
return self.type == "text"
#接收消息
@staticmethod
def at(user_id: str, user_name: str) -> "MessageSegment":
return MessageSegment("at", {
"user_id": user_id,
"user_name": user_name
})
#发送消息
@staticmethod
def text(text: str) -> "MessageSegment":
return MessageSegment("text", {"text": text})
@ -179,7 +188,23 @@ class MessageSerializer:
message: Message
def serialize(self) -> Tuple[str, str]:
return self.message[0].type, json.dumps(self.message[0].data)
segments = list(self.message)
last_segment_type: str = ""
if len(segments) > 1:
msg = {"title": "", "content": [[]]}
for segment in segments:
if segment == "image":
if last_segment_type != "image":
msg["content"].append([])
else:
if last_segment_type == "image":
msg["content"].append([])
msg["content"][-1].append({"tag": segment.type, **segment.data})
last_segment_type = segment.type
return "post", json.dumps(msg)
else:
return self.message[0].type, json.dumps(self.message[0].data)
@dataclass