dycw-utilities 0.134.3__tar.gz → 0.134.4__tar.gz

{dycw_utilities-0.134.3 → dycw_utilities-0.134.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.134.3
+Version: 0.134.4
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12

{dycw_utilities-0.134.3 → dycw_utilities-0.134.4}/pyproject.toml

@@ -105,7 +105,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.134.3"
+version = "0.134.4"
 
 [project.optional-dependencies]
 logging = [
@@ -132,7 +132,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.134.3"
+current_version = "0.134.4"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.134.3 → dycw_utilities-0.134.4}/src/tests/test_asyncio.py

@@ -175,6 +175,29 @@ class TestEnhancedTaskGroup:
             await sleep_td(self.delta)
         assert looper._core_attempts >= 1
 
+    async def test_debug_true(self) -> None:
+        class CustomError(Exception): ...
+
+        async def func() -> None:
+            await sleep_td(self.delta)
+            raise CustomError
+
+        with raises(CustomError):
+            async with EnhancedTaskGroup(debug=True) as tg:
+                _ = await tg.run_or_create_task(func())
+
+    async def test_debug_false(self) -> None:
+        class CustomError(Exception): ...
+
+        async def func() -> None:
+            await sleep_td(self.delta)
+            raise CustomError
+
+        with raises(ExceptionGroup) as exc_info:
+            async with EnhancedTaskGroup(debug=False) as tg:
+                _ = await tg.run_or_create_task(func())
+        assert isinstance(one(exc_info.value.exceptions), CustomError)
+
     async def test_max_tasks_disabled(self) -> None:
         with Timer() as timer:
             async with EnhancedTaskGroup() as tg:

{dycw_utilities-0.134.3 → dycw_utilities-0.134.4}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.134.3"
+__version__ = "0.134.4"

{dycw_utilities-0.134.3 → dycw_utilities-0.134.4}/src/utilities/asyncio.py

@@ -223,6 +223,7 @@ class EnhancedTaskGroup(TaskGroup):
     _semaphore: Semaphore | None
     _timeout: TimeDelta | None
     _error: MaybeType[BaseException]
+    _debug: bool
     _stack: AsyncExitStack
     _timeout_cm: _AsyncGeneratorContextManager[None] | None
 
@@ -233,11 +234,13 @@ class EnhancedTaskGroup(TaskGroup):
         max_tasks: int | None = None,
         timeout: TimeDelta | None = None,
         error: MaybeType[BaseException] = TimeoutError,
+        debug: bool = False,
     ) -> None:
         super().__init__()
         self._semaphore = None if max_tasks is None else Semaphore(max_tasks)
         self._timeout = timeout
         self._error = error
+        self._debug = debug
         self._stack = AsyncExitStack()
         self._timeout_cm = None
 
@@ -254,7 +257,14 @@ class EnhancedTaskGroup(TaskGroup):
         tb: TracebackType | None,
     ) -> None:
         _ = await self._stack.__aexit__(et, exc, tb)
-        _ = await super().__aexit__(et, exc, tb)
+        match self._debug:
+            case True:
+                with suppress(Exception):
+                    _ = await super().__aexit__(et, exc, tb)
+            case False:
+                _ = await super().__aexit__(et, exc, tb)
+            case _ as never:
+                assert_never(never)
 
     @override
     def create_task[T](
@@ -276,6 +286,21 @@ class EnhancedTaskGroup(TaskGroup):
         _ = self._stack.push_async_callback(cm.__aexit__, None, None, None)
         return self.create_task(cm.__aenter__())
 
+    async def run_or_create_task[T](
+        self,
+        coro: _CoroutineLike[T],
+        *,
+        name: str | None = None,
+        context: Context | None = None,
+    ) -> T | Task[T]:
+        match self._debug:
+            case True:
+                return await coro
+            case False:
+                return self.create_task(coro, name=name, context=context)
+            case _ as never:
+                assert_never(never)
+
     async def _wrap_with_semaphore[T](
         self, semaphore: Semaphore, coroutine: _CoroutineLike[T], /
     ) -> T: