horsies 0.1.0a2__py3-none-any.whl → 0.1.0a3__py3-none-any.whl

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/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/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/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/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.dist-info → horsies-0.1.0a3.dist-info}/METADATA

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

{horsies-0.1.0a2.dist-info → horsies-0.1.0a3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-horsies/__init__.py,sha256=Fr_4G82B68cs3p7wShUum2fQlmey1hBf4pAxDz7M6D0,2739
+horsies/__init__.py,sha256=rLxes5IuyzggohtXKRsHoddeyqEy0LsgwDlk09itKXk,2769
 horsies/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 horsies/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 horsies/core/app.py,sha256=sRLVto5oFH0Mb3cwrzJXd2WoPEjoYL2ByJc764DW238,21523
@@ -6,10 +6,10 @@ horsies/core/banner.py,sha256=eCp8crZZWS6JjWoTn6RtCTEpvGmkpr7hWke_Av5-bvI,5716
 horsies/core/cli.py,sha256=XPMRRvZqNg3ph96gWCfo0NJDOsqqe0Cuj_8Wx80sQP0,21057
 horsies/core/errors.py,sha256=M0PMSdsqbOMPOsZsEOmfKTvxdAD5E7BZ0IuDus7HWtk,16640
 horsies/core/logging.py,sha256=p2utHDeOHgvwtSzKTKXh0RwUNCk38ODmgQQkmsD1tWA,2934
-horsies/core/task_decorator.py,sha256=kn1xD76dqF4mjXAut5GWUHwHZ4JIBsavNvW1-8Fr-OA,22449
+horsies/core/task_decorator.py,sha256=mqQYV8iLt9TwsrNQDDsBx3Vo8T5HeLFv6iiWbhF3l-c,23799
 horsies/core/brokers/__init__.py,sha256=a_5xkHhRKY-1PRq6UidMBGq1bX-zeuSdxIvI6GdSiSQ,94
 horsies/core/brokers/listener.py,sha256=pxnnAgLAWqODPV_J0XwUqAhBSrHstL77PSHYEfGoVhc,18637
-horsies/core/brokers/postgres.py,sha256=n_WjM7-Fbg9R5Vao9RJgDSy7eBIL1_pVvnKrjMayzd4,35134
+horsies/core/brokers/postgres.py,sha256=YZSJiQY1Spc2pNF8fjPy_0K9EAD9WAxVnFaV2P6Qf5Q,39073
 horsies/core/codec/serde.py,sha256=-dUEwyE4-DcAnP2CTbJYxBW3rOzXIUMsX_OSZSjNnpU,20043
 horsies/core/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 horsies/core/models/app.py,sha256=I3HKztjNzii75bDwlbB7Jl9CPrFMfjLInUJRacQ39_s,11863
@@ -18,8 +18,8 @@ horsies/core/models/queues.py,sha256=brB8rk_PQZds_VgavpZH3fkgqzsBAvRTySNNAtzikSw
 horsies/core/models/recovery.py,sha256=sI9h7Kbx_Jv23-J_0fTZRrrMgwNqmkX0-dWKTnpikvA,4964
 horsies/core/models/schedule.py,sha256=5mCMsX-12OPz6Gs-nr_GtaJgjxDzuyjuhSTil2XIq50,8113
 horsies/core/models/task_pg.py,sha256=v5U1huJwASh2Saf8GDaRX-IM4mXcD8Fu43kfrFSFDLg,12304
-horsies/core/models/tasks.py,sha256=NBGBw80XniOG3_r1nOB4SE_-N83h8WHQ7FH0tPlr41M,10863
-horsies/core/models/workflow.py,sha256=KZE7l3cjePpTyKqjTOq5erKyptcAia08ai4v59LVtNo,71688
+horsies/core/models/tasks.py,sha256=nQb5z8S-GioO2aZATEvfubzqKtsNBGu-RcCn9kdituw,11592
+horsies/core/models/workflow.py,sha256=_93sQ-He9gbHJ4zOpmjlQ0pcRUPWGEXqzwKUNmN9gRc,71757
 horsies/core/models/workflow_pg.py,sha256=Nd6JDy4Htft9qQ4xKW4U2vhKNe-uUMYYZfg14t-yJnk,9315
 horsies/core/registry/tasks.py,sha256=mm1xb-f2HLUcZJrLgx2ZS-FQtCkuJbfsR-SW2qiMsts,3708
 horsies/core/scheduler/__init__.py,sha256=m0GqCrdTbQNDUV1Fn3UZD5IewAYsV5olMrDRolg3a1I,699
@@ -35,8 +35,8 @@ horsies/core/workflows/__init__.py,sha256=JA-8wcYUHp-wF5OCEqJwKElT8PaZZB1G7iglDZ
 horsies/core/workflows/engine.py,sha256=8Gjmch52g1IObKqzGVMWxZaZDC0diB6TuAtRfqr67l0,86927
 horsies/core/workflows/recovery.py,sha256=xRKQPiYaLcX4vLuvcNGy5cV14eW4Cp_sewD1nXv8gc8,19445
 horsies/core/workflows/registry.py,sha256=ItpjTN8yK9NNSV8Q8OwDnHdQSO-hxDzxeWAyGvExpRk,3194
-horsies-0.1.0a2.dist-info/METADATA,sha256=XJ8Gw2iRCTlKSRkhU3mri_3Q4yuTxRA77lTMmqBJpYs,1321
-horsies-0.1.0a2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-horsies-0.1.0a2.dist-info/entry_points.txt,sha256=6b_OhuNbJ1ky9WSOt3UfNQXETG2YuH0lKimcibILrEE,50
-horsies-0.1.0a2.dist-info/top_level.txt,sha256=ZQ_NurgT-yr3pE4a1svlO_VAhGJ6NFA31vJDLT1yyOA,8
-horsies-0.1.0a2.dist-info/RECORD,,
+horsies-0.1.0a3.dist-info/METADATA,sha256=O-mV7PXCaiEBJfxNVh3sbYZ3qnfWzUU3W6aPGgb_55c,1321
+horsies-0.1.0a3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+horsies-0.1.0a3.dist-info/entry_points.txt,sha256=6b_OhuNbJ1ky9WSOt3UfNQXETG2YuH0lKimcibILrEE,50
+horsies-0.1.0a3.dist-info/top_level.txt,sha256=ZQ_NurgT-yr3pE4a1svlO_VAhGJ6NFA31vJDLT1yyOA,8
+horsies-0.1.0a3.dist-info/RECORD,,