tasks.py

import os
import sys

from invoke import Collection, task

from _docs import bundle, intl, site

# Disable pipenv warning, we run potentially inside the virtualenv already,
# Visual Code e.g. picks it up and there is no harm in that. This is only
# to not confuse people with a pipenv warning that it may not be working.
os.environ["PIPENV_VERBOSITY"] = "-1"
os.environ["PIPENV_IGNORE_VIRTUALENVS"] = "1"


@task
def virtualenv(c):
    """create and install env"""
    c.run(f"{sys.executable} -m pip install -U pipenv")
    c.run(f"{sys.executable} -m pipenv install --dev")

    # Workaround pipenv failing to pin black version due to it being pre-release
    # always.
    c.run(f"{sys.executable} -m pipenv run python -m pip install black==24.2.0")


@task
def run(c, target="build-site"):
    """
    :target: can be `update-docs`, `build-site`, `serve-site`
    """
    c.run(f"{sys.executable} -m pipenv run python update.py --{target}")


ns = Collection(intl, site, bundle, run, virtualenv)