move jinja template tag module

Author: Archmonger

docs/source/about/changelog.rst

@@ -18,7 +18,7 @@ Unreleased
 **Added**
 - :pull:`1113` - Added ``reactpy.ReactPy`` that can be used to run ReactPy in standalone mode.
 - :pull:`1113` - Added ``reactpy.ReactPyMiddleware`` that can be used to run ReactPy with any ASGI compatible framework.
-- :pull:`1113` - Added ``reactpy.jinja.Component`` that can be used alongside ``ReactPyMiddleware`` to embed several ReactPy components into your existing application.
+- :pull:`1113` - Added ``reactpy.templatetags.Jinja`` that can be used alongside ``ReactPyMiddleware`` to embed several ReactPy components into your existing application.
 - :pull:`1113` - Added ``standard``, ``uvicorn``, ``jinja`` installation extras (for example ``pip install reactpy[standard]``).
 - :pull:`1113` - Added support for Python 3.12 and 3.13.
 - :pull:`1264` - Added ``reactpy.use_async_effect`` hook.

src/reactpy/jinja.py

@@ -0,0 +1,3 @@
+from reactpy.templatetags.jinja import Jinja
+
+__all__ = ["Jinja"]

src/reactpy/templatetags/__init__.py

@@ -0,0 +1,44 @@
+from typing import ClassVar
+from uuid import uuid4
+
+from jinja2_simple_tags import StandaloneTag
+
+from reactpy.pyscript.utils import (
+    pyscript_component_html,
+    pyscript_setup_html,
+)
+from reactpy.utils import asgi_component_html
+
+
+class Jinja(StandaloneTag):  # type: ignore
+    safe_output = True
+    tags: ClassVar[set[str]] = {"component", "pyscript_component", "pyscript_setup"}
+
+    def render(self, *args: str, **kwargs: str) -> str:
+        if self.tag_name == "component":
+            return component(*args, **kwargs)
+
+        if self.tag_name == "pyscript_component":
+            return pyscript_component(*args, **kwargs)
+
+        if self.tag_name == "pyscript_setup":
+            return pyscript_setup(*args, **kwargs)
+
+        raise ValueError(f"Unknown tag: {self.tag_name}")
+
+
+def component(dotted_path: str, **kwargs: str) -> str:
+    class_ = kwargs.pop("class", "")
+    if kwargs:
+        raise ValueError(f"Unexpected keyword arguments: {', '.join(kwargs)}")
+    return asgi_component_html(
+        element_id=uuid4().hex, class_=class_, component_path=f"{dotted_path}/"
+    )
+
+
+def pyscript_component(*file_paths: str, initial: str = "", root: str = "root") -> str:
+    return pyscript_component_html(file_paths=file_paths, initial=initial, root=root)
+
+
+def pyscript_setup(*extra_py: str, extra_js: str = "", config: str = "") -> str:
+    return pyscript_setup_html(extra_py=extra_py, extra_js=extra_js, config=config)