✨ more convenient way to write i18n templates

Author: RF-Tar-Railt

src/nonebot_plugin_alconna/uniseg/message.py

@@ -1296,7 +1296,7 @@ def _handle_i18n(self, extra: dict, *args, **kwargs):
             if not isinstance(seg, I18n):
                 self.append(seg)
             else:
-                msg = seg.tp().format(*args, *seg.args, **kwargs, **seg.kwargs, **extra)
+                msg = self.template(str(seg)).format(*args, *seg.args, **kwargs, **seg.kwargs, **extra)
                 if msg.has(I18n):
                     msg._handle_i18n(extra, *seg.args, **seg.kwargs)
                 self.extend(msg)  # type: ignore

src/nonebot_plugin_alconna/uniseg/segment.py

@@ -985,10 +985,8 @@ def __init__(
         self.kwargs = mapping or {}
         self.kwargs.update(kwargs)
 
-    def tp(self):
-        from .message import UniMessage
-
-        return UniMessage.template(lang.require(self.item.scope, self.item.type))
+    def __str__(self):
+        return lang.require(self.item.scope, self.item.type)
 
     def dump(self, **kwargs):
         return {

src/nonebot_plugin_alconna/uniseg/template.py

@@ -5,6 +5,7 @@
 from collections.abc import Mapping, Sequence
 from typing import TYPE_CHECKING, Any, Union, TypeVar, Callable, Optional, cast
 
+from tarina import lang
 import _string  # type: ignore
 from tarina.tools import gen_subclass
 
@@ -18,6 +19,7 @@
 
 _MAPPING = {cls.__name__: cls for cls in gen_subclass(Segment)}
 _PATTERN = re.compile("(" + "|".join(_MAPPING.keys()) + r")\((.*)\)$")
+_I18N_PATTERN = re.compile(r"[^@]+\s*@\s*[^@]+")
 
 
 def _eval(route: str, obj: Any):
@@ -150,6 +152,27 @@ def _vformat(
 
             # if there's a field, output it
             if field_name is not None:
+                if mat := _I18N_PATTERN.match(field_name):
+                    scope, key = mat[0].split("@")
+                    ans = lang.require(scope.strip(), key.strip())
+                    if format_spec:
+                        _kwargs = {}
+                        for part in format_spec.split(","):
+                            if re.match(".+=.+", part):
+                                k, v = part.split("=")
+                                if v in kwargs:
+                                    _kwargs[k] = kwargs[v]
+                                    used_args.add(v)
+                                elif v.startswith("$") and (key := v.split(".")[0]) in kwargs:
+                                    _kwargs[k] = _eval(v[1:], kwargs[key])
+                                else:
+                                    _kwargs[k] = v
+                            elif part in kwargs:
+                                _kwargs[part] = kwargs[part]
+                        if _kwargs:
+                            ans = ans.format_map(_kwargs)
+                    results.append(ans)
+                    continue
                 if field_name == "" and format_spec and (mat := _PATTERN.match(format_spec)):
                     cls, parts = _MAPPING[mat[1]], mat[2].split(",")
                     _args = []

tests/test_i18n.py

@@ -9,7 +9,7 @@
 
 @pytest.mark.asyncio()
 async def test_send(app: App):
-    from nonebot_plugin_alconna import lang, on_alconna
+    from nonebot_plugin_alconna import UniMessage, lang, on_alconna
     from nonebot_plugin_alconna.uniseg.segment import I18n
 
     cmd = on_alconna(Alconna("test", Args["name", str]))
@@ -32,7 +32,7 @@ async def test_send(app: App):
                 "command.test.1": "test!",
                 "command.test.2": "{:At(user, $event.get_user_id())} hello!",
                 "command.test.3": "This is {abcd} test!",
-                "command.test.4": "This is nested: {:I18n(test-i18n, command.test.1)}",
+                "command.test.4": "This is nested: {test-i18n@command.test.1}",
             }
         },
     )
@@ -42,6 +42,7 @@ async def _():
         await cmd.send(I18n("test-i18n", "command.test.1"))
         await cmd.send(I18n("test-i18n", "command.test.2"))
         await cmd.send(cmd.i18n("test-i18n", "command.test.3", abcd="test"))
+        await cmd.send(UniMessage.template("{test-i18n @ command.test.3:abcd=test1}"))
         await cmd.finish(cmd.i18n("test-i18n", "command.test.4"))
 
     lang.select("zh-CN")
@@ -53,6 +54,7 @@ async def _():
         ctx.should_call_send(event, Message("测试!"))
         ctx.should_call_send(event, MessageSegment.mention_user("5678") + " 你好!")
         ctx.should_call_send(event, Message("这是 test 测试!"))
+        ctx.should_call_send(event, Message("这是 test1 测试!"))
         ctx.should_call_send(event, Message("这是嵌套: 测试!"))
         ctx.should_finished()
 
@@ -65,6 +67,7 @@ async def _():
         ctx.should_call_send(event, Message("test!"))
         ctx.should_call_send(event, MessageSegment.mention_user("5678") + " hello!")
         ctx.should_call_send(event, Message("This is test test!"))
+        ctx.should_call_send(event, Message("This is test1 test!"))
         ctx.should_call_send(event, Message("This is nested: test!"))
         ctx.should_finished()