dycw-utilities 0.126.2__py3-none-any.whl → 0.126.4__py3-none-any.whl

{dycw_utilities-0.126.2.dist-info → dycw_utilities-0.126.4.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.126.2
+Version: 0.126.4
 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.32; extra == 'test'
+Requires-Dist: hypothesis<6.133,>=6.132.0; extra == 'test'
 Requires-Dist: pytest-asyncio<1.1,>=1.0.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'
@@ -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.32; extra == 'zzz-test-hypothesis'
+Requires-Dist: hypothesis<6.133,>=6.132.0; 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.126.2.dist-info → dycw_utilities-0.126.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=oG_3_PTYR4RRwJNt7YUNCU570YlGyZIR_E1X-yfdGqY,60
+utilities/__init__.py,sha256=q-AzOWuCGiffIcmyXrrhqNK_TnB_0a0oNDlPGkcdXQE,60
 utilities/altair.py,sha256=Gpja-flOo-Db0PIPJLJsgzAlXWoKUjPU1qY-DQ829ek,9156
 utilities/asyncio.py,sha256=K5Kj7rsM0nA17-b7d7mrNgPR1U_NbkfQmTruq5LBLRA,51778
 utilities/atomicwrites.py,sha256=geFjn9Pwn-tTrtoGjDDxWli9NqbYfy3gGL6ZBctiqSo,5393
@@ -23,12 +23,12 @@ utilities/getpass.py,sha256=DfN5UgMAtFCqS3dSfFHUfqIMZX2shXvwphOz_6J6f6A,103
 utilities/git.py,sha256=wpt5dZ5Oi5931pN24_VLZYaQOvmR0OcQuVtgHzFUN1k,2359
 utilities/hashlib.py,sha256=SVTgtguur0P4elppvzOBbLEjVM3Pea0eWB61yg2ilxo,309
 utilities/http.py,sha256=WcahTcKYRtZ04WXQoWt5EGCgFPcyHD3EJdlMfxvDt-0,946
-utilities/hypothesis.py,sha256=a75izXg9aCBhhDkj_ZgK3TDzlzk38evP8TO7JbYYQvg,46264
+utilities/hypothesis.py,sha256=DwhPpnz9w5H9YxLpoXTRjhTpadDjrX5-yUVT-Kq6Tgc,45216
 utilities/importlib.py,sha256=mV1xT_O_zt_GnZZ36tl3xOmMaN_3jErDWY54fX39F6Y,429
 utilities/ipython.py,sha256=V2oMYHvEKvlNBzxDXdLvKi48oUq2SclRg5xasjaXStw,763
 utilities/iterables.py,sha256=mDqw2_0MUVp-P8FklgcaVTi2TXduH0MxbhTDzzhSBho,44915
 utilities/jupyter.py,sha256=ft5JA7fBxXKzP-L9W8f2-wbF0QeYc_2uLQNFDVk4Z-M,2917
-utilities/libcst.py,sha256=L0IkwAWXPuNpYyceeJGOC8CAwHdVPRpbXC8MQphCXh4,4954
+utilities/libcst.py,sha256=Jto5ppzRzsxn4AD32IS8n0lbgLYXwsVJB6EY8giNZyY,4974
 utilities/lightweight_charts.py,sha256=0xNfcsrgFI0R9xL25LtSm-W5yhfBI93qQNT6HyaXAhg,2769
 utilities/logging.py,sha256=gwo3pusPjnWO1ollrtn1VKYyRAQJTue4SkCbMeNvec4,25715
 utilities/loguru.py,sha256=MEMQVWrdECxk1e3FxGzmOf21vWT9j8CAir98SEXFKPA,3809
@@ -49,8 +49,9 @@ utilities/pickle.py,sha256=Bhvd7cZl-zQKQDFjUerqGuSKlHvnW1K2QXeU5UZibtg,657
 utilities/platform.py,sha256=48IOKx1IC6ZJXWG-b56ZQptITcNFhWRjELW72o2dGTA,2398
 utilities/polars.py,sha256=QlmUpYTqHNkcLnWOQh1TW22W2QyLzvifCvBcbsqhpdE,63272
 utilities/polars_ols.py,sha256=Uc9V5kvlWZ5cU93lKZ-cfAKdVFFw81tqwLW9PxtUvMs,5618
+utilities/pottery.py,sha256=_PA24g2B23etpiF2awZ4oR7RciUgEHu_j7Ls3QKMkjY,1527
 utilities/pqdm.py,sha256=foRytQybmOQ05pjt5LF7ANyzrIa--4ScDE3T2wd31a4,3118
-utilities/psutil.py,sha256=0dXHKW7t9J3uzY3YtHGiRiZVRWr0uozccioZRCVFQfk,3761
+utilities/psutil.py,sha256=RtbLKOoIJhqrJmEoHDBVeSD-KPzshtS0FtRXBP9_w2s,3751
 utilities/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 utilities/pydantic.py,sha256=f6qtR5mO2YMuyvNmbaEj5YeD9eGA4YYfb7Bjzh9jUs0,1845
 utilities/pyinstrument.py,sha256=OJFDh4o1CWIa4aYPYURdQjgap_nvP45KUsCEe94rQHY,829
@@ -60,7 +61,7 @@ utilities/pytest_regressions.py,sha256=-SVT9647Dg6-JcdsiaDKXe3NdOmmrvGevLKWwGjxq
 utilities/python_dotenv.py,sha256=iWcnpXbH7S6RoXHiLlGgyuH6udCupAcPd_gQ0eAenQ0,3190
 utilities/random.py,sha256=lYdjgxB7GCfU_fwFVl5U-BIM_HV3q6_urL9byjrwDM8,4157
 utilities/re.py,sha256=5J4d8VwIPFVrX2Eb8zfoxImDv7IwiN_U7mJ07wR2Wvs,3958
-utilities/redis.py,sha256=DTkNR_caxlx06rH-ugcClroErKsu6NDGMYGIgEng05c,32570
+utilities/redis.py,sha256=U-LN6Li3vPTQ2sGGh85_1eD53payujg4qYX0TfmbLSs,32205
 utilities/reprlib.py,sha256=Re9bk3n-kC__9DxQmRlevqFA86pE6TtVfWjUgpbVOv0,1849
 utilities/rich.py,sha256=t50MwwVBsoOLxzmeVFSVpjno4OW6Ufum32skXbV8-Bs,1911
 utilities/scipy.py,sha256=X6ROnHwiUhAmPhM0jkfEh0-Fd9iRvwiqtCQMOLmOQF8,945
@@ -90,7 +91,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.126.2.dist-info/METADATA,sha256=0QZHN43mRwTBwhoYhygzaYieXnLM6qsG4ANxaqYs4Ls,12851
-dycw_utilities-0.126.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.126.2.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.126.2.dist-info/RECORD,,
+dycw_utilities-0.126.4.dist-info/METADATA,sha256=GiVs9mIgDnNbbpw9H2PkBdxD3LIObcakDua9XR5Xh6g,12849
+dycw_utilities-0.126.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.126.4.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.126.4.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.126.2"
+__version__ = "0.126.4"

utilities/hypothesis.py

@@ -2,22 +2,18 @@ from __future__ import annotations
 
 import builtins
 import datetime as dt
-from contextlib import (
-    AbstractAsyncContextManager,
-    asynccontextmanager,
-    contextmanager,
-    suppress,
-)
+from contextlib import contextmanager
 from dataclasses import dataclass
 from datetime import timezone
 from enum import Enum, auto
 from functools import partial
 from math import ceil, floor, inf, isclose, isfinite, nan
-from os import environ
+from os import environ, getpid
 from pathlib import Path
 from re import search
 from string import ascii_letters, ascii_lowercase, ascii_uppercase, digits, printable
 from subprocess import check_call
+from threading import get_ident
 from typing import (
     TYPE_CHECKING,
     Any,
@@ -92,27 +88,19 @@ from utilities.pathlib import temp_cwd
 from utilities.platform import IS_WINDOWS
 from utilities.sentinel import Sentinel, sentinel
 from utilities.tempfile import TEMP_DIR, TemporaryDirectory
+from utilities.tzlocal import get_now_local
 from utilities.version import Version
 from utilities.zoneinfo import UTC
 
 if TYPE_CHECKING:
-    from collections.abc import (
-        AsyncIterator,
-        Collection,
-        Hashable,
-        Iterable,
-        Iterator,
-        Sequence,
-    )
+    from collections.abc import Collection, Hashable, Iterable, Iterator, Sequence
     from zoneinfo import ZoneInfo
 
     from hypothesis.database import ExampleDatabase
     from numpy.random import RandomState
-    from redis.typing import KeyT
     from sqlalchemy.ext.asyncio import AsyncEngine
 
     from utilities.numpy import NDArrayB, NDArrayF, NDArrayI, NDArrayO
-    from utilities.redis import _TestRedis
     from utilities.sqlalchemy import Dialect, TableOrORMInstOrClass
     from utilities.types import Duration, Number, RoundMode
 
@@ -1422,6 +1410,19 @@ def uint64s(
 ##
 
 
+@composite
+def unique_strs(draw: DrawFn, /) -> str:
+    """Strategy for generating unique strings."""
+    now = get_now_local()
+    pid = getpid()
+    ident = get_ident()
+    key = str(draw(uuids())).replace("-", "")
+    return f"{now:%Y%m%d%H%M%S%f}_{pid}_{ident}_{key}"
+
+
+##
+
+
 @composite
 def versions(draw: DrawFn, /, *, suffix: MaybeSearchStrategy[bool] = False) -> Version:
     """Strategy for generating versions."""
@@ -1434,34 +1435,6 @@ def versions(draw: DrawFn, /, *, suffix: MaybeSearchStrategy[bool] = False) -> V
 ##
 
 
-def yield_test_redis(data: DataObject, /) -> AbstractAsyncContextManager[_TestRedis]:
-    """Strategy for generating test redis clients."""
-    from redis.exceptions import ResponseError  # skipif-ci-and-not-linux
-
-    from utilities.redis import _TestRedis, yield_redis  #  skipif-ci-and-not-linux
-    from utilities.tzlocal import get_now_local  #  skipif-ci-and-not-linux
-
-    now = get_now_local()  # skipif-ci-and-not-linux
-    uuid = data.draw(uuids())  # skipif-ci-and-not-linux
-    key = f"{now}_{uuid}"  # skipif-ci-and-not-linux
-
-    @asynccontextmanager
-    async def func() -> AsyncIterator[_TestRedis]:  # skipif-ci-and-not-linux
-        async with yield_redis(db=15) as redis:  # skipif-ci-and-not-linux
-            keys = cast("list[KeyT]", await redis.keys(pattern=f"{key}_*"))
-            with suppress(ResponseError):
-                _ = await redis.delete(*keys)
-            yield _TestRedis(redis=redis, timestamp=now, uuid=uuid, key=key)
-            keys = cast("list[KeyT]", await redis.keys(pattern=f"{key}_*"))
-            with suppress(ResponseError):
-                _ = await redis.delete(*keys)
-
-    return func()  # skipif-ci-and-not-linux
-
-
-##
-
-
 @composite
 def zoned_datetimes(
     draw: DrawFn,
@@ -1578,7 +1551,7 @@ __all__ = [
     "triples",
     "uint32s",
     "uint64s",
+    "unique_strs",
     "versions",
-    "yield_test_redis",
     "zoned_datetimes",
 ]

utilities/libcst.py

@@ -156,7 +156,7 @@ def render_module(source: str | Module, /) -> str:
         case str() as text:
             try:
                 return check_output(["ruff", "format", "-"], input=text, text=True)
-            except CalledProcessError:
+            except CalledProcessError:  # pragma: no cover
                 return text
         case Module() as module:
             return render_module(module.code)

utilities/pottery.py

@@ -0,0 +1,50 @@
+from __future__ import annotations
+
+import asyncio
+from contextlib import asynccontextmanager, suppress
+from typing import TYPE_CHECKING
+
+from pottery import AIORedlock
+from pottery.exceptions import ReleaseUnlockedLock
+from redis.asyncio import Redis
+
+from utilities.datetime import MILLISECOND, SECOND, datetime_duration_to_float
+from utilities.iterables import always_iterable
+
+if TYPE_CHECKING:
+    from collections.abc import AsyncIterator
+
+    from utilities.types import Duration, MaybeIterable
+
+
+@asynccontextmanager
+async def yield_locked_resource(
+    redis: MaybeIterable[Redis],
+    key: str,
+    /,
+    *,
+    duration: Duration = 10 * SECOND,
+    sleep: Duration = MILLISECOND,
+) -> AsyncIterator[None]:
+    """Yield a locked resource."""
+    masters = (  # skipif-ci-and-not-linux
+        {redis} if isinstance(redis, Redis) else set(always_iterable(redis))
+    )
+    duration_use = datetime_duration_to_float(duration)  # skipif-ci-and-not-linux
+    lock = AIORedlock(  # skipif-ci-and-not-linux
+        key=key,
+        masters=masters,
+        auto_release_time=duration_use,
+        context_manager_timeout=duration_use,
+    )
+    sleep_use = datetime_duration_to_float(sleep)  # skipif-ci-and-not-linux
+    while not await lock.acquire():  # pragma: no cover
+        _ = await asyncio.sleep(sleep_use)
+    try:  # skipif-ci-and-not-linux
+        yield
+    finally:  # skipif-ci-and-not-linux
+        with suppress(ReleaseUnlockedLock):
+            await lock.release()
+
+
+__all__ = ["yield_locked_resource"]

utilities/psutil.py

@@ -31,7 +31,6 @@ class MemoryMonitorService(Looper[None]):
     console: str | None = field(default=None, repr=False)
     path: PathLike = "memory.txt"
     _console: Logger | None = field(init=False, repr=False)
-    _max_age: int | None = field(default=None, init=False, repr=False)
     _path: Path = field(init=False, repr=False)
 
     @override
@@ -40,6 +39,7 @@ class MemoryMonitorService(Looper[None]):
         if self.console is not None:
             self._console = getLogger(self.console)
         self._path = Path(self.path)
+        self._path.parent.mkdir(parents=True, exist_ok=True)
 
     @override
     async def core(self) -> None:

utilities/redis.py

@@ -21,7 +21,6 @@ from typing import (
     overload,
     override,
 )
-from uuid import UUID, uuid4
 
 from redis.asyncio import Redis
 from redis.typing import EncodableT
@@ -32,14 +31,12 @@ from utilities.datetime import (
     SECOND,
     datetime_duration_to_float,
     datetime_duration_to_timedelta,
-    get_now,
 )
 from utilities.errors import ImpossibleCaseError
 from utilities.functions import ensure_int, get_class_name, identity
 from utilities.iterables import always_iterable, one
 
 if TYPE_CHECKING:
-    import datetime as dt
     from collections.abc import (
         AsyncIterator,
         Awaitable,
@@ -984,22 +981,6 @@ def _deserialize(
     return deserializer_use(data)  # skipif-ci-and-not-linux
 
 
-##
-
-
-@dataclass(repr=False, kw_only=True, slots=True)
-class _TestRedis:
-    """A container for a redis client; for testing purposes only."""
-
-    redis: Redis
-    timestamp: dt.datetime = field(default_factory=get_now)
-    uuid: UUID = field(default_factory=uuid4)
-    key: str
-
-
-_ = _TestRedis
-
-
 __all__ = [
     "PublishService",
     "Publisher",