dycw-utilities 0.125.3__py3-none-any.whl → 0.125.5__py3-none-any.whl

{dycw_utilities-0.125.3.dist-info → dycw_utilities-0.125.5.dist-info}/METADATA

@@ -1,16 +1,16 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.125.3
+Version: 0.125.5
 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.27; extra == 'test'
+Requires-Dist: hypothesis<6.132,>=6.131.28; 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'
-Requires-Dist: pytest-lazy-fixtures<1.2,>=1.1.2; extra == 'test'
+Requires-Dist: pytest-lazy-fixtures<1.2,>=1.1.3; extra == 'test'
 Requires-Dist: pytest-only<2.2,>=2.1.2; extra == 'test'
 Requires-Dist: pytest-randomly<3.17,>=3.16.0; extra == 'test'
 Requires-Dist: pytest-regressions<2.8,>=2.7.0; extra == 'test'
@@ -41,7 +41,7 @@ Requires-Dist: whenever<0.9,>=0.8.2; extra == 'zzz-test-click'
 Provides-Extra: zzz-test-contextlib
 Provides-Extra: zzz-test-contextvars
 Provides-Extra: zzz-test-cryptography
-Requires-Dist: cryptography<45.1,>=45.0.2; extra == 'zzz-test-cryptography'
+Requires-Dist: cryptography<45.1,>=45.0.3; extra == 'zzz-test-cryptography'
 Provides-Extra: zzz-test-cvxpy
 Requires-Dist: cvxpy<1.7,>=1.6.5; extra == 'zzz-test-cvxpy'
 Provides-Extra: zzz-test-dataclasses
@@ -78,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.27; extra == 'zzz-test-hypothesis'
+Requires-Dist: hypothesis<6.132,>=6.131.28; 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'
@@ -174,7 +174,7 @@ Requires-Dist: scipy<1.16,>=1.15.3; extra == 'zzz-test-scipy'
 Provides-Extra: zzz-test-sentinel
 Provides-Extra: zzz-test-shelve
 Provides-Extra: zzz-test-slack-sdk
-Requires-Dist: aiohttp<3.12,>=3.11.16; extra == 'zzz-test-slack-sdk'
+Requires-Dist: aiohttp<3.13,>=3.12.0; extra == 'zzz-test-slack-sdk'
 Requires-Dist: slack-sdk<3.36,>=3.35.0; extra == 'zzz-test-slack-sdk'
 Provides-Extra: zzz-test-socket
 Provides-Extra: zzz-test-sqlalchemy

{dycw_utilities-0.125.3.dist-info → dycw_utilities-0.125.5.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=S9hi5nUNGrmUoeHSNNw9CdOdoY68OUGHaOm0V7chzZo,60
+utilities/__init__.py,sha256=DteTSv79-JnoG5SiUi0i4Es8z2F38hAIDg1-tlzG4mc,60
 utilities/altair.py,sha256=Gpja-flOo-Db0PIPJLJsgzAlXWoKUjPU1qY-DQ829ek,9156
 utilities/asyncio.py,sha256=gr2eUx0E6LiCup6VKgUGwh8lAUriGdX2TlK-PZdlvfo,28284
 utilities/atomicwrites.py,sha256=geFjn9Pwn-tTrtoGjDDxWli9NqbYfy3gGL6ZBctiqSo,5393
@@ -26,9 +26,10 @@ utilities/http.py,sha256=WcahTcKYRtZ04WXQoWt5EGCgFPcyHD3EJdlMfxvDt-0,946
 utilities/hypothesis.py,sha256=a75izXg9aCBhhDkj_ZgK3TDzlzk38evP8TO7JbYYQvg,46264
 utilities/importlib.py,sha256=mV1xT_O_zt_GnZZ36tl3xOmMaN_3jErDWY54fX39F6Y,429
 utilities/ipython.py,sha256=V2oMYHvEKvlNBzxDXdLvKi48oUq2SclRg5xasjaXStw,763
-utilities/iterables.py,sha256=prKXBdF5QfLTGC-q4567DwO8xzUng_Z-2a4wBkMqyDo,45360
+utilities/iterables.py,sha256=wqq2lKfEAq1qncE-lR848DHX627aerFaq4BXmLen-NE,43589
 utilities/jupyter.py,sha256=ft5JA7fBxXKzP-L9W8f2-wbF0QeYc_2uLQNFDVk4Z-M,2917
-utilities/lightweight_charts.py,sha256=vyVOzarYhBIOZj2xDhqdbP85qbSKUjdc6Au91rc1W4M,2814
+utilities/libcst.py,sha256=zqZxfvTdcsi83Ph-HC4lP8zyOqAun7eKABeRdhMtoMM,4846
+utilities/lightweight_charts.py,sha256=0xNfcsrgFI0R9xL25LtSm-W5yhfBI93qQNT6HyaXAhg,2769
 utilities/logging.py,sha256=gwo3pusPjnWO1ollrtn1VKYyRAQJTue4SkCbMeNvec4,25715
 utilities/loguru.py,sha256=MEMQVWrdECxk1e3FxGzmOf21vWT9j8CAir98SEXFKPA,3809
 utilities/luigi.py,sha256=fpH9MbxJDuo6-k9iCXRayFRtiVbUtibCJKugf7ygpv0,5988
@@ -46,14 +47,14 @@ utilities/pathlib.py,sha256=31WPMXdLIyXgYOMMl_HOI2wlo66MGSE-cgeelk-Lias,1410
 utilities/period.py,sha256=o4wXYEXVlFomop4-Ra4L0yRP4i99NZFjIe_fa7NdZck,11024
 utilities/pickle.py,sha256=Bhvd7cZl-zQKQDFjUerqGuSKlHvnW1K2QXeU5UZibtg,657
 utilities/platform.py,sha256=NU7ycTvAXAG-fdYmDXaM1m4EOml2cGiaYwaUzfzSqyU,1767
-utilities/polars.py,sha256=tGO3r6S4lFtir6zbdIvZ_kvKBWIha4lnA1eEmV3EvNQ,63632
-utilities/polars_ols.py,sha256=efhXf0gjrHUpQrvS6a7g8yJQJWf_ATKtJnqqF2inCOU,5680
+utilities/polars.py,sha256=QlmUpYTqHNkcLnWOQh1TW22W2QyLzvifCvBcbsqhpdE,63272
+utilities/polars_ols.py,sha256=Uc9V5kvlWZ5cU93lKZ-cfAKdVFFw81tqwLW9PxtUvMs,5618
 utilities/pqdm.py,sha256=foRytQybmOQ05pjt5LF7ANyzrIa--4ScDE3T2wd31a4,3118
 utilities/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 utilities/pydantic.py,sha256=f6qtR5mO2YMuyvNmbaEj5YeD9eGA4YYfb7Bjzh9jUs0,1845
 utilities/pyinstrument.py,sha256=OJFDh4o1CWIa4aYPYURdQjgap_nvP45KUsCEe94rQHY,829
-utilities/pyrsistent.py,sha256=MoDcAqQGlSNkmlS32DCJLw-cZFAfHB6K9kpox_iyI4k,2512
-utilities/pytest.py,sha256=85QUax4g2VBBAqAHtM9wekcSLB7_9O8AKFTaCshztL8,7989
+utilities/pyrsistent.py,sha256=wVOVIe_68AAaa-lUE9y-TEzDawVp1uEIc_zfoDgr5ww,2287
+utilities/pytest.py,sha256=KoHSwJbIY2CHtFUlUr_gnEk7z1DVTaldl8RDQ4tDkG4,7837
 utilities/pytest_regressions.py,sha256=-SVT9647Dg6-JcdsiaDKXe3NdOmmrvGevLKWwGjxq3c,5088
 utilities/python_dotenv.py,sha256=iWcnpXbH7S6RoXHiLlGgyuH6udCupAcPd_gQ0eAenQ0,3190
 utilities/random.py,sha256=lYdjgxB7GCfU_fwFVl5U-BIM_HV3q6_urL9byjrwDM8,4157
@@ -67,7 +68,7 @@ utilities/shelve.py,sha256=HZsMwK4tcIfg3sh0gApx4-yjQnrY4o3V3ZRimvRhoW0,738
 utilities/slack_sdk.py,sha256=Q8UakiB7qo6SUfaBDB0j1N4b8MuFzaD9lG5HGq7rtuw,3200
 utilities/socket.py,sha256=K77vfREvzoVTrpYKo6MZakol0EYu2q1sWJnnZqL0So0,118
 utilities/sqlalchemy.py,sha256=p8vsHaNRoeq5zJouIKyp9piFM26wtm5yR4DkzCMFDSw,35471
-utilities/sqlalchemy_polars.py,sha256=wjJpoUo-yO9E2ujpG_06vV5r2OdvBiQ4yvV6wKCa2Tk,15605
+utilities/sqlalchemy_polars.py,sha256=s7hQNep2O5DTgIRXyN_JRQma7a4DAtNd25tshaZW8iw,15490
 utilities/statsmodels.py,sha256=koyiBHvpMcSiBfh99wFUfSggLNx7cuAw3rwyfAhoKpQ,3410
 utilities/streamlit.py,sha256=U9PJBaKP1IdSykKhPZhIzSPTZsmLsnwbEPZWzNhJPKk,2955
 utilities/sys.py,sha256=h0Xr7Vj86wNalvwJVP1wj5Y0kD_VWm1vzuXZ_jw94mE,2743
@@ -87,7 +88,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.125.3.dist-info/METADATA,sha256=vMdoTZVKyi-BNSOGgeJYjY6uY-icfbVmvUlAo_SwJmM,12852
-dycw_utilities-0.125.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.125.3.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.125.3.dist-info/RECORD,,
+dycw_utilities-0.125.5.dist-info/METADATA,sha256=OP9MTsqYyVzlwUsK9XdBKW3Z_5p4PNCNC3K8Io5dyMs,12851
+dycw_utilities-0.125.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.125.5.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.125.5.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.125.3"
+__version__ = "0.125.5"

utilities/iterables.py

@@ -25,7 +25,6 @@ from typing import (
     Any,
     Generic,
     Literal,
-    Self,
     TypeGuard,
     TypeVar,
     assert_never,
@@ -45,7 +44,7 @@ from utilities.math import (
 )
 from utilities.reprlib import get_repr
 from utilities.sentinel import Sentinel, sentinel
-from utilities.types import Sign, THashable, THashable2, TSupportsAdd, TSupportsLT
+from utilities.types import Sign, THashable, TSupportsAdd, TSupportsLT
 from utilities.zoneinfo import UTC
 
 if TYPE_CHECKING:
@@ -722,57 +721,6 @@ def chunked(iterable: Iterable[_T], n: int, /) -> Iterator[Sequence[_T]]:
 ##
 
 
-class Collection(frozenset[THashable]):
-    """A collection of hashable, sortable items."""
-
-    def __new__(cls, *item_or_items: MaybeIterable[THashable]) -> Self:
-        items = list(chain(*map(always_iterable, item_or_items)))
-        cls.check_items(items)
-        return super().__new__(cls, items)
-
-    def __init__(self, *item_or_items: MaybeIterable[THashable]) -> None:
-        super().__init__()
-        _ = item_or_items
-
-    @override
-    def __and__(self, other: MaybeIterable[THashable], /) -> Self:
-        if isinstance(other, type(self)):
-            return type(self)(super().__and__(other))
-        return self.__and__(type(self)(other))
-
-    @override
-    def __or__(self, other: MaybeIterable[THashable], /) -> Self:
-        if isinstance(other, type(self)):
-            return type(self)(super().__or__(other))
-        return self.__or__(type(self)(other))
-
-    @override
-    def __sub__(self, other: MaybeIterable[THashable], /) -> Self:
-        if isinstance(other, type(self)):
-            return type(self)(super().__sub__(other))
-        return self.__sub__(type(self)(other))
-
-    @classmethod
-    def check_items(cls, items: Iterable[THashable], /) -> None:
-        _ = items
-
-    def filter(self, func: Callable[[THashable], bool], /) -> Self:
-        return type(self)(filter(func, self))
-
-    def map(self, func: Callable[[THashable], THashable2], /) -> Collection[THashable2]:
-        values = cast("Any", map(func, self))
-        return cast("Any", type(self)(values))
-
-    def partition(self, func: Callable[[THashable], bool], /) -> tuple[Self, Self]:
-        from more_itertools import partition
-
-        is_false, is_true = partition(func, self)
-        return type(self)(is_false), type(self)(is_true)
-
-
-##
-
-
 def ensure_hashables(
     *args: Any, **kwargs: Any
 ) -> tuple[list[Hashable], dict[str, Hashable]]:

utilities/libcst.py

@@ -0,0 +1,176 @@
+from __future__ import annotations
+
+from collections.abc import Sequence
+from dataclasses import dataclass
+from subprocess import check_output
+from typing import assert_never, override
+
+from libcst import (
+    AsName,
+    Attribute,
+    BaseExpression,
+    FormattedString,
+    FormattedStringExpression,
+    FormattedStringText,
+    Import,
+    ImportAlias,
+    ImportFrom,
+    ImportStar,
+    Module,
+    Name,
+)
+
+from utilities.errors import ImpossibleCaseError
+
+
+def generate_from_import(
+    module: str, name: str, /, *, asname: str | None = None
+) -> ImportFrom:
+    """Generate an `ImportFrom` object."""
+    alias = ImportAlias(
+        name=Name(name), asname=AsName(Name(asname)) if asname else None
+    )
+    return ImportFrom(module=split_dotted_str(module), names=[alias])
+
+
+def generate_f_string(var: str, suffix: str, /) -> FormattedString:
+    """Generate an f-string."""
+    return FormattedString([
+        FormattedStringExpression(expression=Name(var)),
+        FormattedStringText(suffix),
+    ])
+
+
+def generate_import(module: str, /, *, asname: str | None = None) -> Import:
+    """Generate an `Import` object."""
+    alias = ImportAlias(
+        name=split_dotted_str(module), asname=AsName(Name(asname)) if asname else None
+    )
+    return Import(names=[alias])
+
+
+##
+
+
+@dataclass(kw_only=True, slots=True)
+class _ParseImportOutput:
+    module: str
+    name: str | None = None
+
+
+def parse_import(import_: Import | ImportFrom, /) -> Sequence[_ParseImportOutput]:
+    """Parse an import."""
+    match import_:
+        case Import():
+            return [_parse_import_one(n) for n in import_.names]
+        case ImportFrom():
+            if (attr_or_name := import_.module) is None:
+                raise _ParseImportEmptyModuleError(import_=import_)
+            module = join_dotted_str(attr_or_name)
+            match import_.names:
+                case Sequence() as names:
+                    return [_parse_import_from_one(module, n) for n in names]
+                case ImportStar():
+                    return [_ParseImportOutput(module=module, name="*")]
+                case _ as never:
+                    assert_never(never)
+        case _ as never:
+            assert_never(never)
+
+
+def _parse_import_one(alias: ImportAlias, /) -> _ParseImportOutput:
+    return _ParseImportOutput(module=join_dotted_str(alias.name))
+
+
+def _parse_import_from_one(module: str, alias: ImportAlias, /) -> _ParseImportOutput:
+    match alias.name:
+        case Name(name):
+            return _ParseImportOutput(module=module, name=name)
+        case Attribute() as attr:
+            raise _ParseImportAliasError(module=module, attr=attr)
+        case _ as never:
+            assert_never(never)
+
+
+@dataclass(kw_only=True, slots=True)
+class ParseImportError(Exception): ...
+
+
+@dataclass(kw_only=True, slots=True)
+class _ParseImportEmptyModuleError(ParseImportError):
+    import_: ImportFrom
+
+    @override
+    def __str__(self) -> str:
+        return f"Module must not be None; got {self.import_}"
+
+
+@dataclass(kw_only=True, slots=True)
+class _ParseImportAliasError(ParseImportError):
+    module: str
+    attr: Attribute
+
+    @override
+    def __str__(self) -> str:
+        attr = self.attr
+        return f"Invalid alias name; got module {self.module!r} and attribute '{attr.value}.{attr.attr}'"
+
+
+##
+
+
+def split_dotted_str(dotted: str, /) -> Name | Attribute:
+    """Split a dotted string into a name/attribute."""
+    parts = dotted.split(".")
+    node = Name(parts[0])
+    for part in parts[1:]:
+        node = Attribute(value=node, attr=Name(part))
+    return node
+
+
+def join_dotted_str(name_or_attr: Name | Attribute, /) -> str:
+    """Join a dotted from from a name/attribute."""
+    parts: Sequence[str] = []
+    curr: BaseExpression | Name | Attribute = name_or_attr
+    while True:
+        match curr:
+            case Name(value=value):
+                parts.append(value)
+                break
+            case Attribute(value=value, attr=Name(value=attr_value)):
+                parts.append(attr_value)
+                curr = value
+            case BaseExpression():  # pragma: no cover
+                raise ImpossibleCaseError(case=[f"{curr=}"])
+            case _ as never:
+                assert_never(never)
+    return ".".join(reversed(parts))
+
+
+##
+
+
+def render_module(source: str | Module, /) -> str:
+    """Render a module."""
+    match source:  # skipif-ci
+        case str() as text:
+            return check_output(["ruff", "format", "-"], input=text, text=True)
+        case Module() as module:
+            return render_module(module.code)
+        case _ as never:
+            assert_never(never)
+
+
+##
+
+
+__all__ = [
+    "ParseImportError",
+    "generate_f_string",
+    "generate_from_import",
+    "generate_import",
+    "join_dotted_str",
+    "parse_import",
+    "render_module",
+    "split_dotted_str",
+]

utilities/lightweight_charts.py

@@ -13,7 +13,7 @@ if TYPE_CHECKING:
     from lightweight_charts import AbstractChart, Chart
     from lightweight_charts.abstract import SeriesCommon
     from polars import DataFrame
-    from polars._typing import SchemaDict  # pyright: ignore[reportPrivateImportUsage]
+    from polars._typing import SchemaDict
 
     from utilities.types import PathLike
 

utilities/polars.py

@@ -112,14 +112,14 @@ if TYPE_CHECKING:
     from collections.abc import Set as AbstractSet
 
     from polars._typing import (
-        IntoExpr,  # pyright: ignore[reportPrivateImportUsage]
-        IntoExprColumn,  # pyright: ignore[reportPrivateImportUsage]
-        JoinStrategy,  # pyright: ignore[reportPrivateImportUsage]
-        JoinValidation,  # pyright: ignore[reportPrivateImportUsage]
-        PolarsDataType,  # pyright: ignore[reportPrivateImportUsage]
-        QuantileMethod,  # pyright: ignore[reportPrivateImportUsage]
-        SchemaDict,  # pyright: ignore[reportPrivateImportUsage]
-        TimeUnit,  # pyright: ignore[reportPrivateImportUsage]
+        IntoExpr,
+        IntoExprColumn,
+        JoinStrategy,
+        JoinValidation,
+        PolarsDataType,
+        QuantileMethod,
+        SchemaDict,
+        TimeUnit,
     )
 
     from utilities.numpy import NDArrayB, NDArrayF

utilities/polars_ols.py

@@ -10,9 +10,7 @@ from utilities.functions import is_sequence_of
 from utilities.polars import concat_series, ensure_expr_or_series
 
 if TYPE_CHECKING:
-    from polars._typing import (
-        IntoExprColumn,  # pyright: ignore[reportPrivateImportUsage]
-    )
+    from polars._typing import IntoExprColumn
     from polars_ols import NullPolicy
 
     from utilities.polars import ExprLike

utilities/pyrsistent.py

@@ -5,11 +5,11 @@ from typing import TYPE_CHECKING, Any, TypeVar, dataclass_transform, overload
 from pyrsistent import PRecord as _PRecord
 from pyrsistent import field as _field
 from pyrsistent._field_common import (
-    PFIELD_NO_FACTORY,  # pyright: ignore[reportPrivateImportUsage]
-    PFIELD_NO_INITIAL,  # pyright: ignore[reportPrivateImportUsage]
-    PFIELD_NO_INVARIANT,  # pyright: ignore[reportPrivateImportUsage]
-    PFIELD_NO_SERIALIZER,  # pyright: ignore[reportPrivateImportUsage]
-    PFIELD_NO_TYPE,  # pyright: ignore[reportPrivateImportUsage]
+    PFIELD_NO_FACTORY,
+    PFIELD_NO_INITIAL,
+    PFIELD_NO_INVARIANT,
+    PFIELD_NO_SERIALIZER,
+    PFIELD_NO_TYPE,
 )
 
 if TYPE_CHECKING:

utilities/pytest.py

@@ -35,11 +35,9 @@ if TYPE_CHECKING:
     )
 
 try:  # WARNING: this package cannot use unguarded `pytest` imports
-    from _pytest.config import Config  # pyright: ignore[reportPrivateImportUsage]
-    from _pytest.config.argparsing import (
-        Parser,  # pyright: ignore[reportPrivateImportUsage]
-    )
-    from _pytest.python import Function  # pyright: ignore[reportPrivateImportUsage]
+    from _pytest.config import Config
+    from _pytest.config.argparsing import Parser
+    from _pytest.python import Function
     from pytest import mark, skip
 except ModuleNotFoundError:  # pragma: no cover
     from typing import Any as Config

utilities/sqlalchemy_polars.py

@@ -60,10 +60,7 @@ if TYPE_CHECKING:
         Mapping,
     )
 
-    from polars._typing import (
-        PolarsDataType,  # pyright: ignore[reportPrivateImportUsage]
-        SchemaDict,  # pyright: ignore[reportPrivateImportUsage]
-    )
+    from polars._typing import PolarsDataType, SchemaDict
     from sqlalchemy.sql import ColumnCollection
     from sqlalchemy.sql.base import ReadOnlyColumnCollection
     from tenacity.retry import RetryBaseT