Files
api/path/to/venv/lib/python3.12/site-packages/mammoth/document_matchers.py
2025-12-30 11:27:14 +07:00

96 lines
2.0 KiB
Python

import collections
import cobble
def paragraph(style_id=None, style_name=None, numbering=None):
return ParagraphMatcher(style_id, style_name, numbering)
ParagraphMatcher = collections.namedtuple("ParagraphMatcher", ["style_id", "style_name", "numbering"])
ParagraphMatcher.element_type = "paragraph"
def run(style_id=None, style_name=None):
return RunMatcher(style_id, style_name)
RunMatcher = collections.namedtuple("RunMatcher", ["style_id", "style_name"])
RunMatcher.element_type = "run"
def table(style_id=None, style_name=None):
return TableMatcher(style_id, style_name)
TableMatcher = collections.namedtuple("TableMatcher", ["style_id", "style_name"])
TableMatcher.element_type = "table"
class bold(object):
element_type = "bold"
class italic(object):
element_type = "italic"
class underline(object):
element_type = "underline"
class strikethrough(object):
element_type = "strikethrough"
class all_caps(object):
element_type = "all_caps"
class small_caps(object):
element_type = "small_caps"
def highlight(color=None):
return HighlightMatcher(color=color)
HighlightMatcher = collections.namedtuple("HighlightMatcher", ["color"])
HighlightMatcher.element_type = "highlight"
class comment_reference(object):
element_type = "comment_reference"
BreakMatcher = collections.namedtuple("BreakMatcher", ["break_type"])
BreakMatcher.element_type = "break"
line_break = BreakMatcher("line")
page_break = BreakMatcher("page")
column_break = BreakMatcher("column")
def equal_to(value):
return StringMatcher(_operator_equal_to, value)
def _operator_equal_to(first, second):
return first.upper() == second.upper()
def starts_with(value):
return StringMatcher(_operator_starts_with, value)
def _operator_starts_with(first, second):
return second.upper().startswith(first.upper())
@cobble.data
class StringMatcher(object):
operator = cobble.field()
value = cobble.field()
def matches(self, other):
return self.operator(self.value, other)