diff options
Diffstat (limited to 'mammoth/docx/document_xml.py')
| -rw-r--r-- | mammoth/docx/document_xml.py | 41 |
1 files changed, 37 insertions, 4 deletions
diff --git a/mammoth/docx/document_xml.py b/mammoth/docx/document_xml.py index 71d744c..f8f1656 100644 --- a/mammoth/docx/document_xml.py +++ b/mammoth/docx/document_xml.py @@ -1,14 +1,16 @@ +import os + from .. import documents from .. import results from .. import lists from .xmlparser import node_types -def read_document_xml_element(element, numbering=None): - reader = _create_reader(numbering) +def read_document_xml_element(element, numbering=None, content_types=None, relationships=None, docx_file=None): + reader = _create_reader(numbering=numbering, content_types=content_types, relationships=relationships, docx_file=docx_file) return reader(element) -def _create_reader(numbering): +def _create_reader(numbering, content_types, relationships, docx_file): _handlers = {} _ignored_elements = set([ "w:bookmarkStart", @@ -19,7 +21,9 @@ def _create_reader(numbering): "w:commentRangeStart", "w:commentRangeEnd", "w:commentReference", - "w:del" + "w:del", + "w:pPr", + "w:rPr", ]) def handler(name): @@ -96,6 +100,35 @@ def _create_reader(numbering): def ins(element): return _read_xml_elements(element.children) + + @handler("w:drawing") + def drawing(element): + return _read_xml_elements(element.children) + + @handler("wp:inline") + def inline(element): + alt_text = element.find_child_or_null("wp:docPr").attributes.get("descr") + blips = element.find_children("a:graphic") \ + .find_children("a:graphicData") \ + .find_children("pic:pic") \ + .find_children("pic:blipFill") \ + .find_children("a:blip") + return _read_blips(blips, alt_text) + + def _read_blips(blips, alt_text): + return results.combine(map(lambda blip: _read_blip(blip, alt_text), blips)) + + def _read_blip(element, alt_text): + relationship_id = element.attributes["r:embed"] + image_path = os.path.join("word", relationships[relationship_id].target) + content_type = content_types.find_content_type(image_path) + + def open_image(): + return docx_file.open(image_path) + + image = documents.image(alt_text=alt_text, content_type=content_type, open=open_image) + return results.success(image) + def read(element): handler = _handlers.get(element.name) if handler is None: |