X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/b2f456977cd5884c625f704e024213ad86163929..8315186b58e7fb5ff12472eff11fb6e0a789a51e:/models/pages.py diff --git a/models/pages.py b/models/pages.py index 354293f..5f75494 100644 --- a/models/pages.py +++ b/models/pages.py @@ -43,9 +43,10 @@ class Template(TreeModel): nodes = [] for node in nodelist: try: - for nodelist_name in ('nodelist', 'nodelist_loop', 'nodelist_empty', 'nodelist_true', 'nodelist_false', 'nodelist_main'): - if hasattr(node, nodelist_name): - nodes.extend(nodelist_container_nodes(getattr(node, nodelist_name))) + if hasattr(node, 'child_nodelists'): + for nodelist_name in node.child_nodelists: + if hasattr(node, nodelist_name): + nodes.extend(nodelist_container_nodes(getattr(node, nodelist_name))) if isinstance(node, ContainerNode): nodes.append(node) elif isinstance(node, ExtendsNode):