Fix svg2pdf filter

IPython/nbconvert/preprocessors/convertfigures.py

@@ -1,64 +1,58 b''
 """Module containing a preprocessor 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
 #-----------------------------------------------------------------------------
 
 from .base import Preprocessor
 from IPython.utils.traitlets import Unicode
 
 #-----------------------------------------------------------------------------
 # Classes
 #-----------------------------------------------------------------------------
 
 class ConvertFiguresPreprocessor(Preprocessor):
     """
     Converts all of the outputs in a notebook from one format to another.
     """
 
     from_format = Unicode(config=True, help='Format the converter accepts')
     to_format = Unicode(config=True, help='Format the converter writes')
 
     def __init__(self, **kw):
         """
         Public constructor
         """
         super(ConvertFiguresPreprocessor, self).__init__(**kw)
 
 
     def convert_figure(self, data_format, data):
         raise NotImplementedError()
 
 
     def preprocess_cell(self, cell, resources, cell_index):
         """
         Apply a transformation on each cell,
         
         See base.py
         """
 
         # Loop through all of the datatypes of the outputs in the cell.
-        for index, cell_out in enumerate(cell.get('outputs', [])):
-            for data_type, data in list(cell_out.items()):
-                # this must run *before* extract outputs,
-                # so figure_name and filename do not exist
-                self._convert_figure(cell_out, resources, data_type, data)
-        return cell, resources
+        for output in cell.get('outputs', []):
+            if output.output_type in {'execute_result', 'display_data'} \
+                    and self.from_format in output.data \
+                    and self.to_format not in output.data:
 
+                output.data[self.to_format] = self.convert_figure(
+                            self.from_format, output.data[self.from_format])
 
-    def _convert_figure(self, cell_out, resources, data_type, data):
-        """
-        Convert a figure and output the results to the cell output
-        """
-        if not self.to_format in cell_out and data_type == self.from_format:
-            data = self.convert_figure(data_type, data)
-            cell_out[self.to_format] = data
+        return cell, resources

IPython/nbconvert/preprocessors/svg2pdf.py

@@ -1,82 +1,82 b''
 """Module containing a preprocessor that converts outputs in the notebook from 
 one format to another.
 """
 
 # Copyright (c) IPython Development Team.
 # Distributed under the terms of the Modified BSD License.
 
 import base64
 import io
 import os
 import sys
 import subprocess
 
 from IPython.utils.tempdir import TemporaryDirectory
 from IPython.utils.traitlets import Unicode
 
 from .convertfigures import ConvertFiguresPreprocessor
 
 
 INKSCAPE_APP = '/Applications/Inkscape.app/Contents/Resources/bin/inkscape'
 
 
 class SVG2PDFPreprocessor(ConvertFiguresPreprocessor):
     """
     Converts all of the outputs in a notebook from SVG to PDF.
     """
     
     def _from_format_default(self):
         return 'image/svg+xml'
     def _to_format_default(self):
         return 'application/pdf'
     
     command = Unicode(config=True,
         help="""The command to use for converting SVG to PDF
         
         This string is a template, which will be formatted with the keys
         to_filename and from_filename.
         
         The conversion call must read the SVG from {from_flename},
         and write a PDF to {to_filename}.
         """)
     
     def _command_default(self):
         return self.inkscape + \
                ' --without-gui --export-pdf="{to_filename}" "{from_filename}"'
     
     inkscape = Unicode(config=True, help="The path to Inkscape, if necessary")
     def _inkscape_default(self):
         if sys.platform == "darwin":
             if os.path.isfile(INKSCAPE_APP):
                 return INKSCAPE_APP
         return "inkscape"
 
 
     def convert_figure(self, data_format, data):
         """
         Convert a single SVG figure to PDF.  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)
+            input_filename = os.path.join(tmpdir, 'figure.svg')
             # SVG data is unicode text
             with io.open(input_filename, 'w', encoding='utf8') as f:
                 f.write(data)
 
             #Call conversion application
             output_filename = os.path.join(tmpdir, 'figure.pdf')
             shell = self.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
             # return value expects a filename
             if os.path.isfile(output_filename):
                 with open(output_filename, 'rb') as f:
                     # PDF is a nb supported binary, data type, so base64 encode.
                     return base64.encodestring(f.read())
             else:
                 raise TypeError("Inkscape svg to pdf conversion failed")