1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
import io
import os
import shutil
import subprocess
import tempfile
# An example of how to use LibreOffice and ImageMagick to convert WMF images to
# PNGs.
#
# libreoffice_wmf_conversion uses LibreOffice to convert the image to a PNG.
# This normally creates an image with a large amount of padding, so
# imagemagick_trim can be used to trim the image.
#
# The image can be then be converted using a normal image handler, such as
# mammoth.images.data_uri.
#
# Example usage:
#
# def convert_image(image):
# image = libreoffice_wmf_conversion(image, post_process=imagemagick_trim)
# return mammoth.images.data_uri(image)
#
#
# fileobj = open("document.docx", "rb")
# result = mammoth.convert_to_html(
# fileobj,
# convert_image=compose(
# mammoth.images.data_uri,
# libreoffice_wmf_conversion(post_process=imagemagick_trim),
# ),
# )
#
# libreoffice_wmf_conversion uses LibreOffice to convert the image
_wmf_extensions = {
"image/x-wmf": ".wmf",
"image/x-emf": ".emf",
}
def libreoffice_wmf_conversion(image, post_process=None):
if post_process is None:
post_process = lambda x: x
wmf_extension = _wmf_extensions.get(image.content_type)
if wmf_extension is None:
return image
else:
temporary_directory = tempfile.mkdtemp()
try:
input_path = os.path.join(temporary_directory, "image" + wmf_extension)
with io.open(input_path, "wb") as input_fileobj:
with image.open() as image_fileobj:
shutil.copyfileobj(image_fileobj, input_fileobj)
output_path = os.path.join(temporary_directory, "image.png")
subprocess.check_call([
"libreoffice",
"--headless",
"--convert-to",
"png",
input_path,
"--outdir",
temporary_directory,
])
with io.open(output_path, "rb") as output_fileobj:
output = output_fileobj.read()
def open_image():
return io.BytesIO(output)
return post_process(image.copy(
content_type="image/png",
open=open_image,
))
finally:
shutil.rmtree(temporary_directory)
def imagemagick_trim(image):
command = ["convert", "-", "-trim", "-"]
process = subprocess.Popen(command, stdin=subprocess.PIPE, stdout=subprocess.PIPE)
try:
with image.open() as image_fileobj:
shutil.copyfileobj(image_fileobj, process.stdin)
output, err_output = process.communicate()
except:
process.kill()
process.wait()
raise
return_code = process.poll()
if return_code:
raise subprocess.CalledProcessError(return_code, command)
else:
def open_image():
return io.BytesIO(output)
return image.copy(open=open_image)
|
