From 48725944f4bf4705b6f35438f6bc54fcfff51f00 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Mon, 3 Apr 2017 11:46:40 +0100 Subject: Add transforms.get_descendants() --- mammoth/transforms.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'mammoth/transforms.py') diff --git a/mammoth/transforms.py b/mammoth/transforms.py index 6eb6d92..88db8a1 100644 --- a/mammoth/transforms.py +++ b/mammoth/transforms.py @@ -20,3 +20,22 @@ def _each_element(transform_element): return transform_element(element) return transform_element_and_children + + +def get_descendants(element): + descendants = [] + + def visit(element): + descendants.append(element) + + _visit_descendants(element, visit) + + return descendants + + +def _visit_descendants(element, visit): + if isinstance(element, documents.HasChildren): + for child in element.children: + _visit_descendants(child, visit) + visit(child) + -- cgit v1.2.3