template.py
151 lines
| 4.7 KiB
| text/x-python
|
PythonLexer
/ converters / template.py
Matthias BUSSONNIER
|
r8994 | """Base classes for the notebook conversion pipeline. | ||
This module defines Converter, from which all objects designed to implement | ||||
a conversion of IPython notebooks to some other format should inherit. | ||||
""" | ||||
#----------------------------------------------------------------------------- | ||||
# Copyright (c) 2012, 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 __future__ import print_function, absolute_import | ||||
# Stdlib imports | ||||
import io | ||||
import os | ||||
Matthias BUSSONNIER
|
r9040 | from IPython.utils import path | ||
Matthias BUSSONNIER
|
r8994 | |||
Matthias BUSSONNIER
|
r9183 | from jinja2 import Environment, FileSystemLoader | ||
Matthias BUSSONNIER
|
r9049 | env = Environment( | ||
loader=FileSystemLoader('./templates/'), | ||||
extensions=['jinja2.ext.loopcontrols'] | ||||
) | ||||
Matthias BUSSONNIER
|
r8994 | |||
# IPython imports | ||||
from IPython.nbformat import current as nbformat | ||||
Matthias BUSSONNIER
|
r9183 | from IPython.config.configurable import Configurable | ||
from IPython.utils.traitlets import ( Unicode, Any) | ||||
Matthias BUSSONNIER
|
r8994 | |||
# Our own imports | ||||
from IPython.utils.text import indent | ||||
from .utils import remove_ansi | ||||
Matthias BUSSONNIER
|
r8997 | from markdown import markdown | ||
Matthias BUSSONNIER
|
r9183 | from .utils import highlight, ansi2html | ||
Matthias BUSSONNIER
|
r8994 | #----------------------------------------------------------------------------- | ||
# Class declarations | ||||
#----------------------------------------------------------------------------- | ||||
def rm_fake(strng): | ||||
return strng.replace('/files/', '') | ||||
class ConversionException(Exception): | ||||
pass | ||||
def python_comment(string): | ||||
return '# '+'\n# '.join(string.split('\n')) | ||||
Matthias BUSSONNIER
|
r9040 | |||
def header_body(): | ||||
Matthias BUSSONNIER
|
r9183 | """Return the body of the header as a list of strings.""" | ||
from pygments.formatters import HtmlFormatter | ||||
header = [] | ||||
static = os.path.join(path.get_ipython_package_dir(), | ||||
'frontend', 'html', 'notebook', 'static', | ||||
) | ||||
here = os.path.split(os.path.realpath(__file__))[0] | ||||
css = os.path.join(static, 'css') | ||||
for sheet in [ | ||||
# do we need jquery and prettify? | ||||
# os.path.join(static, 'jquery', 'css', 'themes', 'base', | ||||
# 'jquery-ui.min.css'), | ||||
# os.path.join(static, 'prettify', 'prettify.css'), | ||||
os.path.join(css, 'boilerplate.css'), | ||||
os.path.join(css, 'fbm.css'), | ||||
os.path.join(css, 'notebook.css'), | ||||
os.path.join(css, 'renderedhtml.css'), | ||||
# our overrides: | ||||
os.path.join(here, '..', 'css', 'static_html.css'), | ||||
]: | ||||
with io.open(sheet, encoding='utf-8') as f: | ||||
s = f.read() | ||||
header.append(s) | ||||
pygments_css = HtmlFormatter().get_style_defs('.highlight') | ||||
header.append(pygments_css) | ||||
return header | ||||
inlining = {} | ||||
Matthias BUSSONNIER
|
r9048 | inlining['css'] = header_body() | ||
Matthias BUSSONNIER
|
r9040 | |||
Matthias BUSSONNIER
|
r9049 | |||
def filter_data_type(output): | ||||
Matthias BUSSONNIER
|
r9183 | for fmt in ['html', 'pdf', 'svg', 'latex', 'png', 'jpg', 'jpeg' , 'text']: | ||
Matthias BUSSONNIER
|
r9049 | if fmt in output: | ||
return [fmt] | ||||
env.filters['filter_data_type'] = filter_data_type | ||||
Matthias BUSSONNIER
|
r8994 | env.filters['pycomment'] = python_comment | ||
env.filters['indent'] = indent | ||||
env.filters['rm_fake'] = rm_fake | ||||
env.filters['rm_ansi'] = remove_ansi | ||||
Matthias BUSSONNIER
|
r8997 | env.filters['markdown'] = markdown | ||
env.filters['highlight'] = highlight | ||||
Matthias BUSSONNIER
|
r9039 | env.filters['ansi2html'] = ansi2html | ||
Matthias BUSSONNIER
|
r8994 | |||
class ConverterTemplate(Configurable): | ||||
display_data_priority = ['pdf', 'svg', 'png', 'jpg', 'text'] | ||||
#------------------------------------------------------------------------- | ||||
# Instance-level attributes that are set in the constructor for this | ||||
# class. | ||||
#------------------------------------------------------------------------- | ||||
infile = Any() | ||||
Matthias BUSSONNIER
|
r9010 | |||
Matthias BUSSONNIER
|
r8994 | infile_dir = Unicode() | ||
Matthias BUSSONNIER
|
r8997 | def __init__(self, tplfile='fullhtml', config=None, **kw): | ||
self.template = env.get_template(tplfile+'.tpl') | ||||
Matthias BUSSONNIER
|
r9183 | self.nb = None | ||
super(ConverterTemplate, self).__init__(config=config, **kw) | ||||
Matthias BUSSONNIER
|
r8994 | |||
def process(self): | ||||
converted_cells = [] | ||||
for worksheet in self.nb.worksheets: | ||||
for cell in worksheet.cells: | ||||
cell.type = cell.cell_type | ||||
Matthias BUSSONNIER
|
r9039 | cell.haspyout = False | ||
Matthias BUSSONNIER
|
r9183 | for out in cell.get('outputs', []): | ||
Matthias BUSSONNIER
|
r9039 | if out.output_type == 'pyout': | ||
cell.haspyout = True | ||||
break | ||||
Matthias BUSSONNIER
|
r9010 | converted_cells.append(worksheet) | ||
Matthias BUSSONNIER
|
r8994 | |||
return converted_cells | ||||
Matthias BUSSONNIER
|
r9183 | def convert(self): | ||
Matthias BUSSONNIER
|
r9182 | """ convert the ipynb file | ||
return both the converted ipynb file and a dict containing potential | ||||
other resources | ||||
""" | ||||
Matthias BUSSONNIER
|
r9183 | return self.template.render(worksheets=self.process(), inlining=inlining), {} | ||
Matthias BUSSONNIER
|
r8994 | |||
def read(self, filename): | ||||
"read and parse notebook into NotebookNode called self.nb" | ||||
Matthias BUSSONNIER
|
r9014 | with io.open(filename) as f: | ||
Matthias BUSSONNIER
|
r8994 | self.nb = nbformat.read(f, 'json') | ||