noxfile.py

# noxfile.py
import logging
import os
import sys
import tempfile

import nox

nox.options.sessions = "lint", "safety", "tests", "docs"
locations = "src", "tests", "noxfile.py", "docs/conf.py"

if "GITHUB_TOKEN" in os.environ:
    env = {"GITHUB_TOKEN": os.environ["GITHUB_TOKEN"]}
else:
    logging.error("No GITHUB_TOKEN environment variable set nor any .env file exists")
    sys.exit(0)


@nox.session()
def tests(session):
    session.run("poetry", "install", external=True)
    session.run("pytest", env=env)


@nox.session()
def lint(session):
    args = session.posargs or locations
    session.install(
        "flake8",
        "flake8-annotations",
        "flake8-bandit",
        "flake8-black",
        "flake8-bugbear",
        "flake8-docstrings",
        "flake8-import-order",
        "darglint",
    )
    session.run("flake8", *args)


@nox.session()
def black(session):
    args = session.posargs or locations
    session.install("black")
    session.run("black", *args)


@nox.session()
def safety(session):
    with tempfile.NamedTemporaryFile() as requirements:
        session.run(
            "poetry",
            "export",
            "--dev",
            "--format=requirements.txt",
            "--without-hashes",
            f"--output={requirements.name}",
            external=True,
        )
        session.install("safety")
        session.run("safety", "check", f"--file={requirements.name}", "--full-report")


@nox.session()
def docs(session) -> None:
    """Build the documentation."""
    session.run("poetry", "install", "--no-dev", external=True)
    session.install("sphinx")
    session.install("sphinx-autodoc-typehints")
    session.install("myst_parser")
    session.run("sphinx-build", "docs", "docs/_build")