dycw-utilities 0.110.7__py3-none-any.whl → 0.110.8__py3-none-any.whl

{dycw_utilities-0.110.7.dist-info → dycw_utilities-0.110.8.dist-info}/METADATA

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

{dycw_utilities-0.110.7.dist-info → dycw_utilities-0.110.8.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=RH75kOPQZPuI3PzW-iy4R8ZfTSbKQHPjZwc14TPFLN8,60
+utilities/__init__.py,sha256=USEDB4wi-U2inTRQAJuZ7L5i_LnX5kcGM17wMf2dOck,60
 utilities/altair.py,sha256=Gpja-flOo-Db0PIPJLJsgzAlXWoKUjPU1qY-DQ829ek,9156
 utilities/astor.py,sha256=xuDUkjq0-b6fhtwjhbnebzbqQZAjMSHR1IIS5uOodVg,777
 utilities/asyncio.py,sha256=41oQUurWMvadFK5gFnaG21hMM0Vmfn2WS6OpC0R9mas,14757
@@ -78,7 +78,7 @@ utilities/threading.py,sha256=GvBOp4CyhHfN90wGXZuA2VKe9fGzMaEa7oCl4f3nnPU,1009
 utilities/timer.py,sha256=Rkc49KSpHuC8s7vUxGO9DU55U9I6yDKnchsQqrUCVBs,4075
 utilities/traceback.py,sha256=KwHPLdEbdj0fFhXo8MBfxcvem8A-VXYDwFMNJ6f0cTM,27328
 utilities/types.py,sha256=kVY71hZkcnyYNIlYSse0mLm8yeP3OBkzhDPMME6jXxo,18126
-utilities/typing.py,sha256=gLg4EbE1FX52fJ1d3ji4i08qolwu9qgWt8w_w_Y5DTk,5512
+utilities/typing.py,sha256=WE3UWaVYO6nTkpEo8Ke3xUuxllsJtfOxsZ0-mSH69Nw,6668
 utilities/tzdata.py,sha256=2ZsPmhTVM9Ptrxb4QrWKtKOB9RiH8IOO-A1u7ULdVbg,176
 utilities/tzlocal.py,sha256=42BCquGF54oIqIKe5RGziP4K8Nbm3Ey7uqcNn6m5ge8,534
 utilities/uuid.py,sha256=jJTFxz-CWgltqNuzmythB7iEQ-Q1mCwPevUfKthZT3c,611
@@ -87,7 +87,7 @@ utilities/warnings.py,sha256=un1LvHv70PU-LLv8RxPVmugTzDJkkGXRMZTE2-fTQHw,1771
 utilities/whenever.py,sha256=TjoTAJ1R27-rKXiXzdE4GzPidmYqm0W58XydDXp-QZM,17786
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=-DQz5a0Ikw9jfSZtL0BEQkXOMC9yGn_xiJYNCLMiqEc,1989
-dycw_utilities-0.110.7.dist-info/METADATA,sha256=wFw9aZuOwiyPR1o4QtTmz5fPx9y5gBbKheuvV0XKhA8,13004
-dycw_utilities-0.110.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.110.7.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.110.7.dist-info/RECORD,,
+dycw_utilities-0.110.8.dist-info/METADATA,sha256=ZVWP8hzsSisjUrY3uoJqCIoh5j7pVhSy0PwDigCaJRw,13004
+dycw_utilities-0.110.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.110.8.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.110.8.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.110.7"
+__version__ = "0.110.8"

utilities/typing.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import datetime as dt
 from collections.abc import Mapping, Sequence
+from dataclasses import dataclass
 from itertools import chain
 from pathlib import Path
 from types import NoneType, UnionType
@@ -15,6 +16,7 @@ from typing import (
     TypeGuard,
     Union,  # pyright: ignore[reportDeprecated]
     get_origin,
+    override,
 )
 from typing import get_args as _get_args
 from typing import get_type_hints as _get_type_hints
@@ -64,7 +66,7 @@ def _get_literal_elements_inner(obj: Any, /) -> list[Any]:
 
 
 def get_type_hints(
-    cls: Any,
+    obj: Any,
     /,
     *,
     globalns: StrMapping | None = None,
@@ -72,15 +74,15 @@ def get_type_hints(
     warn_name_errors: bool = False,
 ) -> dict[str, Any]:
     """Get the type hints of an object."""
-    result: dict[str, Any] = cls.__annotations__
+    result: dict[str, Any] = obj.__annotations__
     _ = {Literal, Path, Sentinel, StrMapping, UUID, dt}
     globalns_use = globals() | ({} if globalns is None else dict(globalns))
     localns_use = {} if localns is None else dict(localns)
     try:
-        hints = _get_type_hints(cls, globalns=globalns_use, localns=localns_use)
+        hints = _get_type_hints(obj, globalns=globalns_use, localns=localns_use)
     except NameError as error:
         if warn_name_errors:
-            warn(f"Error getting type hints for {cls!r}; {error}", stacklevel=2)
+            warn(f"Error getting type hints for {obj!r}; {error}", stacklevel=2)
     else:
         result.update({
             key: value
@@ -93,6 +95,44 @@ def get_type_hints(
 ##
 
 
+def get_union_type_classes(obj: Any, /) -> tuple[type[Any], ...]:
+    if not is_union_type(obj):
+        raise _GetUnionTypeClassesNotAUnionTypeError(obj=obj)
+    types_: Sequence[type[Any]] = []
+    for arg in get_args(obj):
+        if isinstance(arg, type):
+            types_.append(arg)
+        elif is_union_type(arg):
+            types_.extend(get_union_type_classes(arg))
+        else:
+            raise _GetUnionTypeClassesNotATypeError(obj=obj, inner=arg)
+    return tuple(types_)
+
+
+@dataclass(kw_only=True, slots=True)
+class GetUnionTypeClassesError(Exception):
+    obj: Any
+
+
+@dataclass(kw_only=True, slots=True)
+class _GetUnionTypeClassesNotAUnionTypeError(GetUnionTypeClassesError):
+    @override
+    def __str__(self) -> str:
+        return f"Object must be a Union type; got {self.obj}"
+
+
+@dataclass(kw_only=True, slots=True)
+class _GetUnionTypeClassesNotATypeError(GetUnionTypeClassesError):
+    inner: Any
+
+    @override
+    def __str__(self) -> str:
+        return f"Union type must contain types only; got {self.inner}"
+
+
+##
+
+
 def is_dict_type(obj: Any, /) -> bool:
     """Check if an object is a dict type annotation."""
     return _is_annotation_of_type(obj, dict)
@@ -207,9 +247,11 @@ def _is_annotation_of_type(obj: Any, origin: Any, /) -> bool:
 
 
 __all__ = [
+    "GetUnionTypeClassesError",
     "contains_self",
     "get_literal_elements",
     "get_type_hints",
+    "get_union_type_classes",
     "is_dict_type",
     "is_frozenset_type",
     "is_list_type",