Extract element_of_type in transforms module

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