diff options
| -rw-r--r-- | mammoth/transforms.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/mammoth/transforms.py b/mammoth/transforms.py index 21bcbf6..d126578 100644 --- a/mammoth/transforms.py +++ b/mammoth/transforms.py @@ -2,9 +2,13 @@ from . import documents def paragraph(transform_paragraph): + return element_of_type(documents.Paragraph, transform_paragraph) + + +def element_of_type(element_type, transform): def transform_element(element): - if isinstance(element, documents.Paragraph): - return transform_paragraph(element) + if isinstance(element, element_type): + return transform(element) else: return element |