ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-01-31 23:51:28 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

♻️ improve segment-string parser

Browse Source
This commit is contained in:
StarHeartHunt 2021-07-08 14:52:34 +08:00
parent 1222972b73
commit f597303698
1 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -18,14 +18,31 @@ class MessageSegment(BaseMessageSegment["Message"]):
def get_message_class(cls) -> Type["Message"]: def get_message_class(cls) -> Type["Message"]:
return Message return Message
@property
def segment_text(self) -> dict:
return {
"image": "[图片]",
"file": "[文件]",
"audio": "[音频]",
"media": "[视频]",
"sticker": "[表情包]",
"interactive": "[消息卡片]",
"hongbao": "[红包]",
"share_calendar_event": "[日程卡片]",
"share_chat": "[群名片]",
"share_user": "[个人名片]",
"system": "[系统消息]",
"location": "[位置]",
"video_chat": "[视频通话]"
}
def __str__(self) -> str: def __str__(self) -> str:
if self.type == "text" or self.type == "hongbao": if self.type in ["text", "hongbao", "a"]:
return str(self.data["text"]) return str(self.data["text"])
elif self.type == "at":
elif self.type == "image": return str(f"@{self.data['user_name']}")
return "[图片]" else:
return self.segment_text.get(self.type, "")
return ""
@overrides(BaseMessageSegment) @overrides(BaseMessageSegment)
def __add__(self, other) -> "Message": def __add__(self, other) -> "Message":