##// END OF EJS Templates
Added writers and supporting code.
Added writers and supporting code.

File last commit:

r11367:b2eb244e
r11367:b2eb244e
Show More
convertsvg.py
70 lines | 2.5 KiB | text/x-python | PythonLexer
"""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")