💩 should implement rich text parser

This commit is contained in:
StarHeartHunt 2021-07-07 22:36:08 +08:00
parent e490be9eb0
commit fe43c8d69a

View File

@ -1,4 +1,5 @@
import json import json
import itertools
from dataclasses import dataclass from dataclasses import dataclass
from typing import Any, Dict, Tuple, Type, Union, Mapping, Iterable from typing import Any, Dict, Tuple, Type, Union, Mapping, Iterable
@ -18,8 +19,17 @@ class MessageSegment(BaseMessageSegment["Message"]):
return Message return Message
def __str__(self) -> str: def __str__(self) -> str:
if self.type == "text" or self.type == "hongbao": if self.type == "post":
return "".join(
str(MessageSegment(seg["tag"], seg))
for seg in itertools.chain(*self.data["content"]))
elif self.type == "text" or self.type == "hongbao":
return str(self.data["text"]) return str(self.data["text"])
elif self.type == "img" or self.type == "image":
return "[图片]"
return "" return ""
@overrides(BaseMessageSegment) @overrides(BaseMessageSegment)