Adding rst and heading cells to the notebook format.

IPython/nbformat/v2/nbbase.py

@@ -119,6 +119,16 b' def new_text_cell(cell_type, source=None, rendered=None):'
     return cell
 
 
+def new_heading_cell(source=None, level=1):
+    """Create a new section cell with a given integer level."""
+    cell = NotebookNode()
+    cell.cell_type = u'heading'
+    if source is not None:
+        cell.source = unicode(source)
+    cell.level = int(level)
+    return cell
+
+
 def new_worksheet(name=None, cells=None):
     """Create a worksheet by name with with a list of cells."""
     ws = NotebookNode()

IPython/nbformat/v2/nbpy.py

@@ -18,7 +18,10 b' Authors:'
 
 import re
 from .rwbase import NotebookReader, NotebookWriter
-from .nbbase import new_code_cell, new_text_cell, new_worksheet, new_notebook
+from .nbbase import (
+    new_code_cell, new_text_cell, new_worksheet,
+    new_notebook, new_heading_cell
+)
 
 #-----------------------------------------------------------------------------
 # Code
@@ -39,28 +42,53 b' class PyReader(NotebookReader):'
         lines = s.splitlines()
         cells = []
         cell_lines = []
+        kwargs = {}
         state = u'codecell'
         for line in lines:
             if line.startswith(u'# <nbformat>') or _encoding_declaration_re.match(line):
                 pass
             elif line.startswith(u'# <codecell>'):
-                cell = self.new_cell(state, cell_lines)
+                cell = self.new_cell(state, cell_lines, **kwargs)
                 if cell is not None:
                     cells.append(cell)
                 state = u'codecell'
                 cell_lines = []
+                kwargs = {}
             elif line.startswith(u'# <htmlcell>'):
-                cell = self.new_cell(state, cell_lines)
+                cell = self.new_cell(state, cell_lines, **kwargs)
                 if cell is not None:
                     cells.append(cell)
                 state = u'htmlcell'
                 cell_lines = []
+                kwargs = {}
             elif line.startswith(u'# <markdowncell>'):
-                cell = self.new_cell(state, cell_lines)
+                cell = self.new_cell(state, cell_lines, **kwargs)
                 if cell is not None:
                     cells.append(cell)
                 state = u'markdowncell'
                 cell_lines = []
+                kwargs = {}
+            elif line.startswith(u'# <rstcell>'):
+                cell = self.new_cell(state, cell_lines, **kwargs)
+                if cell is not None:
+                    cells.append(cell)
+                state = u'rstcell'
+                cell_lines = []
+                kwargs = {}
+            elif line.startswith(u'# <headingcell'):
+                cell = self.new_cell(state, cell_lines, **kwargs)
+                if cell is not None:
+                    cells.append(cell)
+                    cell_lines = []
+                m = re.match(r'# <headingcell level=(?P<level>\d)>',line)
+                if m is not None:
+                    state = u'headingcell'
+                    kwargs = {}
+                    kwargs['level'] = int(m.group('level'))
+                else:
+                    state = u'codecell'
+                    kwargs = {}
+                    cell_lines = []
             else:
                 cell_lines.append(line)
         if cell_lines and state == u'codecell':
@@ -71,7 +99,7 b' class PyReader(NotebookReader):'
         nb = new_notebook(worksheets=[ws])
         return nb
 
-    def new_cell(self, state, lines):
+    def new_cell(self, state, lines, **kwargs):
         if state == u'codecell':
             input = u'\n'.join(lines)
             input = input.strip(u'\n')
@@ -85,6 +113,15 b' class PyReader(NotebookReader):'
             text = self._remove_comments(lines)
             if text:
                 return new_text_cell(u'markdown',source=text)
+        elif state == u'rstcell':
+            text = self._remove_comments(lines)
+            if text:
+                return new_text_cell(u'rst',source=text)
+        elif state == u'headingcell':
+            text = self._remove_comments(lines)
+            level = kwargs.get('level',1)
+            if text:
+                return new_heading_cell(source=text,level=level)
 
     def _remove_comments(self, lines):
         new_lines = []
@@ -135,6 +172,19 b' class PyWriter(NotebookWriter):'
                         lines.extend([u'# <markdowncell>',u''])
                         lines.extend([u'# ' + line for line in input.splitlines()])
                         lines.append(u'')
+                elif cell.cell_type == u'rst':
+                    input = cell.get(u'source')
+                    if input is not None:
+                        lines.extend([u'# <rstcell>',u''])
+                        lines.extend([u'# ' + line for line in input.splitlines()])
+                        lines.append(u'')
+                elif cell.cell_type == u'heading':
+                    input = cell.get(u'source')
+                    level = cell.get(u'level',1)
+                    if input is not None:
+                        lines.extend([u'# <headingcell level=%s>' % level,u''])
+                        lines.extend([u'# ' + line for line in input.splitlines()])
+                        lines.append(u'')
         lines.append('')
         return unicode('\n'.join(lines))
 

IPython/nbformat/v2/rwbase.py

@@ -64,6 +64,8 b' def rejoin_lines(nb):'
                         item = output.get(key, None)
                         if isinstance(item, list):
                             output[key] = u'\n'.join(item)
+            elif cell.cell_type == 'heading':
+                pass
             else: # text cell
                 for key in ['source', 'rendered']:
                     item = cell.get(key, None)
@@ -90,6 +92,8 b' def split_lines(nb):'
                         item = output.get(key, None)
                         if isinstance(item, basestring):
                             output[key] = item.splitlines()
+            elif cell.cell_type == 'heading':
+                pass
             else: # text cell
                 for key in ['source', 'rendered']:
                     item = cell.get(key, None)

IPython/nbformat/v2/tests/nbexamples.py

@@ -4,7 +4,7 b' from base64 import encodestring'
 from ..nbbase import (
     NotebookNode,
     new_code_cell, new_text_cell, new_worksheet, new_notebook, new_output,
-    new_metadata, new_author
+    new_metadata, new_author, new_heading_cell
 )
 
 # some random base64-encoded *bytes*
@@ -32,6 +32,16 b' ws.cells.append(new_text_cell('
     rendered='A random array'
 ))
 
+ws.cells.append(new_text_cell(
+    u'rst',
+    source='A random array',
+))
+
+ws.cells.append(new_heading_cell(
+    u'My Heading',
+    level=2
+))
+
 ws.cells.append(new_code_cell(
     input='a = numpy.random.rand(100)',
     prompt_number=2,
@@ -96,6 +106,14 b' import numpy'
 
 # A random array
 
+# <rstcell>
+
+# A random array
+
+# <headingcell level=2>
+
+# My Heading
+
 # <codecell>
 
 a = numpy.random.rand(100)

IPython/nbformat/v2/tests/test_nbbase.py

@@ -3,7 +3,7 b' from unittest import TestCase'
 from ..nbbase import (
     NotebookNode,
     new_code_cell, new_text_cell, new_worksheet, new_notebook, new_output,
-    new_author, new_metadata
+    new_author, new_metadata, new_heading_cell
 )
 
 class TestCell(TestCase):
@@ -59,6 +59,27 b' class TestCell(TestCase):'
         self.assertEquals(tc.source, u'hi')
         self.assertEquals(tc.rendered, u'hi')
 
+    def test_empty_rst_cell(self):
+        tc = new_text_cell(u'rst')
+        self.assertEquals(tc.cell_type, u'rst')
+        self.assertEquals(u'source' not in tc, True)
+        self.assertEquals(u'rendered' not in tc, True)
+
+    def test_rst_cell(self):
+        tc = new_text_cell(u'rst', 'hi', 'hi')
+        self.assertEquals(tc.source, u'hi')
+        self.assertEquals(tc.rendered, u'hi')
+
+    def test_empty_heading_cell(self):
+        tc = new_heading_cell()
+        self.assertEquals(tc.cell_type, u'heading')
+        self.assertEquals(u'source' not in tc, True)
+
+    def test_heading_cell(self):
+        tc = new_heading_cell(u'My Heading', level=2)
+        self.assertEquals(tc.source, u'My Heading')
+        self.assertEquals(tc.level, 2)
+
 
 class TestWorksheet(TestCase):