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

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

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

{dycw_utilities-0.134.2 → 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.2"
+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.2"
+current_version = "0.134.4"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.134.2 → 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.2 → dycw_utilities-0.134.4}/src/tests/test_parse.py

@@ -45,6 +45,7 @@ from utilities.functions import ensure_path
 from utilities.hypothesis import (
     dates,
     int64s,
+    months,
     numbers,
     paths,
     plain_datetimes,
@@ -67,6 +68,7 @@ from utilities.sentinel import Sentinel, sentinel
 from utilities.text import parse_bool
 from utilities.types import Number
 from utilities.version import Version
+from utilities.whenever import Month
 
 
 class TestSerializeAndParseObject:
@@ -141,6 +143,12 @@ class TestSerializeAndParseObject:
         result = parse_object(TrueOrFalseFutureLit, truth)
         assert result == truth
 
+    @given(month=months())
+    def test_month(self, *, month: Month) -> None:
+        serialized = serialize_object(month)
+        result = parse_object(Month, serialized)
+        assert result == month
+
     def test_none(self) -> None:
         serialized = serialize_object(None)
         result = parse_object(None, serialized)

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

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

{dycw_utilities-0.134.2 → 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:

{dycw_utilities-0.134.2 → dycw_utilities-0.134.4}/src/utilities/parse.py

@@ -51,6 +51,7 @@ from utilities.typing import (
     is_union_type,
 )
 from utilities.version import ParseVersionError, Version, parse_version
+from utilities.whenever import Month
 
 if TYPE_CHECKING:
     from collections.abc import Iterable, Mapping, Sequence
@@ -189,7 +190,16 @@ def _parse_object_type(
             raise _ParseObjectParseError(type_=cls, text=text) from None
     if issubclass(
         cls,
-        (Date, DateDelta, DateTimeDelta, PlainDateTime, Time, TimeDelta, ZonedDateTime),
+        (
+            Date,
+            DateDelta,
+            DateTimeDelta,
+            Month,
+            PlainDateTime,
+            Time,
+            TimeDelta,
+            ZonedDateTime,
+        ),
     ):
         try:
             return cls.parse_common_iso(text)
@@ -453,7 +463,16 @@ def serialize_object(
         return str(obj)
     if isinstance(
         obj,
-        (Date, DateDelta, DateTimeDelta, PlainDateTime, Time, TimeDelta, ZonedDateTime),
+        (
+            Date,
+            DateDelta,
+            DateTimeDelta,
+            Month,
+            PlainDateTime,
+            Time,
+            TimeDelta,
+            ZonedDateTime,
+        ),
     ):
         return obj.format_common_iso()
     if isinstance(obj, Enum):