tasks.py

"""Invoke task definitions for project automation."""

from invoke import task


@task
def install(c):
    """Install project with dev dependencies."""
    c.run("pip install -e '.[dev]'")


@task
def test(c, cov=True, mark=None):
    """Run tests."""
    cmd = "pytest"
    if cov:
        cmd += " --cov=src/ai_project --cov-report=html"
    if mark:
        cmd += f" -m {mark}"
    c.run(cmd)


@task
def format(c):
    """Format code with black."""
    c.run("black src tests")


@task
def lint(c):
    """Lint code with ruff."""
    c.run("ruff check src tests --fix")


@task
def type_check(c):
    """Type check with pyright."""
    c.run("pyright src")


@task
def quality(c):
    """Run all quality checks."""
    format(c)
    lint(c)
    type_check(c)
    test(c)


@task
def pre_commit(c):
    """Install pre-commit hooks."""
    c.run("pre-commit install")


@task
def clean(c):
    """Clean build artifacts."""
    c.run("rm -rf build dist *.egg-info .pytest_cache .coverage htmlcov")
    c.run("find . -type d -name __pycache__ -exec rm -rf {} +")
    c.run("find . -type f -name '*.pyc' -delete")