dycw-utilities 0.134.0__py3-none-any.whl → 0.134.2__py3-none-any.whl

{dycw_utilities-0.134.0.dist-info → dycw_utilities-0.134.2.dist-info}/METADATA

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

{dycw_utilities-0.134.0.dist-info → dycw_utilities-0.134.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=6Kft8t6oHMNZG-gaM4ZQ07SvvDvKryeqHULU_jPaWnE,60
+utilities/__init__.py,sha256=3l_xarfe4CxRZA0THTNWrLa3nhX96z__Dw-8p3WRIqA,60
 utilities/aiolimiter.py,sha256=mD0wEiqMgwpty4XTbawFpnkkmJS6R4JRsVXFUaoitSU,628
 utilities/altair.py,sha256=HeZBVUocjkrTNwwKrClppsIqgNFF-ykv05HfZSoHYno,9104
 utilities/arq.py,sha256=S-sfBfY-E1ErRKf4sSXt2YyCjKvu-pBlOECDfjBebRA,6399
@@ -19,7 +19,7 @@ utilities/eventkit.py,sha256=cV76NIHKDPyrhnZRuhjCXse5CQqgRGXZ7p7Gon4Mu2g,12646
 utilities/fastapi.py,sha256=E8T2J1-N_RbpkN4czthU6NPIxAZDzxy-k_WGJaxeJ48,2671
 utilities/fpdf2.py,sha256=PmPj8ugr_SlxFEpw-9OsI8--mteLQ4MaXv_Cbmf7XXs,1852
 utilities/functions.py,sha256=Teqm7ylOqEcBLHWhRwjKqouUfc6nv_6qO6fILpnOPyA,27937
-utilities/functools.py,sha256=mQv5yPMAy2CPVPnbxvRHhnDX3DnnyF1Mz5DIFWzWcro,1512
+utilities/functools.py,sha256=I00ru2gQPakZw2SHVeKIKXfTv741655s6HI0lUoE0D4,1552
 utilities/getpass.py,sha256=DfN5UgMAtFCqS3dSfFHUfqIMZX2shXvwphOz_6J6f6A,103
 utilities/git.py,sha256=oi7-_l5e9haSANSCvQw25ufYGoNahuUPHAZ6114s3JQ,1191
 utilities/hashlib.py,sha256=SVTgtguur0P4elppvzOBbLEjVM3Pea0eWB61yg2ilxo,309
@@ -53,6 +53,7 @@ utilities/polars_ols.py,sha256=Uc9V5kvlWZ5cU93lKZ-cfAKdVFFw81tqwLW9PxtUvMs,5618
 utilities/pottery.py,sha256=RN3XwOEsVAPXvEfsRPmn3ZSKgTzK_c182PNrtksq-bg,3429
 utilities/pqdm.py,sha256=BTsYPtbKQWwX-iXF4qCkfPG7DPxIB54J989n83bXrIo,3092
 utilities/psutil.py,sha256=0j4YxtVb8VjaaKKiHg6UEK95SUPkEcENgPtLgPJsNv0,3760
+utilities/pudb.py,sha256=Dte7P9J0VWrWTNYvwxZX6kgaEtdTt5ZnnmOAUTjltqU,1771
 utilities/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 utilities/pydantic.py,sha256=CmxCi4sukeHM3JGjJ1Rbp8UAvcx4MZapLg10mFYJ-nk,1771
 utilities/pyinstrument.py,sha256=_Rfq6Gg4NKV2NF0dRYOpK2IRyyePxI7-3UmHIQLYrlQ,852
@@ -89,7 +90,7 @@ utilities/warnings.py,sha256=un1LvHv70PU-LLv8RxPVmugTzDJkkGXRMZTE2-fTQHw,1771
 utilities/whenever.py,sha256=A-yoOqBqrcVD1yDINDsTFDw7dq9-zgUGn_f8CxVUQJs,23332
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=oEH-nL3t4h9uawyZqWDtNtDAl6M-CLpLYGI_nI6DulM,1971
-dycw_utilities-0.134.0.dist-info/METADATA,sha256=ge1UR3zk8DRZIcXfN3RY6DR9oh5TJ-nT-A2YWjlPz_4,1584
-dycw_utilities-0.134.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.134.0.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.134.0.dist-info/RECORD,,
+dycw_utilities-0.134.2.dist-info/METADATA,sha256=jP7BRWqiFgPVdbALoM6g4cmSKGD85iB5_kNWuVV2tms,1584
+dycw_utilities-0.134.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.134.2.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.134.2.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.134.0"
+__version__ = "0.134.2"

utilities/functools.py

@@ -18,16 +18,19 @@ def cache[F: Callable](func: F, /) -> F:
 ##
 
 
+_MAX_SIZE = 128
+
+
 @overload
 def lru_cache[F: Callable](
-    func: F, /, *, max_size: int = ..., typed: bool = ...
+    func: F, /, *, max_size: int = _MAX_SIZE, typed: bool = False
 ) -> F: ...
 @overload
 def lru_cache[F: Callable](
-    func: None = None, /, *, max_size: int = ..., typed: bool = ...
+    func: None = None, /, *, max_size: int = _MAX_SIZE, typed: bool = False
 ) -> Callable[[F], F]: ...
 def lru_cache[F: Callable](
-    func: F | None = None, /, *, max_size: int = 128, typed: bool = False
+    func: F | None = None, /, *, max_size: int = _MAX_SIZE, typed: bool = False
 ) -> F | Callable[[F], F]:
     """Typed version of `lru_cache`."""
     if func is None:

utilities/pudb.py

@@ -0,0 +1,62 @@
+from __future__ import annotations
+
+from asyncio import iscoroutinefunction
+from functools import partial, wraps
+from typing import TYPE_CHECKING, Any, NoReturn, cast, overload
+
+from utilities.os import GetEnvVarError, get_env_var
+
+if TYPE_CHECKING:
+    from collections.abc import Callable
+
+
+_ENV_VAR = "DEBUG"
+
+
+@overload
+def call_pudb[F: Callable](func: F, /, *, env_var: str = _ENV_VAR) -> F: ...
+@overload
+def call_pudb[F: Callable](
+    func: None = None, /, *, env_var: str = _ENV_VAR
+) -> Callable[[F], F]: ...
+def call_pudb[F: Callable](
+    func: F | None = None, /, *, env_var: str = _ENV_VAR
+) -> F | Callable[[F], F]:
+    """Call `pudb` upon failure, if the required environment variable is set."""
+    if func is None:
+        result = partial(call_pudb, env_var=env_var)
+        return cast("Callable[[F], F]", result)
+
+    if not iscoroutinefunction(func):
+
+        @wraps(func)
+        def wrapped_sync(*args: Any, **kwargs: Any) -> Any:
+            try:
+                return func(*args, **kwargs)
+            except Exception as error:  # noqa: BLE001
+                _call_pudb(error, env_var=env_var)
+
+        return cast("F", wrapped_sync)
+
+    @wraps(func)
+    async def wrapped_async(*args: Any, **kwargs: Any) -> Any:
+        try:
+            return await func(*args, **kwargs)
+        except Exception as error:  # noqa: BLE001
+            _call_pudb(error, env_var=env_var)
+
+    return cast("F", wrapped_async)
+
+
+def _call_pudb(error: Exception, /, *, env_var: str = _ENV_VAR) -> NoReturn:
+    try:
+        _ = get_env_var(env_var)
+    except GetEnvVarError:
+        raise error from None
+    from pudb import post_mortem  # pragma: no cover
+
+    post_mortem()  # pragma: no cover
+    raise error  # pragma: no cover
+
+
+__all__ = ["call_pudb"]