Allow block definitions in if/for/match blocks (fixes #335)

rinja-rs · Jun 2, 2020 · 67f345b · 67f345b
1 parent 17ec9e4
commit 67f345b
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/askama_shared/src/heritage.rs b/askama_shared/src/heritage.rs
@@ -75,6 +75,19 @@ impl<'a> Context<'a> {
                             nested.push(nodes);
                         }
                     }
+                    Node::Cond(branches, _) => {
+                        for (_, _, nodes) in branches {
+                            nested.push(nodes);
+                        }
+                    }
+                    Node::Loop(_, _, _, nodes, _) => {
+                        nested.push(nodes);
+                    }
+                    Node::Match(_, _, _, arms, _) => {
+                        for (_, _, _, arm) in arms {
+                            nested.push(arm);
+                        }
+                    }
                     _ => {}
                 }
             }