noxfile.py

import os
from pathlib import Path

import nox

HERE = Path(__file__).resolve(strict=True).parent
PYTHON_VERSION = HERE.joinpath(".python-version").read_text().strip()

nox.options.pythons = [PYTHON_VERSION]
nox.options.sessions = ["run"]
nox.options.reuse_existing_virtualenvs = True
nox.options.error_on_missing_interpreters = True


@nox.session(python=PYTHON_VERSION)
def lint(session: nox.Session) -> None:
    """Run the linter."""
    session.install("-U", "pre-commit")
    session.run("pre-commit", "run", "--all-files", *session.posargs)


@nox.session(python=PYTHON_VERSION)
def update_requirements(session: nox.Session) -> None:
    """Update requirements.txt."""
    session.install("-U", "pip-tools")
    env = os.environ.copy()
    # CUSTOM_COMPILE_COMMAND is a pip-compile option that tells users how to
    # regenerate the constraints files
    env["CUSTOM_COMPILE_COMMAND"] = f"nox -s {session.name}"
    session.run(
        "pip-compile",
        "--allow-unsafe",
        "--upgrade",
        "--generate-hashes",
        "requirements.in",
        env=env,
    )


@nox.session(python=PYTHON_VERSION)
def run(session: nox.Session) -> None:
    """Run manylinux-timeline."""
    session.install(
        "--only-binary",
        ":all:",
        "--require-hashes",
        "-r",
        "requirements.txt",
    )
    session.run("python", "update.py", *session.posargs)


@nox.session(python=PYTHON_VERSION)
def serve(session: nox.Session) -> None:
    session.run("python", "-m", "http.server", "-d", "build")


@nox.session(python=PYTHON_VERSION, venv_backend="none")
def timestamp(session: nox.Session) -> None:
    """Get timestamp for PyPI package cache on GHA"""
    from datetime import datetime, timezone

    print(datetime.now(timezone.utc).isoformat())