dycw-utilities 0.124.0__py3-none-any.whl → 0.125.0__py3-none-any.whl

{dycw_utilities-0.124.0.dist-info → dycw_utilities-0.125.0.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.124.0
+Version: 0.125.0
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12
 Requires-Dist: typing-extensions<4.14,>=4.13.1
 Provides-Extra: test
-Requires-Dist: hypothesis<6.132,>=6.131.21; extra == 'test'
+Requires-Dist: hypothesis<6.132,>=6.131.24; extra == 'test'
 Requires-Dist: pytest-asyncio<0.27,>=0.26.0; extra == 'test'
 Requires-Dist: pytest-cov<6.2,>=6.1.1; extra == 'test'
 Requires-Dist: pytest-instafail<0.6,>=0.5.0; extra == 'test'
@@ -27,8 +27,6 @@ Requires-Dist: img2pdf<0.7,>=0.6.0; extra == 'zzz-test-altair'
 Requires-Dist: polars-lts-cpu<1.31,>=1.30.0; extra == 'zzz-test-altair'
 Requires-Dist: vl-convert-python<1.8,>=1.7.0; extra == 'zzz-test-altair'
 Requires-Dist: whenever<0.9,>=0.8.2; extra == 'zzz-test-altair'
-Provides-Extra: zzz-test-astor
-Requires-Dist: astor<0.9,>=0.8.1; extra == 'zzz-test-astor'
 Provides-Extra: zzz-test-asyncio
 Provides-Extra: zzz-test-atomicwrites
 Requires-Dist: atomicwrites<1.5,>=1.4.1; extra == 'zzz-test-atomicwrites'
@@ -80,7 +78,7 @@ Provides-Extra: zzz-test-hypothesis
 Requires-Dist: aiosqlite<0.22,>=0.21.0; extra == 'zzz-test-hypothesis'
 Requires-Dist: asyncpg<0.31,>=0.30.0; extra == 'zzz-test-hypothesis'
 Requires-Dist: greenlet<3.3,>=3.2.0; extra == 'zzz-test-hypothesis'
-Requires-Dist: hypothesis<6.132,>=6.131.21; extra == 'zzz-test-hypothesis'
+Requires-Dist: hypothesis<6.132,>=6.131.24; extra == 'zzz-test-hypothesis'
 Requires-Dist: luigi<3.7,>=3.6.0; extra == 'zzz-test-hypothesis'
 Requires-Dist: numpy<2.3,>=2.2.6; extra == 'zzz-test-hypothesis'
 Requires-Dist: pathvalidate<3.3,>=3.2.3; extra == 'zzz-test-hypothesis'

{dycw_utilities-0.124.0.dist-info → dycw_utilities-0.125.0.dist-info}/RECORD

@@ -1,6 +1,5 @@
-utilities/__init__.py,sha256=HeZfuK-rk7XFCrcSKl_fSwvax77ZgizR9BO2AjEIbLs,60
+utilities/__init__.py,sha256=QGDJ0f72zDhkWMU5d6JojuaUWYcF1Z-mqiuQN6pmnBo,60
 utilities/altair.py,sha256=Gpja-flOo-Db0PIPJLJsgzAlXWoKUjPU1qY-DQ829ek,9156
-utilities/astor.py,sha256=xuDUkjq0-b6fhtwjhbnebzbqQZAjMSHR1IIS5uOodVg,777
 utilities/asyncio.py,sha256=joGmwv-WiDLYoK4q41TtVgfIT23s8Ok46jT_yEQGjaM,28240
 utilities/atomicwrites.py,sha256=geFjn9Pwn-tTrtoGjDDxWli9NqbYfy3gGL6ZBctiqSo,5393
 utilities/atools.py,sha256=IYMuFSFGSKyuQmqD6v5IUtDlz8PPw0Sr87Cub_gRU3M,1168
@@ -36,7 +35,7 @@ utilities/luigi.py,sha256=fpH9MbxJDuo6-k9iCXRayFRtiVbUtibCJKugf7ygpv0,5988
 utilities/math.py,sha256=-mQgbah-dPJwOEWf3SonrFoVZ2AVxMgpeQ3dfVa-oJA,26764
 utilities/memory_profiler.py,sha256=tf2C51P2lCujPGvRt2Rfc7VEw5LDXmVPCG3z_AvBmbU,962
 utilities/modules.py,sha256=iuvLluJya-hvl1Q25-Jk3dLgx2Es3ck4SjJiEkAlVTs,3195
-utilities/more_itertools.py,sha256=6T0225gBFZtv47-B0JRFOKMz836Wg3Hct79ePPLGpuo,5827
+utilities/more_itertools.py,sha256=fYsGyIPB2s1KRWoH3AkV3CxJxnMLvmidqBf8l1VQnyE,7193
 utilities/numpy.py,sha256=Xn23sA2ZbVNqwUYEgNJD3XBYH6IbCri_WkHSNhg3NkY,26122
 utilities/operator.py,sha256=0M2yZJ0PODH47ogFEnkGMBe_cfxwZR02T_92LZVZvHo,3715
 utilities/optuna.py,sha256=loyJGWTzljgdJaoLhP09PT8Jz6o_pwBOwehY33lHkhw,1923
@@ -88,7 +87,7 @@ utilities/warnings.py,sha256=un1LvHv70PU-LLv8RxPVmugTzDJkkGXRMZTE2-fTQHw,1771
 utilities/whenever.py,sha256=jS31ZAY5OMxFxLja_Yo5Fidi87Pd-GoVZ7Vi_teqVDA,16743
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=-5j7IQ9nb7gR43rdgA7ms05im-XuqhAk9EJnQBXxCoQ,1874
-dycw_utilities-0.124.0.dist-info/METADATA,sha256=-1BZhlwOBb6JkkxdOqznS-XXxhlcvdZfGawiKrjB7Fc,12943
-dycw_utilities-0.124.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.124.0.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.124.0.dist-info/RECORD,,
+dycw_utilities-0.125.0.dist-info/METADATA,sha256=539k9z6K4GTl6bfPw8x0gEN_tWp0guvq15SQrIcaHjg,12852
+dycw_utilities-0.125.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.125.0.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.125.0.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.124.0"
+__version__ = "0.125.0"

utilities/more_itertools.py

@@ -37,25 +37,81 @@ _U = TypeVar("_U")
 
 @overload
 def bucket_mapping(
-    iterable: Iterable[_T], func: Callable[[_T], THashable], /, *, list: Literal[True]
+    iterable: Iterable[_T],
+    func: Callable[[_T], THashable],
+    /,
+    *,
+    transform: Callable[[_T], _U],
+    list: Literal[True],
+) -> Mapping[THashable, Sequence[_U]]: ...
+@overload
+def bucket_mapping(
+    iterable: Iterable[_T],
+    func: Callable[[_T], THashable],
+    /,
+    *,
+    transform: Callable[[_T], _U],
+    list: bool = False,
+) -> Mapping[THashable, Iterator[_U]]: ...
+@overload
+def bucket_mapping(
+    iterable: Iterable[_T],
+    func: Callable[[_T], THashable],
+    /,
+    *,
+    transform: Callable[[_T], _U] | None = None,
+    list: Literal[True],
 ) -> Mapping[THashable, Sequence[_T]]: ...
 @overload
 def bucket_mapping(
-    iterable: Iterable[_T], func: Callable[[_T], THashable], /, *, list: bool = False
+    iterable: Iterable[_T],
+    func: Callable[[_T], THashable],
+    /,
+    *,
+    transform: Callable[[_T], _U] | None = None,
+    list: bool = False,
 ) -> Mapping[THashable, Iterator[_T]]: ...
+@overload
+def bucket_mapping(
+    iterable: Iterable[_T],
+    func: Callable[[_T], THashable],
+    /,
+    *,
+    transform: Callable[[_T], _U] | None = None,
+    list: bool = False,
+) -> (
+    Mapping[THashable, Iterator[_T]]
+    | Mapping[THashable, Iterator[_U]]
+    | Mapping[THashable, Sequence[_T]]
+    | Mapping[THashable, Sequence[_U]]
+): ...
 def bucket_mapping(
     iterable: Iterable[_T],
     func: Callable[[_T], THashable],
     /,
     *,
+    transform: Callable[[_T], _U] | None = None,
     list: bool = False,  # noqa: A002
-) -> Mapping[THashable, Iterator[_T]] | Mapping[THashable, Sequence[_T]]:
+) -> (
+    Mapping[THashable, Iterator[_T]]
+    | Mapping[THashable, Iterator[_U]]
+    | Mapping[THashable, Sequence[_T]]
+    | Mapping[THashable, Sequence[_U]]
+):
     """Bucket the values of iterable into a mapping."""
     b = bucket(iterable, func)
     mapping = {key: b[key] for key in b}
-    if list:
-        return {key: builtins.list(value) for key, value in mapping.items()}
-    return mapping
+    match transform, list:
+        case None, False:
+            return mapping
+        case None, True:
+            return {k: builtins.list(v) for k, v in mapping.items()}
+        case _, False:
+            return {k: map(transform, v) for k, v in mapping.items()}
+        case _, True:
+            return {k: builtins.list(map(transform, v)) for k, v in mapping.items()}
+        case _ as never:
+            assert_never(never)
 
 
 ##

utilities/astor.py

@@ -1,28 +0,0 @@
-from __future__ import annotations
-
-from contextlib import suppress
-from subprocess import CalledProcessError, check_call
-from typing import TYPE_CHECKING
-
-from astor import to_source
-
-from utilities.tempfile import TemporaryDirectory
-
-if TYPE_CHECKING:
-    from ast import Module
-
-
-def module_to_source(module: Module, /) -> str:
-    """Write a module as a `ruff`-formatted string."""
-    src = to_source(module)
-    with TemporaryDirectory() as temp:
-        path = temp.joinpath("temp.py")
-        with path.open(mode="w") as fh:
-            _ = fh.write(src)
-        with suppress(CalledProcessError, FileNotFoundError):
-            _ = check_call(["ruff", "format", str(path)])
-        with path.open() as fh:
-            return fh.read()
-
-
-__all__ = ["module_to_source"]