dycw-utilities 0.129.14__py3-none-any.whl → 0.130.0__py3-none-any.whl

utilities/redis.py

@@ -53,7 +53,7 @@ if TYPE_CHECKING:
     from redis.typing import EncodableT, ResponseT
 
     from utilities.iterables import MaybeIterable
-    from utilities.types import Duration, TypeLike
+    from utilities.types import Duration, MaybeType, TypeLike
 
 
 _K = TypeVar("_K")
@@ -90,7 +90,7 @@ class RedisHashMapKey(Generic[_K, _V]):
     value_serializer: Callable[[_V], bytes] | None = None
     value_deserializer: Callable[[bytes], _V] | None = None
     timeout: Duration | None = None
-    error: type[Exception] = TimeoutError
+    error: MaybeType[BaseException] = TimeoutError
     ttl: Duration | None = None
 
     async def delete(self, redis: Redis, key: _K, /) -> int:
@@ -402,7 +402,7 @@ class RedisKey(Generic[_T]):
     serializer: Callable[[_T], bytes] | None = None
     deserializer: Callable[[bytes], _T] | None = None
     timeout: Duration | None = None
-    error: type[Exception] = TimeoutError
+    error: MaybeType[BaseException] = TimeoutError
     ttl: Duration | None = None
 
     async def delete(self, redis: Redis, /) -> int:

utilities/sqlalchemy.py

@@ -82,7 +82,13 @@ from utilities.iterables import (
 )
 from utilities.reprlib import get_repr
 from utilities.text import snake_case
-from utilities.types import Duration, MaybeIterable, StrMapping, TupleOrStrMapping
+from utilities.types import (
+    Duration,
+    MaybeIterable,
+    MaybeType,
+    StrMapping,
+    TupleOrStrMapping,
+)
 
 _T = TypeVar("_T")
 type _EngineOrConnectionOrAsync = Engine | Connection | AsyncEngine | AsyncConnection
@@ -830,7 +836,7 @@ async def yield_connection(
     /,
     *,
     timeout: Duration | None = None,
-    error: type[Exception] = TimeoutError,
+    error: MaybeType[BaseException] = TimeoutError,
 ) -> AsyncIterator[AsyncConnection]:
     """Yield an async connection."""
     async with timeout_dur(duration=timeout, error=error), engine.begin() as conn:

utilities/sqlalchemy_polars.py

@@ -68,7 +68,7 @@ if TYPE_CHECKING:
     from tenacity.wait import WaitBaseT
 
     import utilities.types
-    from utilities.types import TimeZoneLike
+    from utilities.types import MaybeType, TimeZoneLike
 
 
 async def insert_dataframe(
@@ -303,7 +303,7 @@ async def select_to_dataframe(
     in_clauses_chunk_size: int | None = None,
     chunk_size_frac: float = CHUNK_SIZE_FRAC,
     timeout: utilities.types.Duration | None = None,
-    error: type[Exception] = TimeoutError,
+    error: MaybeType[BaseException] = TimeoutError,
     **kwargs: Any,
 ) -> DataFrame | Iterable[DataFrame] | AsyncIterable[DataFrame]:
     """Read a table from a database into a DataFrame."""