Mustache comments.

Author: chr15m

runprompt

@@ -120,6 +120,9 @@ def render_template(template, variables):
         return ctx
 
     def render(tmpl, ctx):
+        # Remove comments: {{! ... }}
+        tmpl = re.sub(r"\{\{!.*?\}\}", "", tmpl, flags=re.DOTALL)
+
         # Process sections: {{#key}}...{{/key}}
         def section_replace(match):
             key = match.group(1)

tests/test-mustache.py

@@ -85,13 +85,23 @@ def test_combined():
          {"items": [1]}, "have")
 
 
+def test_comments():
+    print("\n--- Comments ---")
+    test("simple comment", "Hello {{! this is a comment }}World", {}, "Hello World")
+    test("comment removes entirely", "{{! comment }}", {}, "")
+    test("comment with variable", "{{! ignore }}{{name}}", {"name": "Alice"}, "Alice")
+    test("multiline comment", "Hello {{! this\nis\nmultiline }}World", {}, "Hello World")
+    test("comment between variables", "{{a}}{{! middle }}{{b}}", {"a": "X", "b": "Y"}, "XY")
+
+
 def main():
     test_basic_interpolation()
     test_dot_notation()
     test_sections()
     test_section_lists()
     test_inverted_sections()
     test_combined()
+    test_comments()
 
     print("\n" + "=" * 40)
     print("Passed: %d, Failed: %d" % (passed, failed))