moose-lib 0.4.211__py3-none-any.whl → 0.4.213__py3-none-any.whl

moose_lib/dmv2.py

@@ -10,7 +10,7 @@ of data infrastructure using Python and Pydantic models.
 """
 import dataclasses
 import datetime
-from typing import Any, Generic, Optional, TypeVar, Callable, Union, Literal
+from typing import Any, Generic, Optional, TypeVar, Callable, Union, Literal, Awaitable
 from pydantic import BaseModel, ConfigDict, AliasGenerator
 from pydantic.alias_generators import to_camel
 from pydantic.fields import FieldInfo
@@ -23,11 +23,24 @@ _streams: dict[str, "Stream"] = {}
 _ingest_apis: dict[str, "IngestApi"] = {}
 _egress_apis: dict[str, "ConsumptionApi"] = {}
 _sql_resources: dict[str, "SqlResource"] = {}
+_workflows: dict[str, "Workflow"] = {}
 
 T = TypeVar('T', bound=BaseModel)
 U = TypeVar('U', bound=BaseModel)
+T_none = TypeVar('T_none', bound=Union[BaseModel, None])
+U_none = TypeVar('U_none', bound=Union[BaseModel, None])
 type ZeroOrMany[T] = Union[T, list[T], None]
 
+type TaskRunFunc[T_none, U_none] = Union[
+    # Case 1: No input, no output
+    Callable[[], None],
+    # Case 2: No input, with output
+    Callable[[], Union[U_none, Awaitable[U_none]]],
+    # Case 3: With input, no output
+    Callable[[T_none], None],
+    # Case 4: With input, with output
+    Callable[[T_none], Union[U_none, Awaitable[U_none]]]
+]
 
 class Columns(Generic[T]):
     """Provides runtime checked column name access for Moose resources.
@@ -810,3 +823,146 @@ class MaterializedView(SqlResource, BaseTypedResource, Generic[T]):
 
         self.target_table = target_table
         self.config = options
+
+
+@dataclasses.dataclass
+class TaskConfig(Generic[T_none, U_none]):
+    """Configuration for a Task.
+
+    Attributes:
+        run: The handler function that executes the task logic.
+        on_complete: Optional list of tasks to run after this task completes.
+        timeout: Optional timeout string (e.g. "5m", "1h").
+        retries: Optional number of retry attempts.
+    """
+    run: TaskRunFunc[T_none, U_none]
+    on_complete: Optional[list["Task[U_none, Any]"]] = None
+    timeout: Optional[str] = None
+    retries: Optional[int] = None
+
+
+class Task(TypedMooseResource, Generic[T_none, U_none]):
+    """Represents a task that can be executed as part of a workflow.
+
+    Tasks are the basic unit of work in a workflow, with typed input and output.
+    They can be chained together using the on_complete configuration.
+
+    Args:
+        name: The name of the task.
+        config: Configuration specifying the task's behavior.
+        t: The Pydantic model defining the task's input schema
+           (passed via `Task[InputModel, OutputModel](...)`).
+           OutputModel can be None for tasks that don't return a value.
+
+    Attributes:
+        config (TaskConfig[T, U]): The configuration for this task.
+        columns (Columns[T]): Helper for accessing input field names safely.
+        name (str): The name of the task.
+        model_type (type[T]): The Pydantic model associated with this task's input.
+    """
+    config: TaskConfig[T_none, U_none]
+
+    def __init__(self, name: str, config: TaskConfig[T_none, U_none], **kwargs):
+        super().__init__()
+        self._set_type(name, self._get_type(kwargs))
+        self.config = config
+
+    @classmethod
+    def _get_type(cls, keyword_args: dict):
+        t = keyword_args.get('t')
+        if t is None:
+            raise ValueError(f"Use `{cls.__name__}[T, U](name='...')` to supply both input and output types")
+        if not isinstance(t, tuple) or len(t) != 2:
+            raise ValueError(f"Use `{cls.__name__}[T, U](name='...')` to supply both input and output types")
+
+        input_type, output_type = t
+        if input_type is not None and (not isinstance(input_type, type) or not issubclass(input_type, BaseModel)):
+            raise ValueError(f"Input type {input_type} is not a Pydantic model or None")
+        if output_type is not None and (not isinstance(output_type, type) or not issubclass(output_type, BaseModel)):
+            raise ValueError(f"Output type {output_type} is not a Pydantic model or None")
+        return t
+
+    def _set_type(self, name: str, t: tuple[type[T_none], type[U_none]]):
+        input_type, output_type = t
+        self._t = input_type
+        self._u = output_type
+        self.name = name
+
+
+@dataclasses.dataclass
+class WorkflowConfig:
+    """Configuration for a workflow.
+
+    Attributes:
+        starting_task: The first task to execute in the workflow.
+        retries: Optional number of retry attempts for the entire workflow.
+        timeout: Optional timeout string for the entire workflow.
+        schedule: Optional cron-like schedule string for recurring execution.
+    """
+    starting_task: Task[Any, Any]
+    retries: Optional[int] = None
+    timeout: Optional[str] = None
+    schedule: Optional[str] = None
+
+
+class Workflow:
+    """Represents a workflow composed of one or more tasks.
+
+    Workflows define a sequence of tasks to be executed, with optional
+    scheduling, retries, and timeouts at the workflow level.
+
+    Args:
+        name: The name of the workflow.
+        config: Configuration specifying the workflow's behavior.
+
+    Attributes:
+        name (str): The name of the workflow.
+        config (WorkflowConfig): The configuration for this workflow.
+    """
+    def __init__(self, name: str, config: WorkflowConfig):
+        self.name = name
+        self.config = config
+        # Register the workflow in the internal registry
+        _workflows[name] = self
+
+    def get_task_names(self) -> list[str]:
+        """Get a list of all task names in this workflow.
+
+        Returns:
+            list[str]: List of task names in the workflow, including all child tasks
+        """
+        def collect_task_names(task: Task) -> list[str]:
+            names = [task.name]
+            if task.config.on_complete:
+                for child in task.config.on_complete:
+                    names.extend(collect_task_names(child))
+            return names
+
+        return collect_task_names(self.config.starting_task)
+
+    def get_task(self, task_name: str) -> Optional[Task]:
+        """Find a task in this workflow by name.
+
+        Args:
+            task_name: The name of the task to find
+
+        Returns:
+            Optional[Task]: The task if found, None otherwise
+        """
+        def find_task(task: Task) -> Optional[Task]:
+            if task.name == task_name:
+                return task
+            if task.config.on_complete:
+                for child in task.config.on_complete:
+                    found = find_task(child)
+                    if found:
+                        return found
+            return None
+
+        return find_task(self.config.starting_task)
+
+def _get_workflows() -> dict[str, Workflow]:
+    return _workflows
+
+def _get_workflow(name: str) -> Optional[Workflow]:
+    return _workflows.get(name)

moose_lib/internal.py

@@ -11,7 +11,7 @@ from typing import Literal, Optional, List, Any
 from pydantic import BaseModel, ConfigDict, AliasGenerator
 import json
 from .data_models import Column, _to_columns
-from moose_lib.dmv2 import _tables, _streams, _ingest_apis, _egress_apis, SqlResource, _sql_resources
+from moose_lib.dmv2 import _tables, _streams, _ingest_apis, _egress_apis, SqlResource, _sql_resources, _workflows
 from moose_lib.dmv2 import OlapTable, View, MaterializedView
 from pydantic.alias_generators import to_camel
 from pydantic.json_schema import JsonSchemaValue
@@ -131,6 +131,22 @@ class EgressApiConfig(BaseModel):
     version: Optional[str] = None
     metadata: Optional[dict] = None
 
+class WorkflowJson(BaseModel):
+    """Internal representation of a workflow configuration for serialization.
+
+    Attributes:
+        name: Name of the workflow.
+        retries: Optional number of retry attempts for the entire workflow.
+        timeout: Optional timeout string for the entire workflow.
+        schedule: Optional cron-like schedule string for recurring execution.
+    """
+    model_config = model_config
+
+    name: str
+    retries: Optional[int] = None
+    timeout: Optional[str] = None
+    schedule: Optional[str] = None
+
 class InfrastructureSignatureJson(BaseModel):
     """Represents the unique signature of an infrastructure component (Table, Topic, etc.).
 
@@ -175,6 +191,7 @@ class InfrastructureMap(BaseModel):
         ingest_apis: Dictionary mapping ingest API names to their configurations.
         egress_apis: Dictionary mapping egress API names to their configurations.
         sql_resources: Dictionary mapping SQL resource names to their configurations.
+        workflows: Dictionary mapping workflow names to their configurations.
     """
     model_config = model_config
 
@@ -183,6 +200,7 @@ class InfrastructureMap(BaseModel):
     ingest_apis: dict[str, IngestApiConfig]
     egress_apis: dict[str, EgressApiConfig]
     sql_resources: dict[str, SqlResourceConfig]
+    workflows: dict[str, WorkflowJson]
 
 
 def _map_sql_resource_ref(r: Any) -> InfrastructureSignatureJson:
@@ -233,6 +251,7 @@ def to_infra_map() -> dict:
     ingest_apis = {}
     egress_apis = {}
     sql_resources = {}
+    workflows = {}
 
     for name, table in _tables.items():
         engine = table.config.engine
@@ -308,12 +327,21 @@ def to_infra_map() -> dict:
             metadata=getattr(resource, "metadata", None),
         )
 
+    for name, workflow in _workflows.items():
+        workflows[name] = WorkflowJson(
+            name=workflow.name,
+            retries=workflow.config.retries,
+            timeout=workflow.config.timeout,
+            schedule=workflow.config.schedule,
+        )
+
     infra_map = InfrastructureMap(
         tables=tables,
         topics=topics,
         ingest_apis=ingest_apis,
         egress_apis=egress_apis,
-        sql_resources=sql_resources
+        sql_resources=sql_resources,
+        workflows=workflows
     )
 
     return infra_map.model_dump(by_alias=True)

{moose_lib-0.4.211.dist-info → moose_lib-0.4.213.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: moose_lib
-Version: 0.4.211
+Version: 0.4.213
 Home-page: https://www.fiveonefour.com/moose
 Author: Fiveonefour Labs Inc.
 Author-email: support@fiveonefour.com
@@ -11,6 +11,7 @@ Requires-Dist: pydantic==2.10.6
 Requires-Dist: temporalio==1.9.0
 Requires-Dist: kafka-python-ng==2.2.2
 Requires-Dist: redis==6.2.0
+Requires-Dist: humanfriendly==10.0
 Dynamic: author
 Dynamic: author-email
 Dynamic: description

{moose_lib-0.4.211.dist-info → moose_lib-0.4.213.dist-info}/RECORD

@@ -3,8 +3,8 @@ moose_lib/blocks.py,sha256=_wdvC2NC_Y3MMEnB71WTgWbeQ--zPNHk19xjToJW0C0,3185
 moose_lib/commons.py,sha256=BV5X78MuOWHiZV9bsWSN69JIvzTNWUi-gnuMiAtaO8A,2489
 moose_lib/data_models.py,sha256=R6do1eQqHK6AZ4GTP5tOPtSZaltjZurfx9_Asji7Dwc,8529
 moose_lib/dmv2-serializer.py,sha256=CL_Pvvg8tJOT8Qk6hywDNzY8MYGhMVdTOw8arZi3jng,49
-moose_lib/dmv2.py,sha256=jSqN9ST7xhYCqUP4kJLDxIHMCsRvbkiTJAd27kb6Nfk,31803
-moose_lib/internal.py,sha256=gREvC3XxBFN4i7JL5uMj0riCu_JUO2YyiMZvCokg1ME,13101
+moose_lib/dmv2.py,sha256=87TbxF1cJLUA6vOpPs_zez9Y9obfxadwe1t4KgqrO-U,37599
+moose_lib/internal.py,sha256=nNb1e-grgJ3jpo4H_4_JsKrmH6wL202zWsaVCmvk3yE,14085
 moose_lib/main.py,sha256=In-u7yA1FsLDeP_2bhIgBtHY_BkXaZqDwf7BxwyC21c,8471
 moose_lib/query_param.py,sha256=AB5BKu610Ji-h1iYGMBZKfnEFqt85rS94kzhDwhWJnc,6288
 moose_lib/tasks.py,sha256=6MXA0j7nhvQILAJVTQHCAsquwrSOi2zAevghAc_7kXs,1554
@@ -16,7 +16,7 @@ tests/__init__.py,sha256=0Gh4yzPkkC3TzBGKhenpMIxJcRhyrrCfxLSfpTZnPMQ,53
 tests/conftest.py,sha256=ZVJNbnr4DwbcqkTmePW6U01zAzE6QD0kNAEZjPG1f4s,169
 tests/test_moose.py,sha256=mBsx_OYWmL8ppDzL_7Bd7xR6qf_i3-pCIO3wm2iQNaA,2136
 tests/test_redis_client.py,sha256=d9_MLYsJ4ecVil_jPB2gW3Q5aWnavxmmjZg2uYI3LVo,3256
-moose_lib-0.4.211.dist-info/METADATA,sha256=agj4yuAWI5wQByL5C5aaL4jWs8HWyqICf_TrSe0ocWg,603
-moose_lib-0.4.211.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-moose_lib-0.4.211.dist-info/top_level.txt,sha256=XEns2-4aCmGp2XjJAeEH9TAUcGONLnSLy6ycT9FSJh8,16
-moose_lib-0.4.211.dist-info/RECORD,,
+moose_lib-0.4.213.dist-info/METADATA,sha256=T1Cywpt6LEEhL8OFyFb0wy15WKteaV7NHIBG83Pqgng,638
+moose_lib-0.4.213.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+moose_lib-0.4.213.dist-info/top_level.txt,sha256=XEns2-4aCmGp2XjJAeEH9TAUcGONLnSLy6ycT9FSJh8,16
+moose_lib-0.4.213.dist-info/RECORD,,