Merge pull request #3749 from minrk/pytransform

convert IPython syntax to Python syntax in nbconvert python template

adds ipython2python filter and uses it in python.tpl

IPython/nbconvert/exporters/exporter.py

@@ -54,6 +54,7 b' default_filters = {'
         'get_lines': filters.get_lines,
         'highlight2html': filters.highlight2html,
         'highlight2latex': filters.highlight2latex,
+        'ipython2python': filters.ipython2python,
         'markdown2latex': filters.markdown2latex,
         'markdown2rst': filters.markdown2rst,
         'comment_lines': filters.comment_lines,

IPython/nbconvert/filters/strings.py

@@ -19,6 +19,8 b' templates.'
 import re
 import textwrap
 from xml.etree import ElementTree
+
+from IPython.core.interactiveshell import InteractiveShell
 from IPython.utils import py3compat
 
 #-----------------------------------------------------------------------------
@@ -32,7 +34,8 b' __all__ = ['
     'strip_dollars',
     'strip_files_prefix',
     'comment_lines',
-    'get_lines'
+    'get_lines',
+    'ipython2python',
 ]
 
 
@@ -150,3 +153,15 b' def get_lines(text, start=None,end=None):'
     
     # Return the right lines.
     return "\n".join(lines[start:end]) #re-join
+
+def ipython2python(code):
+    """Transform IPython syntax to pure Python syntax
+
+    Parameters
+    ----------
+
+    code : str
+        IPython code, to be transformed to pure Python
+    """
+    shell = InteractiveShell.instance()
+    return shell.input_transformer_manager.transform_cell(code)

IPython/nbconvert/templates/python.tpl

@@ -7,7 +7,7 b''
 {% block output_prompt %}
 # Out[{{cell.prompt_number}}]:{% endblock output_prompt %}
 
-{% block input %}{{ cell.input }}
+{% block input %}{{ cell.input | ipython2python }}
 {% endblock input %}