"""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 from IPython.utils.tempdir import TemporaryDirectory from .convertfigures import ConvertFiguresTransformer #----------------------------------------------------------------------------- # Constants #----------------------------------------------------------------------------- INKSCAPE_COMMAND = "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) #Call conversion application output_filename = os.path.join(tmpdir, 'figure.pdf') shell = INKSCAPE_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) as f: return f.read() else: return TypeError("Inkscape svg to png conversion failed")