tasks.py

from sys import platform
from typing import Optional

from invoke import task


@task
def install(c):
    c.run("conda env update --prune -f environment.yml", pty=_use_pty())


@task
def precommit_install(c):
    c.run("pre-commit install", pty=_use_pty())


@task
def clean_preproc(c):
    c.run("find .cache/ -name 'preproc' -exec rm -r {} +")
    c.run("find .cache/ -name 'cache*' -delete")
    c.run("find .cache/ -name 'tmp*' -delete")


@task
def clean_cache(c):
    c.run("find . -name '*.pyc' -exec rm -f {} +")
    c.run("find . -name '*.pyo' -exec rm -f {} +")
    c.run("find . -name '*~' -exec rm -f {} +")
    c.run("find . -name '__pycache__' -exec rm -fr {} +")
    c.run("rm -fr .mypy_cache")


@task
def clean_test(c):
    c.run("rm -fr .tox/")
    c.run("rm -f .coverage")
    c.run("rm -fr htmlcov/")
    c.run("rm -fr .pytest_cache")


@task
def clean_build(c):
    c.run("rm -fr dist")


@task
def clean(c):
    clean_cache(c)
    clean_test(c)
    clean_build(c)


@task
def build(c):
    clean(c)
    c.run("poetry build", pty=_use_pty())


@task
def docker(c, build=True, tag=False, push=False):
    version = _package_version(c)

    if build:
        c.run("docker build -t perceiver-io .")
    if tag:
        c.run(f"docker tag perceiver-io perceiver-io:{version}")
    if push:
        # TODO: implement push
        ...


@task(aliases=["cc"])
def code_check(c):
    c.run("pre-commit run --all-files", pty=_use_pty())


@task
def test(c, cov=False, cov_report=None):
    _run_pytest(c, "tests --durations=25 --color=yes", cov, cov_report, _use_pty())


def _use_pty():
    return platform != "win32"


def _package_version(c):
    return c.run("poetry version", dry=False, hide="out").stdout.split(" ")[1].strip()


def _run_pytest(c, test_dir, cov=False, cov_report=None, pty=True):
    c.run(f"pytest {test_dir} {_pytest_cov_options(cov, cov_report)}", pty=pty)


def _pytest_cov_options(use_cov: bool, cov_reports: Optional[str]):
    if not use_cov:
        return ""

    cov_report_types = cov_reports.split(",") if cov_reports else []
    cov_report_types = ["term"] + cov_report_types
    cov_report_params = [f"--cov-report {r}" for r in cov_report_types]
    return f"--cov {' '.join(cov_report_params)}"