horsies 0.1.0a2__tar.gz → 0.1.0a4__tar.gz

{horsies-0.1.0a2 → horsies-0.1.0a4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: horsies
-Version: 0.1.0a2
+Version: 0.1.0a4
 Summary: A Python library for distributed task execution
 Author: Suleyman Ozkeskin
 License-Expression: MIT

{horsies-0.1.0a2 → horsies-0.1.0a4}/horsies/__init__.py

@@ -14,6 +14,7 @@ from .core.models.tasks import (
     LibraryErrorCode,
     SubWorkflowError,
     RetryPolicy,
+    TaskInfo,
 )
 from .core.models.queues import QueueMode, CustomQueueConfig
 from .core.models.workflow import (
@@ -67,6 +68,7 @@ __all__ = [
     'LibraryErrorCode',
     'SubWorkflowError',
     'RetryPolicy',
+    'TaskInfo',
     'QueueMode',
     'CustomQueueConfig',
     'TaskStatus',

{horsies-0.1.0a2 → horsies-0.1.0a4}/horsies/core/brokers/postgres.py

@@ -16,6 +16,7 @@ from horsies.core.codec.serde import (
     loads_json,
     task_result_from_json,
 )
+from horsies.core.models.tasks import TaskInfo
 from horsies.core.utils.loop_runner import LoopRunner
 from horsies.core.logging import get_logger
 
@@ -854,6 +855,134 @@ class PostgresBroker:
         """
         return self._loop_runner.call(self.get_result_async, task_id, timeout_ms)
 
+    async def get_task_info_async(
+        self,
+        task_id: str,
+        *,
+        include_result: bool = False,
+        include_failed_reason: bool = False,
+    ) -> 'TaskInfo | None':
+        """Fetch metadata for a task by ID."""
+        await self._ensure_initialized()
+
+        async with self.session_factory() as session:
+            base_columns = [
+                'id',
+                'task_name',
+                'status',
+                'queue_name',
+                'priority',
+                'retry_count',
+                'max_retries',
+                'next_retry_at',
+                'sent_at',
+                'claimed_at',
+                'started_at',
+                'completed_at',
+                'failed_at',
+                'worker_hostname',
+                'worker_pid',
+                'worker_process_name',
+            ]
+            if include_result:
+                base_columns.append('result')
+            if include_failed_reason:
+                base_columns.append('failed_reason')
+
+            query = text(
+                f"""
+                SELECT {', '.join(base_columns)}
+                FROM horsies_tasks
+                WHERE id = :id
+            """
+            )
+            result = await session.execute(query, {'id': task_id})
+            row = result.fetchone()
+            if row is None:
+                return None
+
+            result_value = None
+            failed_reason = None
+
+            idx = 0
+            task_id_value = row[idx]
+            idx += 1
+            task_name = row[idx]
+            idx += 1
+            status = TaskStatus(row[idx])
+            idx += 1
+            queue_name = row[idx]
+            idx += 1
+            priority = row[idx]
+            idx += 1
+            retry_count = row[idx] or 0
+            idx += 1
+            max_retries = row[idx] or 0
+            idx += 1
+            next_retry_at = row[idx]
+            idx += 1
+            sent_at = row[idx]
+            idx += 1
+            claimed_at = row[idx]
+            idx += 1
+            started_at = row[idx]
+            idx += 1
+            completed_at = row[idx]
+            idx += 1
+            failed_at = row[idx]
+            idx += 1
+            worker_hostname = row[idx]
+            idx += 1
+            worker_pid = row[idx]
+            idx += 1
+            worker_process_name = row[idx]
+            idx += 1
+
+            if include_result:
+                raw_result = row[idx]
+                idx += 1
+                if raw_result:
+                    result_value = task_result_from_json(loads_json(raw_result))
+
+            if include_failed_reason:
+                failed_reason = row[idx]
+
+            return TaskInfo(
+                task_id=task_id_value,
+                task_name=task_name,
+                status=status,
+                queue_name=queue_name,
+                priority=priority,
+                retry_count=retry_count,
+                max_retries=max_retries,
+                next_retry_at=next_retry_at,
+                sent_at=sent_at,
+                claimed_at=claimed_at,
+                started_at=started_at,
+                completed_at=completed_at,
+                failed_at=failed_at,
+                worker_hostname=worker_hostname,
+                worker_pid=worker_pid,
+                worker_process_name=worker_process_name,
+                result=result_value,
+                failed_reason=failed_reason,
+            )
+
+    def get_task_info(
+        self,
+        task_id: str,
+        *,
+        include_result: bool = False,
+        include_failed_reason: bool = False,
+    ) -> 'TaskInfo | None':
+        """Synchronous wrapper for get_task_info_async()."""
+        return self._loop_runner.call(
+            self.get_task_info_async,
+            task_id,
+            include_result=include_result,
+            include_failed_reason=include_failed_reason,
+        )
+
     def close(self) -> None:
         """
         Synchronous cleanup (runs close_async in background loop).

{horsies-0.1.0a2 → horsies-0.1.0a4}/horsies/core/models/tasks.py

@@ -1,6 +1,7 @@
 # app/core/models/tasks.py
 from __future__ import annotations
 import datetime
+from dataclasses import dataclass
 from typing import (
     TYPE_CHECKING,
     Any,
@@ -20,6 +21,7 @@ from enum import Enum
 if TYPE_CHECKING:
     from horsies.core.models.workflow import SubWorkflowSummary
 
+from horsies.core.types.status import TaskStatus
 T = TypeVar('T')  # success payload
 E = TypeVar('E')  # error payload (TaskError )
 
@@ -219,6 +221,30 @@ class TaskResult(Generic[T, E]):
                 raise ValueError('Result is not error - check is_err() first')
 
 
+@dataclass
+class TaskInfo:
+    """Metadata for a broker-backed task."""
+
+    task_id: str
+    task_name: str
+    status: TaskStatus
+    queue_name: str
+    priority: int
+    retry_count: int
+    max_retries: int
+    next_retry_at: datetime.datetime | None
+    sent_at: datetime.datetime | None
+    claimed_at: datetime.datetime | None
+    started_at: datetime.datetime | None
+    completed_at: datetime.datetime | None
+    failed_at: datetime.datetime | None
+    worker_hostname: str | None
+    worker_pid: int | None
+    worker_process_name: str | None
+    result: TaskResult[Any, TaskError] | None = None
+    failed_reason: str | None = None
+
+
 class RetryPolicy(BaseModel):
     """
     Retry policy configuration for tasks.

{horsies-0.1.0a2 → horsies-0.1.0a4}/horsies/core/models/workflow.py

@@ -1374,6 +1374,7 @@ class WorkflowContext(BaseModel):
 class WorkflowTaskInfo:
     """Information about a task within a workflow."""
 
+    node_id: str | None
     index: int
     name: str
     status: WorkflowTaskStatus
@@ -1700,7 +1701,7 @@ class WorkflowHandle:
         async with self.broker.session_factory() as session:
             result = await session.execute(
                 text("""
-                    SELECT task_index, task_name, status, result, started_at, completed_at
+                    SELECT node_id, task_index, task_name, status, result, started_at, completed_at
                     FROM horsies_workflow_tasks
                     WHERE workflow_id = :wf_id
                     ORDER BY task_index
@@ -1710,14 +1711,15 @@ class WorkflowHandle:
 
             return [
                 WorkflowTaskInfo(
-                    index=row[0],
-                    name=row[1],
-                    status=WorkflowTaskStatus(row[2]),
-                    result=task_result_from_json(loads_json(row[3]))
-                    if row[3]
+                    node_id=row[0],
+                    index=row[1],
+                    name=row[2],
+                    status=WorkflowTaskStatus(row[3]),
+                    result=task_result_from_json(loads_json(row[4]))
+                    if row[4]
                     else None,
-                    started_at=row[4],
-                    completed_at=row[5],
+                    started_at=row[5],
+                    completed_at=row[6],
                 )
                 for row in result.fetchall()
             ]

{horsies-0.1.0a2 → horsies-0.1.0a4}/horsies/core/task_decorator.py

@@ -23,6 +23,7 @@ if TYPE_CHECKING:
     from horsies.core.app import Horsies
     from horsies.core.models.tasks import TaskOptions
     from horsies.core.models.tasks import TaskError, TaskResult
+    from horsies.core.models.tasks import TaskInfo
 
 from horsies.core.models.tasks import TaskResult, TaskError, LibraryErrorCode
 from horsies.core.models.workflow import WorkflowContextMissingIdError
@@ -191,6 +192,46 @@ class TaskHandle(Generic[T]):
                 case result:
                     return result
 
+    def info(
+        self,
+        *,
+        include_result: bool = False,
+        include_failed_reason: bool = False,
+    ) -> 'TaskInfo | None':
+        """Fetch metadata for this task from the broker."""
+        if not self._broker_mode or not self._app:
+            raise RuntimeError(
+                'TaskHandle.info() requires a broker-backed task handle '
+                '(use .send() or .send_async())'
+            )
+
+        broker = self._app.get_broker()
+        return broker.get_task_info(
+            self.task_id,
+            include_result=include_result,
+            include_failed_reason=include_failed_reason,
+        )
+
+    async def info_async(
+        self,
+        *,
+        include_result: bool = False,
+        include_failed_reason: bool = False,
+    ) -> 'TaskInfo | None':
+        """Async version of info()."""
+        if not self._broker_mode or not self._app:
+            raise RuntimeError(
+                'TaskHandle.info_async() requires a broker-backed task handle '
+                '(use .send() or .send_async())'
+            )
+
+        broker = self._app.get_broker()
+        return await broker.get_task_info_async(
+            self.task_id,
+            include_result=include_result,
+            include_failed_reason=include_failed_reason,
+        )
+
     def set_immediate_result(
         self,
         result: TaskResult[T, TaskError],

{horsies-0.1.0a2 → horsies-0.1.0a4}/horsies/core/types/status.py

@@ -10,20 +10,20 @@ from enum import Enum
 class TaskStatus(Enum):
     """Task execution status"""
 
-    PENDING = 'pending'  # It awaits to be a candidate for execution.
+    PENDING = 'PENDING'  # It awaits to be a candidate for execution.
     # Default status when the task is sent.
 
     CLAIMED = (
-        'claimed'  # It has been claimed by a worker but not yet started executing.
+        'CLAIMED'  # It has been claimed by a worker but not yet started executing.
     )
 
-    RUNNING = 'running'  # It is being executed by a process.
+    RUNNING = 'RUNNING'  # It is being executed by a process.
 
-    COMPLETED = 'completed'  # It has been executed successfully.
+    COMPLETED = 'COMPLETED'  # It has been executed successfully.
 
-    FAILED = 'failed'  # It has failed to be executed.
-    CANCELLED = 'cancelled'  # It has been cancelled.
-    REQUEUED = 'requeued'  # It has been requeued after a failure.
+    FAILED = 'FAILED'  # It has failed to be executed.
+    CANCELLED = 'CANCELLED'  # It has been cancelled.
+    REQUEUED = 'REQUEUED'  # It has been requeued after a failure.
 
     @property
     def is_terminal(self) -> bool:

{horsies-0.1.0a2 → horsies-0.1.0a4}/horsies.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: horsies
-Version: 0.1.0a2
+Version: 0.1.0a4
 Summary: A Python library for distributed task execution
 Author: Suleyman Ozkeskin
 License-Expression: MIT

{horsies-0.1.0a2 → horsies-0.1.0a4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "horsies"
-version = "0.1.0a2"
+version = "0.1.0a4"
 description = "A Python library for distributed task execution"
 readme = "README.md"
 license = "MIT"