hid internals

Author: Andrei Neagu

packages/service-library/src/servicelib/long_running_tasks/lrt_api.py

@@ -1,7 +1,7 @@
 from typing import Any
 
 from .models import TaskBase, TaskId, TaskStatus
-from .task import RegisteredTaskName, TaskContext, TasksManager, TrackedTask
+from .task import RegisteredTaskName, TaskContext, TasksManager
 
 
 async def start_task(
@@ -52,10 +52,7 @@ async def start_task(
 def list_tasks(
     tasks_manager: TasksManager, task_context: TaskContext | None
 ) -> list[TaskBase]:
-    tracked_tasks: list[TrackedTask] = tasks_manager.list_tasks(
-        with_task_context=task_context
-    )
-    return [TaskBase(task_id=t.task_id) for t in tracked_tasks]
+    return tasks_manager.list_tasks(with_task_context=task_context)
 
 
 def get_task_status(

packages/service-library/src/servicelib/long_running_tasks/task.py

@@ -23,7 +23,7 @@
     TaskNotFoundError,
     TaskNotRegisteredError,
 )
-from .models import TaskId, TaskStatus, TrackedTask
+from .models import TaskBase, TaskId, TaskStatus, TrackedTask
 
 _logger = logging.getLogger(__name__)
 
@@ -175,12 +175,14 @@ async def _stale_tasks_monitor_worker(self) -> None:
                 task_id, with_task_context=None, reraise_errors=False
             )
 
-    def list_tasks(self, with_task_context: TaskContext | None) -> list[TrackedTask]:
+    def list_tasks(self, with_task_context: TaskContext | None) -> list[TaskBase]:
         if not with_task_context:
-            return list(self._tracked_tasks.values())
+            return [
+                TaskBase(task_id=task.task_id) for task in self._tracked_tasks.values()
+            ]
 
         return [
-            task
+            TaskBase(task_id=task.task_id)
             for task in self._tracked_tasks.values()
             if task.task_context == with_task_context
         ]