aboutsummaryrefslogtreecommitdiff
path: root/mammoth/transforms.py
blob: 6eb6d92a91b4911183ed409c07b1c97f4b526279 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from . import documents


def paragraph(transform_paragraph):
    def transform_element(element):
        if isinstance(element, documents.Paragraph):
            return transform_paragraph(element)
        else:
            return element
    
    return _each_element(transform_element)


def _each_element(transform_element):
    def transform_element_and_children(element):
        if isinstance(element, documents.HasChildren):
            children = list(map(transform_element, element.children))
            element = element.copy(children=children)
        
        return transform_element(element)
    
    return transform_element_and_children