Add transform_document argument to convert_to_html

1 files changed, 4 insertions, 2 deletions

diff --git a/mammoth/__init__.py b/mammoth/__init__.py
index 90ab764..59e349f 100644
--- a/mammoth/__init__.py
+++ b/mammoth/__init__.py
@@ -2,8 +2,10 @@ from .results import Result
 from . import docx, conversion, options
 
 
-def convert_to_html(fileobj, **kwargs):
-    return docx.read(fileobj).bind(lambda document: 
+def convert_to_html(fileobj, transform_document=None, **kwargs):
+    if transform_document is None:
+        transform_document = lambda x: x
+    return docx.read(fileobj).map(transform_document).bind(lambda document: 
         conversion.convert_document_element_to_html(
             document,
             **options.read_options(kwargs)