aboutsummaryrefslogtreecommitdiff
path: root/mammoth/style_reader/document_matcher_reader.py
blob: ba08e09e9c2c4cbfac050ee76dedca1793c11ffe (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
from parsimonious.grammar import Grammar

from .. import document_matchers


def read_document_matcher(string):
    root_node = _grammar.parse(string)
    element_node = root_node.children[0]
    if element_node.expr_name == "paragraph":
        return _read_paragraph_node(element_node)
    elif element_node.expr_name == "run":
        return _read_run_node(element_node)
    

def _read_paragraph_node(paragraph_node):
    style_name = _read_style_node(paragraph_node.children[1])
    return document_matchers.paragraph(style_name=style_name)
    

def _read_run_node(run_node):
    style_name = _read_style_node(run_node.children[1])
    return document_matchers.run(style_name=style_name)


def _read_style_node(style_node):
    if style_node.children:
        return style_node.children[0].children[1].text
    else:
        return None


_grammar = Grammar(r"""
matcher = paragraph / run

paragraph = "p" style_name?

run = "r" style_name?

style_name = "." style_identifier

style_identifier = ~"[A-Z0-9]*"i
""")