mm-std 0.3.18__py3-none-any.whl → 0.3.20__py3-none-any.whl

mm_std/concurrency/async_task_runner.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import asyncio
 from collections.abc import Awaitable
 from dataclasses import dataclass
+from logging import Logger
 from typing import Any
 
 
@@ -26,17 +27,21 @@ class AsyncTaskRunner:
         task_id: str
         awaitable: Awaitable[Any]
 
-    def __init__(self, max_concurrent_tasks: int, timeout: float | None = None, name: str | None = None) -> None:
+    def __init__(
+        self, max_concurrent_tasks: int, timeout: float | None = None, name: str | None = None, logger: Logger | None = None
+    ) -> None:
         """
         :param max_concurrent_tasks: Maximum number of tasks that can run concurrently.
         :param timeout: Optional overall timeout in seconds for running all tasks.
         :param name: Optional name for the runner.
+        :param logger: Optional logger for task exceptions.
         """
         if timeout is not None and timeout <= 0:
             raise ValueError("Timeout must be positive if specified.")
         self.max_concurrent_tasks: int = max_concurrent_tasks
         self.timeout: float | None = timeout
         self.name = name
+        self.logger = logger
         self.semaphore: asyncio.Semaphore = asyncio.Semaphore(max_concurrent_tasks)
         self._tasks: list[AsyncTaskRunner.Task] = []
         self._was_run: bool = False
@@ -67,6 +72,9 @@ class AsyncTaskRunner:
         self._task_ids.add(task_id)
         self._tasks.append(AsyncTaskRunner.Task(task_id, awaitable))
 
+    def _task_name(self, task_id: str) -> str:
+        return f"{self.name}-{task_id}" if self.name else task_id
+
     async def run(self) -> AsyncTaskRunner.Result:
         """
         Executes all added tasks with concurrency limited by the semaphore.
@@ -89,10 +97,12 @@ class AsyncTaskRunner:
                     res: Any = await task.awaitable
                     results[task.task_id] = res
                 except Exception as e:
+                    if self.logger:
+                        self.logger.exception(f"Task '{self._task_name(task.task_id)}' raised an exception")
                     exceptions[task.task_id] = e
 
         # Create asyncio tasks for all runner tasks
-        tasks = [asyncio.create_task(run_task(task)) for task in self._tasks]
+        tasks = [asyncio.create_task(run_task(task), name=self._task_name(task.task_id)) for task in self._tasks]
 
         try:
             if self.timeout is not None:

{mm_std-0.3.18.dist-info → mm_std-0.3.20.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: mm-std
-Version: 0.3.18
+Version: 0.3.20
 Requires-Python: >=3.12
 Requires-Dist: aiohttp-socks~=0.10.1
 Requires-Dist: aiohttp~=3.11.14
 Requires-Dist: cryptography~=44.0.2
 Requires-Dist: pydantic-settings>=2.8.1
-Requires-Dist: pydantic~=2.10.6
+Requires-Dist: pydantic~=2.11.0
 Requires-Dist: pydash~=8.0.5
 Requires-Dist: python-dotenv~=1.1.0
 Requires-Dist: requests[socks]~=2.32.3

{mm_std-0.3.18.dist-info → mm_std-0.3.20.dist-info}/RECORD

@@ -21,10 +21,10 @@ mm_std/zip.py,sha256=axzF1BwcIygtfNNTefZH7hXKaQqwe-ZH3ChuRWr9dnk,396
 mm_std/concurrency/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mm_std/concurrency/async_decorators.py,sha256=xEpyipzp3ZhaPHtdeTE-Ikrt67SUTFKBE6LQPeoeh6Q,1735
 mm_std/concurrency/async_scheduler.py,sha256=qS3QKMA0xpoxCZWjDW1ItAwKMTQ5h8esXMMRA0eXtxE,5644
-mm_std/concurrency/async_task_runner.py,sha256=zYC2Jv5taUh8dnyDfWwh394SkzTXtdE9hOhvjV2FWKc,4493
+mm_std/concurrency/async_task_runner.py,sha256=KyTp4aHt-qBc-qJpXIutFaTP9ykupG7uKafghAwbBvg,4948
 mm_std/concurrency/sync_decorators.py,sha256=syCQBOmN7qPO55yzgJB2rbkh10CVww376hmyvs6e5tA,1080
 mm_std/concurrency/sync_scheduler.py,sha256=j4tBL_cBI1spr0cZplTA7N2CoYsznuORMeRN8rpR6gY,2407
 mm_std/concurrency/sync_task_runner.py,sha256=s5JPlLYLGQGHIxy4oDS-PN7O9gcy-yPZFoNm8RQwzcw,1780
-mm_std-0.3.18.dist-info/METADATA,sha256=AUl95Lq0McMhB-E8PsxqPEbYsxuiQI0cA1t5AX208qg,415
-mm_std-0.3.18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mm_std-0.3.18.dist-info/RECORD,,
+mm_std-0.3.20.dist-info/METADATA,sha256=bcvPFhVCzwNBhB8qz0DW1AZY_jeKqdFIuMiOHHz-22I,415
+mm_std-0.3.20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mm_std-0.3.20.dist-info/RECORD,,