aclose context blocks

pallets · Apr 7, 2024 · 22cffd0 · 22cffd0
1 parent 3bca0ab
commit 22cffd0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/jinja2/compiler.py b/src/jinja2/compiler.py
@@ -973,14 +973,18 @@ def visit_Block(self, node: nodes.Block, frame: Frame) -> None:
                 f"yield from context.blocks[{node.name!r}][0]({context})", node
             )
         else:
+            self.writeline("gen = context.blocks[{node.name!r}][0]({context})")
+            self.writeline("try:")
+            self.indent()
             self.writeline(
-                f"{self.choose_async()}for event in"
-                f" context.blocks[{node.name!r}][0]({context}):",
+                f"{self.choose_async()}for event in gen:"
                 node,
             )
             self.indent()
             self.simple_write("event", frame)
             self.outdent()
+            self.outdent()
+            self.write_line(f"finally: {self.choose_async('await gen.aclose()', 'gen.close()')}")
 
         self.outdent(level)