🍻 format spec for UniMessage and Segment

Author: RF-Tar-Railt

src/nonebot_plugin_alconna/uniseg/adapters/onebot11/builder.py

@@ -16,9 +16,11 @@ def get_adapter(cls) -> SupportAdapter:
 
     @build("at")
     def at(self, seg: MessageSegment):
-        if seg.data["qq"] != "all":
-            return At("user", str(seg.data["qq"]))
-        return AtAll()
+        if seg.data["qq"] == "all":
+            return AtAll()
+        if int(seg.data["qq"]) == 0:
+            return AtAll()
+        return At("user", str(seg.data["qq"]))
 
     @build("face")
     def face(self, seg: MessageSegment):

src/nonebot_plugin_alconna/uniseg/message.py

@@ -626,6 +626,11 @@ def __str__(self) -> str:
     def __repr__(self) -> str:
         return "[" + ", ".join(repr(seg) for seg in self) + "]"
 
+    def __format__(self, format_spec):
+        if format_spec in ("#", "*"):
+            return "".join(format(seg, format_spec) for seg in self)
+        return format(self.__str__(), format_spec)
+
     @classmethod
     def template(cls, format_string: str | UniMessage) -> UniMessageTemplate:
         """创建消息模板。

src/nonebot_plugin_alconna/uniseg/segment.py

@@ -53,6 +53,16 @@ class Segment:
     def __str__(self):
         return f"[{self.__class__.__name__.lower()}]"
 
+    def __format__(self, format_spec):
+        if format_spec == "*":
+            if not isinstance(self, Text):
+                attrs = ",".join(f"{k}={v}" for k, v in self.data.items() if v is not None)
+                return f"[{self.__class__.__name__.lower()}:{attrs}]"
+            return self.__str__()
+        if format_spec == "#":
+            return self.__rich__() if isinstance(self, Text) else self.__str__()
+        return format(self.__str__(), format_spec)
+
     @overload
     def __add__(self: TS, item: str) -> "UniMessage[Union[TS, Text]]": ...
 
@@ -161,15 +171,16 @@ def load(cls, data: dict) -> Self:
     "underline": "4",
     "flash": "5",
     "blink": "5",
-    "rapid": "6",
     "sparkle": "5",
+    "rapid": "6",
     "code": "7",
     "inverse": "7",
     "spoiler": "8",
     "invisible": "8",
     "strikethrough": "9",
     "delete": "9",
     "cross": "9",
+    "normal": "22",
     "black": "30",
     "red": "31",
     "green": "32",
@@ -350,7 +361,8 @@ def __rich__(self):
         result.append(text[:left])
         for scale in scales:
             prefix = ";".join(f"{STYLE_TYPE_MAP[style]}" for style in styles[scale])
-            result.append(f"\033[{prefix}m{text[scale[0] : scale[1]]}\033[0m")
+            result.append(f"\033[{prefix}m{text[scale[0] : scale[1]]}")
+        result.append("\033[0m")
         right = scales[-1][1]
         result.append(text[right:])
         return "".join(result)

tests/test_uniseg.py

@@ -9,6 +9,11 @@
 from tests.fake import fake_group_message_event_v11
 
 
+def ansi(*code) -> str:
+    codes = ";".join(str(c) for c in code)
+    return f"\033[{codes}m"
+
+
 def test_uniseg():
     from nonebot_plugin_alconna.uniseg import FallbackSegment
     from nonebot_plugin_alconna import Text, Other, Video, Segment, select
@@ -28,7 +33,7 @@ def test_uniseg():
         Text("man").italic(0, 1),
     ]
     assert text1.replace("o", "e") == Text("helle werld man").color("red", 0, -3).italic(3, -2)
-
+    assert f"{text1:#}" == f"{ansi(31)}hel{ansi(31, 3)}lo world {ansi(3)}m{ansi(0)}an"
     pat = select(Text)
     assert pat.first.validate(Text("foobar")).value() == Text("foobar")
     assert pat.first.validate(Video(url="foobar")(Text("foobar"))).value() == Text("foobar")
@@ -60,6 +65,7 @@ def test_unimsg():
         "At(flag='channel', target='456', display=None), "
         "At(flag='role', target='789', display=None)]"
     )
+    assert f"{msg1:*}" == "[at:flag=user,target=123][at:flag=channel,target=456][at:flag=role,target=789]"
     assert msg1.select(At).filter(lambda x: x.flag == "user") == UniMessage.at("123")
     assert msg1.select(At).map(lambda x: x.target) == ["123", "456", "789"]
     assert UniMessage.text("123").at("456").export_sync(adapter="OneBot V11") == MessageSegment.text(