Jupytext plugin for the pre-commit package, compatible with black and flake8

the pre-commit package (described in this blog) provides a framework for managing precommit hooks. It would be great if jupytext had an option that would allow it to be a pre-commit hook with that package – essentially it would just need to suppress the stdout message and return 0, adding the translated file to the index. Specifically, this script is working for me now:

import sys
import subprocess
from pathlib import Path

cmd = f"jupytext --to py:percent {sys.argv[1]}"
notebook = Path(sys.argv[1])
pyfile = str(notebook.with_suffix(".py"))
status1, output1 = subprocess.getstatusoutput(cmd)
cmd = f"git add {pyfile}"
status2, output2 = subprocess.getstatusoutput(cmd)
with open("jupytext_logfile.txt", "w") as f:
    f.write(
        (f"{sys.argv}\n" f"{status1}\n" f"{output1}\n" f"{status2}\n" f"{output2}\n")
    )
sys.exit(0)

when I save it as pre_commit_jupytext.py and call it with a pre-commit config entry in .pre-commit-config.yaml that looks like this:

repos
-   repo: local
    hooks:
    - id: jupytext
      name: jupytext
      entry: python a301/scripts/pre_commit_jupytext.py
      language: system

but I think smoothing this a little with a command line option would be a nice addition.

[mwouts]

Thank you @KwatME for your kind words. Well you may be interested in the jupytext.read(s)/write(s) methods. They work similarly to nbformat.read(s)/write(s) (which you could use in place of json.load and json.dump to open/save notebook files).

[KwatMDPhD]

Agree. I think jupytext is wonderful because it can do so much and well liked by the community. I want to see how I can start using it 😃 Thanks @mwouts