svg2pdf.py
81 lines
| 3.0 KiB
| text/x-python
|
PythonLexer
Jonathan Frederic
|
r11367 | """Module containing a transformer that converts outputs in the notebook from | ||
one format to another. | ||||
""" | ||||
#----------------------------------------------------------------------------- | ||||
# Copyright (c) 2013, the IPython Development Team. | ||||
# | ||||
# Distributed under the terms of the Modified BSD License. | ||||
# | ||||
# The full license is in the file COPYING.txt, distributed with this software. | ||||
#----------------------------------------------------------------------------- | ||||
#----------------------------------------------------------------------------- | ||||
# Imports | ||||
#----------------------------------------------------------------------------- | ||||
import os | ||||
Jonathan Frederic
|
r11385 | import sys | ||
import subprocess | ||||
Jonathan Frederic
|
r11367 | from IPython.utils.tempdir import TemporaryDirectory | ||
from .convertfigures import ConvertFiguresTransformer | ||||
#----------------------------------------------------------------------------- | ||||
# Constants | ||||
#----------------------------------------------------------------------------- | ||||
Jonathan Frederic
|
r11384 | INKSCAPE_COMMAND = 'inkscape --without-gui --export-pdf="{to_filename}" "{from_filename}"' | ||
INKSCAPE_OSX_COMMAND = '/Applications/Inkscape.app/Contents/Resources/bin/inkscape --without-gui --export-pdf="{to_filename}" "{from_filename}"' | ||||
Jonathan Frederic
|
r11367 | |||
#----------------------------------------------------------------------------- | ||||
# Classes | ||||
#----------------------------------------------------------------------------- | ||||
class ConvertSvgTransformer(ConvertFiguresTransformer): | ||||
""" | ||||
Converts all of the outputs in a notebook from one format to another. | ||||
""" | ||||
def __init__(self, **kw): | ||||
""" | ||||
Constructor | ||||
""" | ||||
super(ConvertSvgTransformer, self).__init__(['svg'], 'pdf', **kw) | ||||
def convert_figure(self, data_format, data): | ||||
""" | ||||
Convert a single Svg figure. Returns converted data. | ||||
""" | ||||
#Work in a temporary directory | ||||
with TemporaryDirectory() as tmpdir: | ||||
#Write fig to temp file | ||||
input_filename = os.path.join(tmpdir, 'figure.' + data_format) | ||||
with open(input_filename, 'w') as f: | ||||
f.write(data) | ||||
Jonathan Frederic
|
r11384 | #Determine command (different on Mac OSX) | ||
command = INKSCAPE_COMMAND | ||||
if sys.platform == 'darwin': | ||||
command = INKSCAPE_OSX_COMMAND | ||||
Jonathan Frederic
|
r11367 | #Call conversion application | ||
output_filename = os.path.join(tmpdir, 'figure.pdf') | ||||
Jonathan Frederic
|
r11384 | shell = command.format(from_filename=input_filename, | ||
to_filename=output_filename) | ||||
Jonathan Frederic
|
r11367 | subprocess.call(shell, shell=True) #Shell=True okay since input is trusted. | ||
#Read output from drive | ||||
if os.path.isfile(output_filename): | ||||
Jonathan Frederic
|
r11374 | with open(output_filename, 'rb') as f: | ||
return f.read().encode("base64") #PDF is a nb supported binary | ||||
#data type, so base64 encode. | ||||
Jonathan Frederic
|
r11367 | else: | ||
return TypeError("Inkscape svg to png conversion failed") | ||||
Jonathan Frederic
|
r11379 | #TODO: ERROR: which notebook crashed exporter | ||