From 438d88240b5bfa99494562fc39817967fbfbb2f0 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Tue, 4 Apr 2017 19:45:45 +0100 Subject: Extract element_of_type in transforms module --- mammoth/transforms.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'mammoth/transforms.py') 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 -- cgit v1.2.3