From 209b32cfdae122ed0ca516d5c94a2ad91d616e1f Mon Sep 17 00:00:00 2001
From: octo-gone <nklimin007@gmail.com>
Date: Thu, 8 Aug 2024 23:33:41 +0300
Subject: [PATCH] feat: render template when instantiating

---
 .github/workflows/publish.yml      |  6 ----
 .github/workflows/release.yml      |  6 ----
 README.md                          | 57 ++++++++++++++++++++++++++++++
 pyproject.toml                     |  2 +-
 src/jinja2_components/component.py |  4 +++
 src/jinja2_components/ext.py       |  3 +-
 6 files changed, 63 insertions(+), 15 deletions(-)

diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index cae6ea3..b0ebd17 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -18,12 +18,6 @@ jobs:
         uses: actions/setup-python@v5
         with:
           python-version: '3.11'
-      - name: Cache Poetry
-        id: cache-poetry
-        uses: actions/cache@v4
-        with:
-          path: ~/.poetry
-          key: poetry
       - name: Install Poetry
         if: steps.cache-poetry.outputs.cache-hit != 'true'
         run: |
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index d7a65aa..5e9e6a7 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -18,12 +18,6 @@ jobs:
         uses: actions/setup-python@v5
         with:
           python-version: '3.11'
-      - name: Cache Poetry
-        id: cache-poetry
-        uses: actions/cache@v4
-        with:
-          path: ~/.poetry
-          key: poetry
       - name: Install Poetry
         if: steps.cache-poetry.outputs.cache-hit != 'true'
         run: |
diff --git a/README.md b/README.md
index 4cf9ebf..8c3d65b 100644
--- a/README.md
+++ b/README.md
@@ -131,3 +131,60 @@ print(template.render())
 #   <button>3</button>
 # </button>
 ```
+
+### Rendering component from code
+
+The idea behind this feature is to pass arguments and get rendered component inside the code.
+
+```python
+@register(name="button")
+class Button(Component):
+    template_str = "<button>{{ body }}</button>"
+
+@register(name="menu")
+class Menu(Component):
+    template_str = """\
+<div class="menu">\
+{% for btn in buttons %}
+  {% button body=btn %}\
+{% endfor %}
+</button>
+"""
+
+template = env.from_string("{% menu buttons=[1, 2, 3] %}")
+print(template.render())
+# <div class="menu">
+#   <button>1</button>
+#   <button>2</button>
+#   <button>3</button>
+# </button>
+
+print(Menu(env, buttons=[1, 2, 3]))
+# <div class="menu">
+#   <button>1</button>
+#   <button>2</button>
+#   <button>3</button>
+# </button>
+```
+
+Also it is possible to pass initialized components (because the are strings).
+
+```python
+@register(name="menu")
+class Menu(Component):
+    template_str = """\
+<div class="menu">\
+{% for btn in buttons %}
+  {{ btn }}\
+{% endfor %}
+</button>
+"""
+
+rendered = Menu(env, buttons=[Button(env, body=1), Button(env, body=2), Button(env, body=3)])
+print(rendered)
+# <div class="menu">
+#   <button>1</button>
+#   <button>2</button>
+#   <button>3</button>
+# </button>
+```
\ No newline at end of file
diff --git a/pyproject.toml b/pyproject.toml
index 136b9df..f4e2e34 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "jinja2-components"
-version = "0.0.2"
+version = "0.0.3"
 description = "Streamlined way to build and reuse your Jinja2 templates from code"
 authors = ["octo-gone <nklimin007@gmail.com>"]
 readme = "README.md"
diff --git a/src/jinja2_components/component.py b/src/jinja2_components/component.py
index 001d668..ee79a81 100644
--- a/src/jinja2_components/component.py
+++ b/src/jinja2_components/component.py
@@ -10,6 +10,10 @@ class Component:
     template_name: t.ClassVar[t.Optional[str]]
     block: t.ClassVar[bool] = False
 
+    def __new__(cls, env: "Environment", *args, **kwargs):
+        template = cls.get_template(env, *args, **kwargs)
+        return template.render(cls.get_context(*args, **kwargs))
+
     @classmethod
     def get_context(cls, *args, **kwargs):
         return kwargs
diff --git a/src/jinja2_components/ext.py b/src/jinja2_components/ext.py
index 66b4015..8c9896e 100644
--- a/src/jinja2_components/ext.py
+++ b/src/jinja2_components/ext.py
@@ -81,5 +81,4 @@ def parse_args(self, parser: "Parser"):
 
     def render(self, tag: str, *args, **kwargs):
         component = self.components[tag]
-        template = component.get_template(self.environment, *args, **kwargs)
-        return template.render(component.get_context(*args, **kwargs))
+        return component(self.environment, *args, **kwargs)