"""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 import sys import subprocess from IPython.utils.tempdir import TemporaryDirectory from .convertfigures import ConvertFiguresTransformer #----------------------------------------------------------------------------- # Constants #----------------------------------------------------------------------------- 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}"' #----------------------------------------------------------------------------- # 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) #Determine command (different on Mac OSX) command = INKSCAPE_COMMAND if sys.platform == 'darwin': command = INKSCAPE_OSX_COMMAND #Call conversion application output_filename = os.path.join(tmpdir, 'figure.pdf') shell = command.format(from_filename=input_filename, to_filename=output_filename) subprocess.call(shell, shell=True) #Shell=True okay since input is trusted. #Read output from drive if os.path.isfile(output_filename): with open(output_filename, 'rb') as f: return f.read().encode("base64") #PDF is a nb supported binary #data type, so base64 encode. else: return TypeError("Inkscape svg to png conversion failed") #TODO: ERROR: which notebook crashed exporter