✨ lang.dispatch(scope)

src/tarina/lang/__init__.py

@@ -157,6 +157,19 @@ def _get_scopes(root: Path) -> dict[str, dict[str, dict[str, str]]]:
     return result
 
 
+class _LangScope:
+    def __init__(self, scope: str):
+        self.scope = scope
+
+    def require(self, type: str, subtype: str | None = None, locale: str | None = None):
+        if subtype:
+            type = f"{type}.{subtype}"
+        return lang.require(self.scope, type, locale)
+
+    def set(self, type: str, content: str, locale: str | None = None):
+        return lang.set(self.scope, type, content, locale)
+
+
 @final
 class _LangConfig:
     def __init__(self):
@@ -296,6 +309,9 @@ def set(self, scope: str, type: str, content: str, locale: str | None = None):
             raise ValueError(self.__langs[locale]["lang"]["error.type"].format(target=type, locale=locale, scope=scope))
         self.__langs[locale].setdefault(scope, {})[type] = content
 
+    def dispatch(self, scope: str) -> _LangScope:
+        return _LangScope(scope)
+
     def __repr__(self):
         return f"<LangConfig: {self.__locale}>"
 

tests/test_main.py

@@ -159,6 +159,9 @@ def test_lang():
     lang.select("test")
     assert lang.require("lang", "error.type") == "test"
     assert lang.require("lang", "error.locale") == "'{target}' 不是合法的语种"
+    scope_lang = lang.dispatch("lang")
+    assert scope_lang.require("error.type") == "test"
+    assert scope_lang.require("error.locale") == "'{target}' 不是合法的语种"
     lang.load_file(Path(__file__).parent / "en-UK.yml")
     assert lang.locales == {"zh-CN", "test", "en-US", "en-UK"}