Add 'repos/61a2fae8d94ede667f4a/' from commit 'cfd24fbd92e4e85f7a2a64386c401c91fa9f7baf'

git-subtree-dir: repos/61a2fae8d94ede667f4a
git-subtree-mainline: 0c1d679eb2
git-subtree-split: cfd24fbd92

repos/61a2fae8d94ede667f4a/remove_output.py

@@ -0,0 +1,28 @@
+"""
+Usage: python remove_output.py notebook.ipynb [ > without_output.ipynb ]
+Modified from remove_output by Minrk
+
+"""
+import sys
+import io
+import os
+from IPython.nbformat.current import read, write
+
+
+def remove_outputs(nb):
+    """remove the outputs from a notebook"""
+    for ws in nb.worksheets:
+        for cell in ws.cells:
+            if cell.cell_type == 'code':
+                cell.outputs = []
+
+if __name__ == '__main__':
+    fname = sys.argv[1]
+    with io.open(fname, 'r') as f:
+        nb = read(f, 'json')
+    remove_outputs(nb)
+    base, ext = os.path.splitext(fname)
+    new_ipynb = "%s_removed%s" % (base, ext)
+    with io.open(new_ipynb, 'w', encoding='utf8') as f:
+        write(nb, f, 'json')
+    print "wrote %s" % new_ipynb