noxfile.py

"""Noxfile."""

import shutil
from pathlib import Path

import nox

nox.options.default_venv_backend = "none"
nox.options.sessions = ["lints"]


CLEANABLE_TARGETS = [
    # Generated bytecode
    "./**/__pycache__",
    "./**/*.pyc",
    "./**/*.pyo",
    # Centralized tool cache
    "./.cache",
]


@nox.session
def lints(session: nox.Session) -> None:
    """Run lints."""
    session.run("prek", "run", "--all-files")
    session.run("ruff", "format", ".")
    session.run("ruff", "check", "--fix", ".")
    session.run("mypy", "--strict", "tools/", "noxfile.py")
    session.run("ty", "check", ".")


@nox.session
def clean(_: nox.Session) -> None:
    """Clean cache, .pyc, .pyo, and test/build artifact files from project."""
    for searchpath in CLEANABLE_TARGETS:
        for filepath in Path().glob(searchpath):
            if filepath.is_dir():
                shutil.rmtree(filepath)
            else:
                filepath.unlink()