Function _setup_for_driver

noxfile.py:63–113

(
    session: nox.Session,
    cmd: List[str],
    basename: str,
    greenlet: bool = False,
)

def _setup_for_driver(
    session: nox.Session,
    cmd: List[str],
    basename: str,
    greenlet: bool = False,
) -> None:

    class="cm"># install driver deps listed out in pyproject.toml
    nogreenlet_deps = fclass="st">"tests-{basename.replace(&class="cm">#x27;_', '-')}"
    greenlet_deps = fclass="st">"tests-{basename.replace(&class="cm">#x27;_', '-')}-asyncio"

    deps = nox.project.dependency_groups(
        pyproject,
        (greenlet_deps if greenlet else nogreenlet_deps),
    )
    if deps:
        session.install(*deps)

    class="cm"># set up top level ``--db`` sent to pytest command line, which looks
    class="cm"># up a base URL in the [db] section of setup.cfg.   Environment variable
    class="cm"># substitution used by CI is also available.

    class="cm"># e.g. TOX_POSTGRESQL, TOX_MYSQL, etc.
    dburl_env = fclass="st">"TOX_{basename.upper()}"
    class="cm"># e.g. --db=postgresql, --db=mysql, etc.
    default_dburl = fclass="st">"--db={basename}"
    cmd.extend(os.environ.get(dburl_env, default_dburl).split())

    class="cm"># set up extra drivers using --dbdriver.   this first looks in
    class="cm"># an environment variable before making use of the DB_CLI_NAMES
    class="cm"># lookup table

    class="cm"># e.g. EXTRA_PG_DRIVERS, EXTRA_MYSQL_DRIVERS, etc.
    if basename == class="st">"postgresql":
        extra_driver_env = class="st">"EXTRA_PG_DRIVERS"
    else:
        extra_driver_env = fclass="st">"EXTRA_{basename.upper()}_DRIVERS"
    env_dbdrivers = os.environ.get(extra_driver_env, None)
    if env_dbdrivers:
        cmd.extend(env_dbdrivers.split())
        return

    class="cm"># use fixed names in DB_CLI_NAMES
    extra_drivers: Dict[str, Set[str]] = DB_CLI_NAMES[basename]
    dbdrivers = extra_drivers[class="st">"nogreenlet"]
    if greenlet:
        dbdrivers.update(extra_drivers[class="st">"greenlet"])

    class="cm"># use equals sign so that we avoid
    class="cm"># https://github.com/pytest-dev/pytest/issues/13913
    cmd.extend([fclass="st">"--dbdriver={dbdriver}" for dbdriver in dbdrivers])


pyproject = nox.project.load_toml(class="st">"pyproject.toml")