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 base64
import io
import os
import shutil
import subprocess
import tempfile
from . import html
def img_element(func):
def convert_image(image):
attributes = func(image).copy()
if image.alt_text:
attributes["alt"] = image.alt_text
return [html.element("img", attributes)]
return convert_image
# Undocumented, but retained for backwards-compatibility with 0.3.x
inline = img_element
@img_element
def data_uri(image):
with image.open() as image_bytes:
encoded_src = base64.b64encode(image_bytes.read()).decode("ascii")
return {
"src": "data:{0};base64,{1}".format(image.content_type, encoded_src)
}
_wmf_extensions = {
"image/x-wmf": ".wmf",
"image/x-emf": ".emf",
}
def libreoffice_wmf_conversion(post_process=None):
if post_process is None:
post_process = lambda x: x
def convert_image(image):
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)
return convert_image
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)
|
