Remove output cells and execution counts

remove_output.py

"""
Usage: python remove_output.py notebook.ipynb [ > without_output.ipynb ]

Modified from remove_output by:
damianavila (https://gist.github.com/damianavila/5305869)
"""
import sys
import io
import os
from IPython.nbformat import read, write


def remove_outputs(nb):
    """remove the outputs from a notebook"""
    for cell in nb.cells:
        if cell.cell_type == 'code':
            cell.outputs = []
            cell.execution_count = None

if __name__ == '__main__':
    fname = sys.argv[1]
    with io.open(fname, 'r') as f:
        nb = read(f, 4)
        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, 4)
        print "wrote %s" % new_ipynb