diff options
Diffstat (limited to 'mammoth/__init__.py')
| -rw-r--r-- | mammoth/__init__.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/mammoth/__init__.py b/mammoth/__init__.py index 297cf24..7735fcf 100644 --- a/mammoth/__init__.py +++ b/mammoth/__init__.py @@ -1,5 +1,6 @@ from . import docx, conversion, options, images, underline from .raw_text import extract_raw_text_from_element +from .docx.style_map import write_style_map, read_style_map __all__ = ["convert_to_html", "extract_raw_text", "images", "underline"] @@ -15,6 +16,8 @@ def convert_to_markdown(*args, **kwargs): def convert(fileobj, transform_document=None, id_prefix=None, **kwargs): if transform_document is None: transform_document = lambda x: x + if kwargs.get("style_map") is None: + kwargs["style_map"] = read_style_map(fileobj) return options.read_options(kwargs).bind(lambda convert_options: docx.read(fileobj).map(transform_document).bind(lambda document: conversion.convert_document_element_to_html( @@ -28,3 +31,10 @@ def convert(fileobj, transform_document=None, id_prefix=None, **kwargs): def extract_raw_text(fileobj): return docx.read(fileobj).map(extract_raw_text_from_element) + + +def embed_style_map(fileobj, style_map): + write_style_map(fileobj, style_map) + +def read_embedded_style_map(fileobj): + return read_style_map(fileobj) |