jupyter_delete_output.py

# -*- coding: utf-8 -*-

# This file is part of CoCalc: Copyright © 2020 Sagemath, Inc.
# License: AGPLv3 s.t. "Commons Clause" – read LICENSE.md for details

# Based on See https://gist.github.com/damianavila/5305869

from __future__ import absolute_import
import sys, io, os
from nbformat.v4 import reads, writes


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


def main():
    for fname in sys.argv[1:]:
        nb = reads(io.open(fname, 'r').read())
        remove_outputs(nb)
        base, ext = os.path.splitext(fname)
        new_ipynb = "%s-no-output%s" % (base, ext)
        io.open(new_ipynb, 'w', encoding='utf8').write(writes(nb))


if __name__ == "__main__":
    main()