From 5b7a5f64f48a9283ecaf0a740fa5ac0181c8b150 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sat, 14 Dec 2013 18:19:45 +0000 Subject: Add style argument to convert_to_html --- mammoth/__init__.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/mammoth/__init__.py b/mammoth/__init__.py index 987226d..7491fb6 100644 --- a/mammoth/__init__.py +++ b/mammoth/__init__.py @@ -2,15 +2,20 @@ from .results import Result from . import docx, conversion, style_reader, lists -def convert_to_html(fileobj): +def convert_to_html(fileobj, styles=None): + if styles is None: + styles = _default_styles + return docx.read(fileobj).bind(lambda document: - conversion.convert_document_element_to_html(document, styles=_create_default_styles()) + conversion.convert_document_element_to_html(document, styles=_read_styles(styles)) ) -def _create_default_styles(): - lines = filter(None, map(lambda line: line.strip(), _default_styles.split("\n"))) +def _read_styles(style_text): + lines = filter(None, map(lambda line: line.strip(), style_text.split("\n"))) return lists.map(style_reader.read_style, lines) + + _default_styles = """ p.Heading1 => h1:fresh -- cgit v1.2.3