-
Notifications
You must be signed in to change notification settings - Fork 5
/
noxfile.py
86 lines (69 loc) · 2.4 KB
/
noxfile.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
"""Nox sessions."""
from pathlib import Path
from textwrap import dedent
import nox
from nox_poetry import session as noxsession
locations = "bask", "noxfile.py"
nox.options.sessions = ("pre-commit", "tests")
python_versions = ["3.8", "3.9"]
def activate_virtualenv_in_precommit_hooks(session):
"""Activate virtualenv in hooks installed by pre-commit.
This function patches git hooks installed by pre-commit to activate the
session's virtual environment. This allows pre-commit to locate hooks in
that environment when invoked from git.
Args:
session: The Session object.
"""
if session.bin is None:
return
virtualenv = session.env.get("VIRTUAL_ENV")
if virtualenv is None:
return
hookdir = Path(".git") / "hooks"
if not hookdir.is_dir():
return
for hook in hookdir.iterdir():
if hook.name.endswith(".sample") or not hook.is_file():
continue
text = hook.read_text()
bindir = repr(session.bin)[1:-1] # strip quotes
if not (
Path("A") == Path("a") and bindir.lower() in text.lower() or bindir in text
):
continue
lines = text.splitlines()
if not (lines[0].startswith("#!") and "python" in lines[0].lower()):
continue
header = dedent(
f"""\
import os
os.environ["VIRTUAL_ENV"] = {virtualenv!r}
os.environ["PATH"] = os.pathsep.join((
{session.bin!r},
os.environ.get("PATH", ""),
))
"""
)
lines.insert(1, header)
hook.write_text("\n".join(lines))
@noxsession(python=python_versions)
def tests(session):
"""Run the test suite."""
session.install(".[data]")
session.install("pytest", "nox", "nox-poetry")
session.run("pytest", *session.posargs)
@noxsession(python="3.9")
def black(session):
"""Run black code formatter."""
args = session.posargs or locations
session.install("black")
session.run("black", *args)
@noxsession(name="pre-commit", python="3.9")
def precommit(session):
args = session.posargs or ["run", "--all-files", "--show-diff-on-failure"]
session.install(
"pre-commit", "black", "flake8", "doc8", "isort", "flake8-bugbear",
)
session.run("pre-commit", *args)
if args and args[0] == "install":
activate_virtualenv_in_precommit_hooks(session)