wt-task 0.1.0__tar.gz

wt_task-0.1.0/PKG-INFO

@@ -0,0 +1,200 @@
+Metadata-Version: 2.4
+Name: wt-task
+Version: 0.1.0
+Summary: Task decorator and execution features for wt ecosystem
+Author: Ecoscope Workflows Team
+License: Apache-2.0
+Keywords: workflow,task,execution,decorator
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: wt-contracts<1.0.0,>=0.1.0
+Requires-Dist: pydantic<3.0.0,>=2.0.0
+Requires-Dist: typing-extensions>=4.0.0; python_version < "3.11"
+Provides-Extra: gcp
+Requires-Dist: opentelemetry-api>=1.20.0; extra == "gcp"
+Requires-Dist: opentelemetry-sdk>=1.20.0; extra == "gcp"
+Requires-Dist: opentelemetry-exporter-gcp-trace>=1.6.0; extra == "gcp"
+
+# wt-task
+
+Task decorator and execution features for the wt ecosystem.
+
+## Overview
+
+`wt-task` provides the `@task` decorator and task execution capabilities for workflow systems. It implements the `TaskProtocol` from `wt-contracts`, enabling type-safe task execution across the wt ecosystem.
+
+## Features
+
+- **Dual-purpose `task` decorator**: Works as both a decorator (`@task`) and wrapper function (`task(func)`)
+- **Execution methods**: `call`, `map`, `mapvalues` for different execution patterns
+- **Partial application**: Bind arguments with `.partial()` before execution
+- **Validation**: Pydantic-based validation with `.validate()`
+- **Error handling**: Wrap errors with task instance context
+- **Conditional skipping**: Skip execution based on conditions with `.skipif()`
+- **Custom executors**: Switch execution backends with `.set_executor()`
+- **OpenTelemetry tracing**: Optional tracing support (requires `wt-task[gcp]`)
+
+## Installation
+
+```bash
+# Basic installation
+pip install wt-task
+
+# With tracing support
+pip install wt-task[gcp]
+
+# Development installation
+cd wt/wt-task
+uv sync
+```
+
+## Usage
+
+### Basic Task Definition
+
+```python
+from wt_task import task
+
+@task
+def add(a: int, b: int) -> int:
+    return a + b
+
+# Direct call
+result = add(1, 2)  # 3
+
+# Explicit call
+result = add.call(1, 2)  # 3
+```
+
+### Partial Application
+
+```python
+@task
+def multiply(a: int, b: int) -> int:
+    return a * b
+
+# Bind one argument
+multiply_by_2 = multiply.partial(b=2)
+result = multiply_by_2.call(a=5)  # 10
+```
+
+### Mapping Over Values
+
+```python
+@task
+def square(x: int) -> int:
+    return x * x
+
+# Map over a sequence
+results = square.map("x", [1, 2, 3, 4])  # [1, 4, 9, 16]
+```
+
+### Mapping Over Key-Value Pairs
+
+```python
+@task
+def process(data: str) -> int:
+    return len(data)
+
+# Map over key-value pairs, preserving keys
+results = process.mapvalues("data", [
+    ("a", "hello"),
+    ("b", "world"),
+])
+# [("a", 5), ("b", 5)]
+```
+
+### Validation
+
+```python
+@task
+def add_numbers(a: int, b: int) -> int:
+    return a + b
+
+# Parse string inputs to ints
+result = add_numbers.validate().call("10", "20")  # 30
+```
+
+### Error Handling
+
+```python
+@task
+def divide(a: int, b: int) -> float:
+    return a / b
+
+# Add task instance context to errors
+task_with_id = divide.set_task_instance_id("task-1")
+try:
+    task_with_id.handle_errors().call(10, 0)
+except TaskInstanceError as e:
+    print(e)  # Task instance 'task-1' raised ZeroDivisionError(...)
+```
+
+### Method Chaining
+
+All methods return a new task instance, enabling clean method chaining:
+
+```python
+result = (
+    add
+    .partial(a=10)
+    .validate()
+    .set_task_instance_id("add-task")
+    .handle_errors()
+    .call(b="20")
+)  # 30
+```
+
+## Architecture
+
+`wt-task` is designed to work seamlessly with other wt packages:
+
+- **wt-contracts**: Implements the `TaskProtocol` interface
+- **wt-compiler**: Generated DAG code uses `task()` wrapper function
+- **wt-registry**: Task functions are registered separately with `@register`
+
+## Development
+
+```bash
+# Create environment and install dependencies
+cd wt/wt-task
+uv sync
+
+# Run tests
+uv run pytest
+
+# Type checking
+uv run mypy src/
+
+# Linting
+uv run ruff check .
+
+# Formatting
+uv run ruff format .
+```
+
+## Testing
+
+The package includes comprehensive tests for all features:
+
+```bash
+# Run all tests
+uv run pytest
+
+# Run with coverage
+uv run pytest --cov=wt_task --cov-report=html
+
+# Run specific test file
+uv run pytest tests/test_decorator.py
+```
+
+## License
+
+Apache-2.0

wt_task-0.1.0/README.md

@@ -0,0 +1,176 @@
+# wt-task
+
+Task decorator and execution features for the wt ecosystem.
+
+## Overview
+
+`wt-task` provides the `@task` decorator and task execution capabilities for workflow systems. It implements the `TaskProtocol` from `wt-contracts`, enabling type-safe task execution across the wt ecosystem.
+
+## Features
+
+- **Dual-purpose `task` decorator**: Works as both a decorator (`@task`) and wrapper function (`task(func)`)
+- **Execution methods**: `call`, `map`, `mapvalues` for different execution patterns
+- **Partial application**: Bind arguments with `.partial()` before execution
+- **Validation**: Pydantic-based validation with `.validate()`
+- **Error handling**: Wrap errors with task instance context
+- **Conditional skipping**: Skip execution based on conditions with `.skipif()`
+- **Custom executors**: Switch execution backends with `.set_executor()`
+- **OpenTelemetry tracing**: Optional tracing support (requires `wt-task[gcp]`)
+
+## Installation
+
+```bash
+# Basic installation
+pip install wt-task
+
+# With tracing support
+pip install wt-task[gcp]
+
+# Development installation
+cd wt/wt-task
+uv sync
+```
+
+## Usage
+
+### Basic Task Definition
+
+```python
+from wt_task import task
+
+@task
+def add(a: int, b: int) -> int:
+    return a + b
+
+# Direct call
+result = add(1, 2)  # 3
+
+# Explicit call
+result = add.call(1, 2)  # 3
+```
+
+### Partial Application
+
+```python
+@task
+def multiply(a: int, b: int) -> int:
+    return a * b
+
+# Bind one argument
+multiply_by_2 = multiply.partial(b=2)
+result = multiply_by_2.call(a=5)  # 10
+```
+
+### Mapping Over Values
+
+```python
+@task
+def square(x: int) -> int:
+    return x * x
+
+# Map over a sequence
+results = square.map("x", [1, 2, 3, 4])  # [1, 4, 9, 16]
+```
+
+### Mapping Over Key-Value Pairs
+
+```python
+@task
+def process(data: str) -> int:
+    return len(data)
+
+# Map over key-value pairs, preserving keys
+results = process.mapvalues("data", [
+    ("a", "hello"),
+    ("b", "world"),
+])
+# [("a", 5), ("b", 5)]
+```
+
+### Validation
+
+```python
+@task
+def add_numbers(a: int, b: int) -> int:
+    return a + b
+
+# Parse string inputs to ints
+result = add_numbers.validate().call("10", "20")  # 30
+```
+
+### Error Handling
+
+```python
+@task
+def divide(a: int, b: int) -> float:
+    return a / b
+
+# Add task instance context to errors
+task_with_id = divide.set_task_instance_id("task-1")
+try:
+    task_with_id.handle_errors().call(10, 0)
+except TaskInstanceError as e:
+    print(e)  # Task instance 'task-1' raised ZeroDivisionError(...)
+```
+
+### Method Chaining
+
+All methods return a new task instance, enabling clean method chaining:
+
+```python
+result = (
+    add
+    .partial(a=10)
+    .validate()
+    .set_task_instance_id("add-task")
+    .handle_errors()
+    .call(b="20")
+)  # 30
+```
+
+## Architecture
+
+`wt-task` is designed to work seamlessly with other wt packages:
+
+- **wt-contracts**: Implements the `TaskProtocol` interface
+- **wt-compiler**: Generated DAG code uses `task()` wrapper function
+- **wt-registry**: Task functions are registered separately with `@register`
+
+## Development
+
+```bash
+# Create environment and install dependencies
+cd wt/wt-task
+uv sync
+
+# Run tests
+uv run pytest
+
+# Type checking
+uv run mypy src/
+
+# Linting
+uv run ruff check .
+
+# Formatting
+uv run ruff format .
+```
+
+## Testing
+
+The package includes comprehensive tests for all features:
+
+```bash
+# Run all tests
+uv run pytest
+
+# Run with coverage
+uv run pytest --cov=wt_task --cov-report=html
+
+# Run specific test file
+uv run pytest tests/test_decorator.py
+```
+
+## License
+
+Apache-2.0

wt_task-0.1.0/pyproject.toml

@@ -0,0 +1,143 @@
+[build-system]
+requires = ["setuptools>=64", "setuptools-scm>=8"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "wt-task"
+dynamic = ["version"]
+description = "Task decorator and execution features for wt ecosystem"
+readme = "README.md"
+requires-python = ">=3.10"
+license = { text = "Apache-2.0" }
+authors = [
+    { name = "Ecoscope Workflows Team" }
+]
+keywords = ["workflow", "task", "execution", "decorator"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: Apache Software License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+]
+dependencies = [
+    "wt-contracts>=0.1.0,<1.0.0",
+    "pydantic>=2.0.0,<3.0.0",
+    "typing-extensions>=4.0.0; python_version < '3.11'",
+]
+
+[project.optional-dependencies]
+gcp = [
+    "opentelemetry-api>=1.20.0",
+    "opentelemetry-sdk>=1.20.0",
+    "opentelemetry-exporter-gcp-trace>=1.6.0",
+]
+
+[dependency-groups]
+dev = [
+    "pytest>=7.0.0",
+    "pytest-cov>=4.0.0",
+    "pytest-asyncio>=0.21.0",
+    "mypy>=1.0.0",
+    "ruff>=0.1.0",
+]
+
+[tool.uv.sources]
+# Development override - uses local editable wt-contracts
+# Published packages will reference PyPI version normally
+wt-contracts = { path = "../wt-contracts", editable = true }
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools_scm]
+# Version comes from git tags matching this pattern
+tag_regex = "^wt-task/v(?P<version>[0-9.]+)$"
+# Root is the git repository root (parent of wt/)
+root = ".."
+# Generate version file
+version_file = "src/wt_task/_version.py"
+git_describe_command = ["git", "describe", "--dirty", "--tags", "--long", "--match", "wt-task/v*"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = ["test_*.py"]
+addopts = "--doctest-modules --doctest-continue-on-failure --cov=wt_task --cov-report=term-missing"
+
+[tool.mypy]
+python_version = "3.10"
+strict = true
+warn_return_any = true
+warn_unused_configs = true
+disallow_untyped_defs = true
+disallow_any_generics = true
+disallow_subclassing_any = true
+disallow_untyped_calls = true
+disallow_incomplete_defs = true
+check_untyped_defs = true
+no_implicit_optional = true
+warn_redundant_casts = true
+warn_unused_ignores = true
+warn_no_return = true
+
+[[tool.mypy.overrides]]
+module = [
+    "opentelemetry",
+    "opentelemetry.*",
+]
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = [
+    "wt_task.tracing._config",
+    "wt_task.tracing._decorator",
+]
+warn_unused_ignores = false
+
+[tool.ruff]
+line-length = 100
+target-version = "py310"
+
+[tool.ruff.lint]
+select = [
+    "E",  # pycodestyle errors
+    "W",  # pycodestyle warnings
+    "F",  # pyflakes
+    "I",  # isort
+    "B",  # flake8-bugbear
+    "C4", # flake8-comprehensions
+    "UP", # pyupgrade
+]
+ignore = [
+    "E501", # line too long (handled by formatter)
+]
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["F401"] # unused imports in __init__.py are often intentional
+
+# ============================================================================
+# Pixi Build Configuration (conda package building)
+# ============================================================================
+
+[tool.pixi.package]
+name = "wt-task"
+version = "0.1.0"
+
+[tool.pixi.package.build]
+backend = { name = "pixi-build-python", version = "*", channels = ["conda-forge", "https://prefix.dev/pixi-build-backends"] }
+
+[tool.pixi.package.build.config]
+noarch = true
+
+[tool.pixi.package.host-dependencies]
+python = ">=3.10,<3.16"
+setuptools = ">=64"
+setuptools-scm = ">=8"
+pip = "*"
+
+[tool.pixi.package.run-dependencies]
+python = ">=3.10,<3.16"
+pydantic = ">=2.0.0,<3.0.0"
+wt-contracts = ">=0.1.0,<1.0.0"

wt_task-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

wt_task-0.1.0/src/wt_task/__init__.py

@@ -0,0 +1,54 @@
+"""Task decorator and execution features for wt ecosystem.
+
+This package provides the @task decorator and task execution capabilities
+for workflow systems. It implements the TaskProtocol from wt-contracts.
+"""
+
+from .async_task import AsyncTask
+from .decorator import Task, task
+from .exceptions import TaskInstanceError, handle_errors
+from .executors import (
+    AsyncExecutor,
+    Future,
+    FutureSequence,
+    PythonExecutor,
+    SyncExecutor,
+)
+from .skip import (
+    SKIP_SENTINEL,
+    SkippedDependencyFallback,
+    SkipSentinel,
+    skipif,
+    unpack_listlike,
+)
+from .sync_task import SyncTask
+
+__all__ = [
+    # Main decorator and task types
+    "task",
+    "Task",
+    "SyncTask",
+    "AsyncTask",
+    # Exceptions
+    "TaskInstanceError",
+    "handle_errors",
+    # Executors
+    "SyncExecutor",
+    "AsyncExecutor",
+    "PythonExecutor",
+    "Future",
+    "FutureSequence",
+    # Skip utilities
+    "SkipSentinel",
+    "SKIP_SENTINEL",
+    "SkippedDependencyFallback",
+    "skipif",
+    "unpack_listlike",
+]
+
+# Version will be set by setuptools-scm
+try:
+    from ._version import __version__, __version_tuple__
+except ImportError:
+    __version__ = "0.0.0.dev0"
+    __version_tuple__ = (0, 0, 0, "dev0")

wt_task-0.1.0/src/wt_task/_version.py

@@ -0,0 +1,34 @@
+# file generated by setuptools-scm
+# don't change, don't track in version control
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
+
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from typing import Tuple
+    from typing import Union
+
+    VERSION_TUPLE = Tuple[Union[int, str], ...]
+    COMMIT_ID = Union[str, None]
+else:
+    VERSION_TUPLE = object
+    COMMIT_ID = object
+
+version: str
+__version__: str
+__version_tuple__: VERSION_TUPLE
+version_tuple: VERSION_TUPLE
+commit_id: COMMIT_ID
+__commit_id__: COMMIT_ID
+
+__version__ = version = '0.1.0'
+__version_tuple__ = version_tuple = (0, 1, 0)
+
+__commit_id__ = commit_id = 'gfa96100'