Strip Jupyter notebook outputs as Git pre-commit hook

strip-ipython.py

#!/usr/bin/env python3
"""
Strip Jupyter notebook outputs as Git pre-commit hook

Using Git >= 2.54

git config --global hook.linter.event pre-commit
git config --global hook.linter.command '$HOME/linters/ipython.py'
"""

import subprocess

cmd = ["git", "diff", "--staged", "--name-only", "--diff-filter=ACMR", "-z"]
out = subprocess.check_output(cmd)

files = [f for f in out.decode("utf-8", errors="surrogateescape").split("\0") if f]
notebooks = [f for f in files if f.endswith(".ipynb")]

if notebooks:
    cmd = ["jupyter", "nbconvert", "--clear-output", "--inplace", *notebooks]
    print(" ".join(cmd))
    subprocess.check_call(cmd)

    subprocess.check_call(["git", "add", "--", *notebooks])