dycw-utilities 0.136.6__tar.gz → 0.136.8__tar.gz

{dycw_utilities-0.136.6 → dycw_utilities-0.136.8}/PKG-INFO

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

{dycw_utilities-0.136.6 → dycw_utilities-0.136.8}/pyproject.toml

@@ -51,7 +51,7 @@ luigi = ["luigi >= 3.6.0, < 3.7"]
 math-test = ["numpy"]
 memory-profiler = ["memory-profiler >= 0.61.0, < 0.62"]
 more-itertools = ["more-itertools >= 10.7.0, < 10.8"]
-numpy = ["numpy >= 2.3.0, < 2.4"]
+numpy = ["numpy >= 2.3.1, < 2.4"]
 operator = ["polars-lts-cpu"]
 optuna = ["optuna >= 4.4.0, < 4.5"]
 orjson = ["orjson >= 3.10.18, < 3.11"]
@@ -80,7 +80,7 @@ python-dotenv = ["python-dotenv >= 1.1.0, < 1.2"]
 redis = ["redis >= 6.2.0, < 6.3", "orjson"]
 redis-test = ["pytest-rerunfailures"]
 reprlib-test = ["rich"]
-scipy = ["scipy >= 1.15.3, < 1.16"]
+scipy = ["scipy >= 1.16.0, < 1.17"]
 sklearn = ["scikit-learn >= 1.7.0, < 1.8"]
 slack-sdk = ["slack-sdk >= 3.35.0, < 3.36"]
 slack-sdk-test = ["aiohttp"]
@@ -105,7 +105,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.136.6"
+version = "0.136.8"
 
 [project.optional-dependencies]
 logging = [
@@ -133,7 +133,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.136.6"
+current_version = "0.136.8"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.136.6 → dycw_utilities-0.136.8}/src/tests/test_pottery.py

@@ -66,7 +66,7 @@ class TestYieldAccess:
     ) -> None:
         with Timer() as timer:
             await _func_access(num_tasks, unique_str(), num_locks=num_locks)
-        assert (min_multiple * _DELTA) <= timer <= (3 * min_multiple * _DELTA)
+        assert (min_multiple * _DELTA) <= timer <= (5 * min_multiple * _DELTA)
 
     async def test_error_num_locks(self) -> None:
         key = unique_str()

{dycw_utilities-0.136.6 → dycw_utilities-0.136.8}/src/tests/test_sqlalchemy.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
+from enum import Enum, StrEnum, auto
 from itertools import chain
 from typing import TYPE_CHECKING, Any, Literal, cast, overload, override
 from uuid import uuid4
@@ -69,6 +70,8 @@ from utilities.sqlalchemy import (
     create_async_engine,
     ensure_tables_created,
     ensure_tables_dropped,
+    enum_name,
+    enum_values,
     get_chunk_size,
     get_column_names,
     get_columns,
@@ -278,6 +281,25 @@ class TestEnsureTablesDropped:
                 _ = await conn.execute(sel)
 
 
+class TestEnumName:
+    def test_main(self) -> None:
+        class Example(Enum): ...
+
+        result = enum_name(Example)
+        assert result == "example_enum"
+
+
+class TestEnumValues:
+    def test_main(self) -> None:
+        class Example(StrEnum):
+            true = auto()
+            false = auto()
+
+        result = enum_values(Example)
+        expected = ["true", "false"]
+        assert result == expected
+
+
 class TestGetChunkSize:
     @given(chunk_size_frac=floats(0.0, 1.0), scaling=int32s(min_value=0))
     @settings(

{dycw_utilities-0.136.6 → dycw_utilities-0.136.8}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.136.6"
+__version__ = "0.136.8"

{dycw_utilities-0.136.6 → dycw_utilities-0.136.8}/src/utilities/sqlalchemy.py

@@ -86,6 +86,8 @@ from utilities.types import MaybeIterable, MaybeType, StrMapping, TupleOrStrMapp
 from utilities.whenever import SECOND
 
 if TYPE_CHECKING:
+    from enum import Enum, StrEnum
+
     from whenever import TimeDelta
 
 type _EngineOrConnectionOrAsync = Engine | Connection | AsyncEngine | AsyncConnection
@@ -284,6 +286,22 @@ async def ensure_tables_dropped(
 ##
 
 
+def enum_name(enum: type[Enum], /) -> str:
+    """Get the name of an Enum."""
+    return f"{snake_case(get_class_name(enum))}_enum"
+
+
+##
+
+
+def enum_values(enum: type[StrEnum], /) -> list[str]:
+    """Get the values of a StrEnum."""
+    return [e.value for e in enum]
+
+
+##
+
+
 def get_chunk_size(
     engine_or_conn: _EngineOrConnectionOrAsync,
     /,
@@ -1171,6 +1189,8 @@ __all__ = [
     "create_async_engine",
     "ensure_tables_created",
     "ensure_tables_dropped",
+    "enum_name",
+    "enum_values",
     "get_chunk_size",
     "get_column_names",
     "get_columns",

{dycw_utilities-0.136.6 → dycw_utilities-0.136.8}/src/utilities/more_itertools.py

@@ -28,50 +28,59 @@ if TYPE_CHECKING:
 
 
 @overload
-def bucket_mapping[T, U, UH: Hashable](
+def bucket_mapping[T, UH: Hashable](
     iterable: Iterable[T],
     func: Callable[[T], UH],
     /,
     *,
-    pre: Callable[[T], U],
+    pre: None = None,
+    post: None = None,
+) -> Mapping[UH, Iterator[T]]: ...
+@overload
+def bucket_mapping[T, UH: Hashable](
+    iterable: Iterable[T],
+    func: Callable[[T], UH],
+    /,
+    *,
+    pre: None = None,
     post: Literal["list"],
-) -> Mapping[UH, list[U]]: ...
+) -> Mapping[UH, list[T]]: ...
 @overload
-def bucket_mapping[T, U, UH: Hashable](
+def bucket_mapping[T, UH: Hashable](
     iterable: Iterable[T],
     func: Callable[[T], UH],
     /,
     *,
-    pre: Callable[[T], U],
+    pre: None = None,
     post: Literal["tuple"],
-) -> Mapping[UH, tuple[U, ...]]: ...
+) -> Mapping[UH, tuple[T, ...]]: ...
 @overload
-def bucket_mapping[T, U, UH: Hashable](
+def bucket_mapping[T, UH: Hashable](
     iterable: Iterable[T],
     func: Callable[[T], UH],
     /,
     *,
-    pre: Callable[[T], U],
+    pre: None = None,
     post: Literal["set"],
-) -> Mapping[UH, set[U]]: ...
+) -> Mapping[UH, set[T]]: ...
 @overload
-def bucket_mapping[T, U, UH: Hashable](
+def bucket_mapping[T, UH: Hashable](
     iterable: Iterable[T],
     func: Callable[[T], UH],
     /,
     *,
-    pre: Callable[[T], U],
+    pre: None = None,
     post: Literal["frozenset"],
-) -> Mapping[UH, frozenset[U]]: ...
+) -> Mapping[UH, frozenset[T]]: ...
 @overload
-def bucket_mapping[T, U, UH: Hashable](
+def bucket_mapping[T, UH: Hashable](
     iterable: Iterable[T],
     func: Callable[[T], UH],
     /,
     *,
-    pre: Callable[[T], U] | None = None,
+    pre: None = None,
     post: Literal["unique"],
-) -> Mapping[UH, U]: ...
+) -> Mapping[UH, T]: ...
 @overload
 def bucket_mapping[T, U, UH: Hashable](
     iterable: Iterable[T],
@@ -82,59 +91,50 @@ def bucket_mapping[T, U, UH: Hashable](
     post: None = None,
 ) -> Mapping[UH, Iterator[U]]: ...
 @overload
-def bucket_mapping[T, UH: Hashable](
+def bucket_mapping[T, U, UH: Hashable](
     iterable: Iterable[T],
     func: Callable[[T], UH],
     /,
     *,
-    pre: None = None,
+    pre: Callable[[T], U],
     post: Literal["list"],
-) -> Mapping[UH, list[T]]: ...
+) -> Mapping[UH, list[U]]: ...
 @overload
-def bucket_mapping[T, UH: Hashable](
+def bucket_mapping[T, U, UH: Hashable](
     iterable: Iterable[T],
     func: Callable[[T], UH],
     /,
     *,
-    pre: None = None,
+    pre: Callable[[T], U],
     post: Literal["tuple"],
-) -> Mapping[UH, tuple[T, ...]]: ...
+) -> Mapping[UH, tuple[U, ...]]: ...
 @overload
-def bucket_mapping[T, UH: Hashable](
+def bucket_mapping[T, U, UH: Hashable](
     iterable: Iterable[T],
     func: Callable[[T], UH],
     /,
     *,
-    pre: None = None,
+    pre: Callable[[T], U],
     post: Literal["set"],
-) -> Mapping[UH, set[T]]: ...
+) -> Mapping[UH, set[U]]: ...
 @overload
-def bucket_mapping[T, UH: Hashable](
+def bucket_mapping[T, U, UH: Hashable](
     iterable: Iterable[T],
     func: Callable[[T], UH],
     /,
     *,
-    pre: None = None,
+    pre: Callable[[T], U],
     post: Literal["frozenset"],
-) -> Mapping[UH, frozenset[T]]: ...
+) -> Mapping[UH, frozenset[U]]: ...
 @overload
-def bucket_mapping[T, UH: Hashable](
+def bucket_mapping[T, U, UH: Hashable](
     iterable: Iterable[T],
     func: Callable[[T], UH],
     /,
     *,
-    pre: None = None,
+    pre: Callable[[T], U] | None = None,
     post: Literal["unique"],
-) -> Mapping[UH, T]: ...
-@overload
-def bucket_mapping[T, UH: Hashable](
-    iterable: Iterable[T],
-    func: Callable[[T], UH],
-    /,
-    *,
-    pre: None = None,
-    post: None = None,
-) -> Mapping[UH, Iterator[T]]: ...
+) -> Mapping[UH, U]: ...
 @overload
 def bucket_mapping[T, U, UH: Hashable](
     iterable: Iterable[T],