noxfile.py

from __future__ import annotations

from pathlib import Path
from typing import TYPE_CHECKING, Literal

import nox

if TYPE_CHECKING:
    from nox import Session

PYTHON_VERSIONS = ["3.9", "3.10", "3.11", "3.12", "3.13"]
PYRIGHT_PYTHON_PYLANCE_VERSION: Literal["latest-release", "latest-prerelease"] = "latest-release"
PYRIGHT_PYTHON_FORCE_VERSION: str | None = None
COMMON_PYTEST_OPTIONS = [
    "--cov",
    "src",
    "--cov-append",
    "--cov-report",
    "term-missing",
    "-n",
    "2",
    "--showlocals",
]

here = Path(__file__).parent


nox.options.error_on_external_run = True
nox.options.default_venv_backend = "uv"


@nox.session(name="unit", python=PYTHON_VERSIONS, tags=["tests"])
def unit_tests(session: Session) -> None:
    (here / ".coverage").unlink(missing_ok=True)
    session.run_install(
        "uv",
        "sync",
        "--all-extras",
        "--group=test",
        env={"UV_PROJECT_ENVIRONMENT": session.virtualenv.location},
    )
    test_files: list[str] = session.posargs or []
    session.run("pytest", *COMMON_PYTEST_OPTIONS, "-vv", *test_files)


@nox.session(name="pyright", python=PYTHON_VERSIONS, tags=["lint"])
def pyright(session: Session) -> None:
    session.run_install(
        "uv",
        "sync",
        "--all-extras",
        "--group=lint",
        env={"UV_PROJECT_ENVIRONMENT": session.virtualenv.location},
    )
    env = {"PYRIGHT_PYTHON_PYLANCE_VERSION": PYRIGHT_PYTHON_PYLANCE_VERSION}
    if PYRIGHT_PYTHON_FORCE_VERSION:
        env["PYRIGHT_PYTHON_FORCE_VERSION"] = PYRIGHT_PYTHON_FORCE_VERSION
    session.run("pyright", "--version", external=True, env=env)
    session.run("pyright", *session.posargs, external=True, env=env)


@nox.session(name="vulture", python=PYTHON_VERSIONS, tags=["lint"])
def vulture(session: Session) -> None:
    session.run_install(
        "uv",
        "sync",
        "--group=lint",
        env={"UV_PROJECT_ENVIRONMENT": session.virtualenv.location},
    )
    session.run("vulture", "src/", "--min-confidence=100", "--sort-by-size", external=True)