noxfile.py

from __future__ import annotations

import nox

PY38 = "3.8"
PY39 = "3.9"
PY310 = "3.10"
PY311 = "3.11"
PY312 = "3.12"
PY_VERSIONS = [PY38, PY39, PY310, PY311, PY312]
PY_DEFAULT = PY38

DJ32 = "3.2"
DJ40 = "4.0"
DJ41 = "4.1"
DJ42 = "4.2"
DJMAIN = "main"
DJMAIN_MIN_PY = PY310
DJ_VERSIONS = [DJ32, DJ40, DJ41, DJ42, DJMAIN]
DJ_DEFAULT = DJ32

PSYCOPG2 = "2"
PSYCOPG3 = "3"
PSYCOPG_VERSIONS = [PSYCOPG2, PSYCOPG3]
PSYCOPG_DEFAULT = PSYCOPG3


def version(ver: str) -> tuple[int, ...]:
    """Convert a string version to a tuple of ints, e.g. "3.10" -> (3, 10)"""
    return tuple(map(int, ver.split(".")))


def should_skip(python: str, django: str, psycopg: str) -> tuple(bool, str | None):
    """Return True if the test should be skipped"""
    if django == DJMAIN and version(python) < version(DJMAIN_MIN_PY):
        return True, f"Django {DJMAIN} requires Python {DJMAIN_MIN_PY}+"

    if django == DJ32 and version(python) >= version(PY312):
        return True, f"Django {DJ32} requires Python < {PY312}"

    if psycopg == PSYCOPG3 and version(python) >= version(PY312):
        return True, f"psycopg3 requires Python < {PY312}"

    return False, None


@nox.session(python=PY_VERSIONS)
@nox.parametrize("django", DJ_VERSIONS)
@nox.parametrize("psycopg", PSYCOPG_VERSIONS)
def tests(session, django, psycopg):
    skip = should_skip(session.python, django, psycopg)
    if skip[0]:
        session.skip(skip[1])

    session.install(".[test]")

    if django == DJMAIN:
        session.install("https://github.com/django/django/archive/refs/heads/main.zip")
    else:
        session.install(f"django=={django}")

    if psycopg == PSYCOPG2:
        session.install("psycopg2-binary")
    elif psycopg == PSYCOPG3:
        session.install("psycopg[binary]")

    session.run("pytest")