diff --git a/nbconvert.py b/nbconvert.py
index 485dde3..f4ea5bd 100755
--- a/nbconvert.py
+++ b/nbconvert.py
@@ -56,14 +56,17 @@ class Converter(object):
def dispatch(self, cell_type):
"""return cell_type dependent render method, for example render_code
"""
+ # XXX: unknown_cell here is RST specific - make it generic
return getattr(self, 'render_' + cell_type, unknown_cell)
def convert(self):
lines = []
+ lines.extend(self.optional_header())
for cell in self.nb.worksheets[0].cells:
conv_fn = self.dispatch(cell.cell_type)
lines.extend(conv_fn(cell))
lines.append('')
+ lines.extend(self.optional_footer())
return '\n'.join(lines)
def render(self):
@@ -87,6 +90,12 @@ class Converter(object):
f.write(self.output.encode(encoding))
return infile
+ def optional_header():
+ pass
+
+ def optional_footer():
+ pass
+
def render_heading(self, cell):
"""convert a heading cell
@@ -201,6 +210,99 @@ class ConverterRST(Converter):
return lines
+class ConverterQuickHTML(Converter):
+ extension = 'html'
+ figures_counter = 0
+
+ def optional_header(self):
+ # XXX: inject the IPython standard CSS into here
+ s = """
+
+
+
+
+ """
+ return s.splitlines()
+
+ def optional_footer(self):
+ s = """
+
+ """
+ return s.splitlines()
+
+ @DocInherit
+ def render_heading(self, cell):
+ marker = cell.level
+ return ['\n {0}\n'.format(cell.source, marker)]
+
+ @DocInherit
+ def render_code(self, cell):
+ if not cell.input:
+ return []
+
+ lines = ['']
+ lines.append('| In [%s]: | ' % cell.prompt_number)
+ lines.append("
\n".join(cell.input.splitlines()))
+ lines.append(' |
')
+
+ for output in cell.outputs:
+ lines.append(' | ')
+ conv_fn = self.dispatch(output.output_type)
+ lines.extend(conv_fn(output))
+ lines.append(' |
')
+
+ lines.append('
')
+ return lines
+
+ @DocInherit
+ def render_markdown(self, cell):
+ return [""+cell.source+"
"]
+
+ @DocInherit
+ def render_plaintext(self, cell):
+ return [""+cell.source+"
"]
+
+ @DocInherit
+ def render_pyout(self, output):
+ lines = ['| Out[%s]: |
' % output.prompt_number, '']
+
+ # output is a dictionary like object with type as a key
+ if 'latex' in output:
+ lines.append("")
+ lines.extend(indent(output.latex))
+ lines.append("")
+
+ if 'text' in output:
+ lines.append("")
+ lines.extend(indent(output.text))
+ lines.append("")
+
+ return lines
+
+ @DocInherit
+ def render_display_data(self, output):
+ lines = []
+
+ if 'png' in output:
+ infile = 'nb_figure_%s.png' % self.figures_counter
+ fullname = os.path.join(self.dirpath, infile)
+ with open(fullname, 'w') as f:
+ f.write(output.png.decode('base64'))
+
+ self.figures_counter += 1
+ lines.append(' ' % infile)
+ lines.append('')
+
+ return lines
+
+ @DocInherit
+ def render_stream(self, output):
+ lines = []
+
+ if 'text' in output:
+ lines.append(output.text)
+
+ return lines
def rst2simplehtml(infile):
"""Convert a rst file to simplified html suitable for blogger.
@@ -261,6 +363,9 @@ def main(infile, format='rst'):
converter = ConverterRST(infile)
rstfname = converter.render()
rst2simplehtml(rstfname)
+ elif format == 'quick-html':
+ converter = ConverterQuickHTML(infile)
+ rstfname = converter.render()
if __name__ == '__main__':
|