feat(formatter): format set blocks

closes #646
djlint · May 18, 2023 · d88371c · d88371c
1 parent 17faf5f
commit d88371c
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/src/djlint/settings.py b/src/djlint/settings.py
@@ -510,6 +510,7 @@ def __init__(
                 | raw
                 | blocktrans(?!late)
                 | blocktranslate
+                | set(?!(?:(?!%}).)*=)
             """
             + self.custom_blocks
             + r")"
@@ -658,6 +659,7 @@ def __init__(
             | raw
             | blocktrans(?!late)
             | blocktranslate
+            | set(?!(?:(?!%}).)*=)
 
             """
             + self.custom_blocks
@@ -708,6 +710,7 @@ def __init__(
             | endblocktrans(?!late)
             | blocktranslate
             | endblocktranslate
+            | set(?!(?:(?!%}).)*=)
             """
             + self.custom_blocks
             + r""")

diff --git a/tests/test_nunjucks/test_set.py b/tests/test_nunjucks/test_set.py
@@ -122,6 +122,23 @@
         ({}),
         id="indent py style list",
     ),
+    pytest.param(
+        (
+            '{% set cta %}{% include "partials/cta.njk" %}<div></div>{% endset %}\n'
+            "{%-set posts = collections.docs-%}\n"
+            "{%asdf%}"
+        ),
+        (
+            "{% set cta %}\n"
+            '    {% include "partials/cta.njk" %}\n'
+            "    <div></div>\n"
+            "{% endset %}\n"
+            "{%- set posts = collections.docs -%}\n"
+            "{% asdf %}\n"
+        ),
+        ({}),
+        id="set block",
+    ),
 ]