mypy

Author: Andrei Neagu

packages/service-library/src/servicelib/aiohttp/long_running_tasks/_manager.py

@@ -4,7 +4,8 @@
 from settings_library.redis import RedisSettings
 
 from ...long_running_tasks.base_long_running_manager import BaseLongRunningManager
-from ...long_running_tasks.task import Namespace, TaskContext, TasksManager
+from ...long_running_tasks.models import TaskContext
+from ...long_running_tasks.task import Namespace, TasksManager
 from ._constants import APP_LONG_RUNNING_MANAGER_KEY
 from ._request import get_task_context
 

packages/service-library/src/servicelib/aiohttp/long_running_tasks/_server.py

@@ -16,8 +16,8 @@
     DEFAULT_STALE_TASK_CHECK_INTERVAL,
     DEFAULT_STALE_TASK_DETECT_TIMEOUT,
 )
-from ...long_running_tasks.models import TaskGet
-from ...long_running_tasks.task import RegisteredTaskName, TaskContext
+from ...long_running_tasks.models import TaskContext, TaskGet
+from ...long_running_tasks.task import RegisteredTaskName
 from ..typing_extension import Handler
 from . import _routes
 from ._constants import (

packages/service-library/src/servicelib/long_running_tasks/_store/redis.py

@@ -44,24 +44,24 @@ def _get_key(self, store_type: str, name: str) -> str:
     async def get_task_data(self, task_id: TaskId) -> TaskData | None:
         result: Any | None = await self.redis.hget(
             self._get_redis_hash_key(STORE_TYPE_TASK_DATA), task_id
-        )
+        )  # type: ignore[misc]
         return TypeAdapter(TaskData).validate_json(result) if result else None
 
     async def set_task_data(self, task_id: TaskId, value: TaskData) -> None:
         await self.redis.hset(
             self._get_redis_hash_key(STORE_TYPE_TASK_DATA),
             task_id,
             value.model_dump_json(),
-        )
+        )  # type: ignore[misc]
 
     async def list_tasks_data(self) -> list[TaskData]:
         result: list[Any] = await self.redis.hvals(
             self._get_redis_hash_key(STORE_TYPE_TASK_DATA)
-        )
+        )  # type: ignore[misc]
         return [TypeAdapter(TaskData).validate_json(item) for item in result]
 
     async def delete_task_data(self, task_id: TaskId) -> None:
-        await self.redis.hdel(self._get_redis_hash_key(STORE_TYPE_TASK_DATA), task_id)
+        await self.redis.hdel(self._get_redis_hash_key(STORE_TYPE_TASK_DATA), task_id)  # type: ignore[misc]
 
     async def set_as_cancelled(
         self, task_id: TaskId, with_task_context: TaskContext | None
@@ -70,12 +70,12 @@ async def set_as_cancelled(
             self._get_redis_hash_key(STORE_TYPE_CANCELLED_TASKS),
             task_id,
             json_dumps(with_task_context),
-        )
+        )  # type: ignore[misc]
 
     async def get_cancelled(self) -> dict[TaskId, TaskContext | None]:
         result: dict[str, str | None] = await self.redis.hgetall(
             self._get_redis_hash_key(STORE_TYPE_CANCELLED_TASKS)
-        )
+        )  # type: ignore[misc]
         return {
             task_id: (json_loads(context) if context else None)
             for task_id, context in result.items()

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

@@ -5,8 +5,8 @@
 from servicelib.logging_errors import create_troubleshootting_log_kwargs
 
 from .errors import TaskNotCompletedError, TaskNotFoundError
-from .models import TaskBase, TaskId, TaskStatus
-from .task import RegisteredTaskName, TaskContext, TasksManager
+from .models import TaskBase, TaskContext, TaskId, TaskStatus
+from .task import RegisteredTaskName, TasksManager
 
 _logger = logging.getLogger(__name__)
 

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

@@ -111,7 +111,9 @@ def __init__(
         namespace: Namespace,
     ):
         # Task groups: Every taskname maps to multiple asyncio.Task within TrackedTask model
-        self._tasks_data: BaseStore = _StorageClass(redis_settings, namespace)
+        self._tasks_data: BaseStore = _StorageClass(
+            redis_settings, namespace
+        )  # type:ignore[call-arg]
         self._created_tasks: dict[TaskId, asyncio.Task] = {}
 
         self.stale_task_check_interval = stale_task_check_interval