Remove output from Jupyter notebook from the command line (dev version 1.0)

remove_jupyter_notebook_cells_output.py

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 = []

def main(fname):
    with io.open(fname, 'r', encoding='latin-1') as f:
        nb = read(f, 'json')

    remove_outputs(nb)
    base, ext = os.path.splitext(fname)
    new_ipynb = f"{base}_removed.ipynb"
    with io.open(new_ipynb, 'w', encoding='latin-1') as f:
        write(nb, f, 'json')

if __name__ == '__main__':
    main(sys.argv[1])