Jupyter Notebook Output Cleaning Script

clean_nbs.py

import io
import sys

from nbformat import read, write

# Handle either stdin or a filename
if __name__ == '__main__':
    for filename in sys.argv[1:]:
        print(filename, file=sys.stderr)
        ipynb = read(filename, 4)
        ipynb.metadata.signature = ''
        for cell in ipynb.cells:
            if "outputs" in cell:
                cell["outputs"] = []
            if "execution_count" in cell:
                cell["execution_count"] = None

        ipynb.metadata.language_info["version"] = None

        with io.open(filename, mode="w", encoding='utf-8') as fh:
            write(ipynb, fh)