hatchet-sdk 0.47.1__py3-none-any.whl → 1.0.0a1__py3-none-any.whl

hatchet_sdk/features/scheduled.py

@@ -1,10 +1,9 @@
 import datetime
-from typing import Any, Coroutine, Dict, List, Optional, Union
+from typing import List, Optional, Union
 
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 
 from hatchet_sdk.client import Client
-from hatchet_sdk.clients.rest.models.cron_workflows import CronWorkflows
 from hatchet_sdk.clients.rest.models.cron_workflows_order_by_field import (
     CronWorkflowsOrderByField,
 )
@@ -12,24 +11,28 @@ from hatchet_sdk.clients.rest.models.scheduled_workflows import ScheduledWorkflo
 from hatchet_sdk.clients.rest.models.scheduled_workflows_list import (
     ScheduledWorkflowsList,
 )
+from hatchet_sdk.clients.rest.models.scheduled_workflows_order_by_field import (
+    ScheduledWorkflowsOrderByField,
+)
 from hatchet_sdk.clients.rest.models.workflow_run_order_by_direction import (
     WorkflowRunOrderByDirection,
 )
+from hatchet_sdk.utils.typing import JSONSerializableMapping
 
 
-class CreateScheduledTriggerInput(BaseModel):
+class CreateScheduledTriggerConfig(BaseModel):
     """
     Schema for creating a scheduled workflow run.
 
     Attributes:
-        input (Dict[str, Any]): The input data for the scheduled workflow.
-        additional_metadata (Dict[str, str]): Additional metadata associated with the future run (e.g. ["key1:value1", "key2:value2"]).
+        input (JSONSerializableMapping): The input data for the scheduled workflow.
+        additional_metadata (JSONSerializableMapping): Additional metadata associated with the future run (e.g. ["key1:value1", "key2:value2"]).
         trigger_at (Optional[datetime.datetime]): The datetime when the run should be triggered.
     """
 
-    input: Dict[str, Any] = {}
-    additional_metadata: Dict[str, str] = {}
-    trigger_at: Optional[datetime.datetime] = None
+    input: JSONSerializableMapping = Field(default_factory=dict)
+    additional_metadata: JSONSerializableMapping = Field(default_factory=dict)
+    trigger_at: datetime.datetime
 
 
 class ScheduledClient:
@@ -51,14 +54,13 @@ class ScheduledClient:
             _client (Client): The client instance to be used for REST interactions.
         """
         self._client = _client
-        self.aio: "ScheduledClientAsync" = ScheduledClientAsync(_client)
 
     def create(
         self,
         workflow_name: str,
         trigger_at: datetime.datetime,
-        input: Dict[str, Any],
-        additional_metadata: Dict[str, str],
+        input: JSONSerializableMapping,
+        additional_metadata: JSONSerializableMapping,
     ) -> ScheduledWorkflows:
         """
         Creates a new scheduled workflow run asynchronously.
@@ -66,14 +68,14 @@ class ScheduledClient:
         Args:
             workflow_name (str): The name of the scheduled workflow.
             trigger_at (datetime.datetime): The datetime when the run should be triggered.
-            input (Dict[str, Any]): The input data for the scheduled workflow.
-            additional_metadata (Dict[str, str]): Additional metadata associated with the future run as a key-value pair (e.g. {"key1": "value1", "key2": "value2"}).
+            input (JSONSerializableMapping): The input data for the scheduled workflow.
+            additional_metadata (JSONSerializableMapping): Additional metadata associated with the future run as a key-value pair (e.g. {"key1": "value1", "key2": "value2"}).
 
         Returns:
             ScheduledWorkflows: The created scheduled workflow instance.
         """
 
-        validated_input = CreateScheduledTriggerInput(
+        validated_input = CreateScheduledTriggerConfig(
             trigger_at=trigger_at, input=input, additional_metadata=additional_metadata
         )
 
@@ -91,16 +93,17 @@ class ScheduledClient:
         Args:
             scheduled (Union[str, ScheduledWorkflows]): The scheduled workflow trigger ID or ScheduledWorkflows instance to delete.
         """
-        id_ = scheduled
-        if isinstance(scheduled, ScheduledWorkflows):
-            id_ = scheduled.metadata.id
-        self._client.rest.schedule_delete(id_)
+        self._client.rest.schedule_delete(
+            scheduled.metadata.id
+            if isinstance(scheduled, ScheduledWorkflows)
+            else scheduled
+        )
 
     def list(
         self,
-        offset: Optional[int] = None,
-        limit: Optional[int] = None,
-        workflow_id: Optional[str] = None,
+        offset: int | None = None,
+        limit: int | None = None,
+        workflow_id: str | None = None,
         additional_metadata: Optional[List[str]] = None,
         order_by_field: Optional[CronWorkflowsOrderByField] = None,
         order_by_direction: Optional[WorkflowRunOrderByDirection] = None,
@@ -109,9 +112,9 @@ class ScheduledClient:
         Retrieves a list of scheduled workflows based on provided filters.
 
         Args:
-            offset (Optional[int]): The starting point for the list.
-            limit (Optional[int]): The maximum number of items to return.
-            workflow_id (Optional[str]): Filter by specific workflow ID.
+            offset (int | None): The starting point for the list.
+            limit (int | None): The maximum number of items to return.
+            workflow_id (str | None): Filter by specific workflow ID.
             additional_metadata (Optional[List[str]]): Filter by additional metadata keys (e.g. ["key1:value1", "key2:value2"]).
             order_by_field (Optional[CronWorkflowsOrderByField]): Field to order the results by.
             order_by_direction (Optional[WorkflowRunOrderByDirection]): Direction to order the results.
@@ -138,37 +141,18 @@ class ScheduledClient:
         Returns:
             ScheduledWorkflows: The requested scheduled workflow instance.
         """
-        id_ = scheduled
-        if isinstance(scheduled, ScheduledWorkflows):
-            id_ = scheduled.metadata.id
-        return self._client.rest.schedule_get(id_)
-
-
-class ScheduledClientAsync:
-    """
-    Asynchronous client for managing scheduled workflows.
-
-    Attributes:
-        _client (Client): The underlying client used to interact with the REST API asynchronously.
-    """
-
-    _client: Client
-
-    def __init__(self, _client: Client) -> None:
-        """
-        Initializes the ScheduledClientAsync with a given Client instance.
-
-        Args:
-            _client (Client): The client instance to be used for asynchronous REST interactions.
-        """
-        self._client = _client
+        return self._client.rest.schedule_get(
+            scheduled.metadata.id
+            if isinstance(scheduled, ScheduledWorkflows)
+            else scheduled
+        )
 
-    async def create(
+    async def aio_create(
         self,
         workflow_name: str,
         trigger_at: datetime.datetime,
-        input: Dict[str, Any],
-        additional_metadata: Dict[str, str],
+        input: JSONSerializableMapping,
+        additional_metadata: JSONSerializableMapping,
     ) -> ScheduledWorkflows:
         """
         Creates a new scheduled workflow run asynchronously.
@@ -176,44 +160,45 @@ class ScheduledClientAsync:
         Args:
             workflow_name (str): The name of the scheduled workflow.
             trigger_at (datetime.datetime): The datetime when the run should be triggered.
-            input (Dict[str, Any]): The input data for the scheduled workflow.
-            additional_metadata (Dict[str, str]): Additional metadata associated with the future run.
+            input (JSONSerializableMapping): The input data for the scheduled workflow.
+            additional_metadata (JSONSerializableMapping): Additional metadata associated with the future run.
 
         Returns:
             ScheduledWorkflows: The created scheduled workflow instance.
         """
-        return await self._client.rest.aio.schedule_create(
+        return await self._client.rest.aio_create_schedule(
             workflow_name, trigger_at, input, additional_metadata
         )
 
-    async def delete(self, scheduled: Union[str, ScheduledWorkflows]) -> None:
+    async def aio_delete(self, scheduled: Union[str, ScheduledWorkflows]) -> None:
         """
         Deletes a scheduled workflow asynchronously.
 
         Args:
             scheduled (Union[str, ScheduledWorkflows]): The scheduled workflow trigger ID or ScheduledWorkflows instance to delete.
         """
-        id_ = scheduled
-        if isinstance(scheduled, ScheduledWorkflows):
-            id_ = scheduled.metadata.id
-        await self._client.rest.aio.schedule_delete(id_)
+        await self._client.rest.aio_delete_schedule(
+            scheduled.metadata.id
+            if isinstance(scheduled, ScheduledWorkflows)
+            else scheduled
+        )
 
-    async def list(
+    async def aio_list(
         self,
-        offset: Optional[int] = None,
-        limit: Optional[int] = None,
-        workflow_id: Optional[str] = None,
+        offset: int | None = None,
+        limit: int | None = None,
+        workflow_id: str | None = None,
         additional_metadata: Optional[List[str]] = None,
-        order_by_field: Optional[CronWorkflowsOrderByField] = None,
+        order_by_field: Optional[ScheduledWorkflowsOrderByField] = None,
         order_by_direction: Optional[WorkflowRunOrderByDirection] = None,
     ) -> ScheduledWorkflowsList:
         """
         Retrieves a list of scheduled workflows based on provided filters asynchronously.
 
         Args:
-            offset (Optional[int]): The starting point for the list.
-            limit (Optional[int]): The maximum number of items to return.
-            workflow_id (Optional[str]): Filter by specific workflow ID.
+            offset (int | None): The starting point for the list.
+            limit (int | None): The maximum number of items to return.
+            workflow_id (str | None): Filter by specific workflow ID.
             additional_metadata (Optional[List[str]]): Filter by additional metadata keys (e.g. ["key1:value1", "key2:value2"]).
             order_by_field (Optional[CronWorkflowsOrderByField]): Field to order the results by.
             order_by_direction (Optional[WorkflowRunOrderByDirection]): Direction to order the results.
@@ -221,7 +206,7 @@ class ScheduledClientAsync:
         Returns:
             ScheduledWorkflowsList: A list of scheduled workflows matching the criteria.
         """
-        return await self._client.rest.aio.schedule_list(
+        return await self._client.rest.aio_list_schedule(
             offset=offset,
             limit=limit,
             workflow_id=workflow_id,
@@ -230,7 +215,7 @@ class ScheduledClientAsync:
             order_by_direction=order_by_direction,
         )
 
-    async def get(
+    async def aio_get(
         self, scheduled: Union[str, ScheduledWorkflows]
     ) -> ScheduledWorkflows:
         """
@@ -242,7 +227,8 @@ class ScheduledClientAsync:
         Returns:
             ScheduledWorkflows: The requested scheduled workflow instance.
         """
-        id_ = scheduled
-        if isinstance(scheduled, ScheduledWorkflows):
-            id_ = scheduled.metadata.id
-        return await self._client.rest.aio.schedule_get(id_)
+        return await self._client.rest.aio_get_schedule(
+            scheduled.metadata.id
+            if isinstance(scheduled, ScheduledWorkflows)
+            else scheduled
+        )