dycw-utilities 0.131.16__py3-none-any.whl → 0.131.18__py3-none-any.whl

{dycw_utilities-0.131.16.dist-info → dycw_utilities-0.131.18.dist-info}/METADATA

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

{dycw_utilities-0.131.16.dist-info → dycw_utilities-0.131.18.dist-info}/RECORD

@@ -1,7 +1,7 @@
-utilities/__init__.py,sha256=I-o_EPHIejt3ggoklYx3Vx_Wi2XLiPZ3YlzjyX4JBKI,61
+utilities/__init__.py,sha256=653kI0IdCJCyIvZc8ekfN8j8dOnjrRtGJKRx9UDXVvM,61
 utilities/aiolimiter.py,sha256=mD0wEiqMgwpty4XTbawFpnkkmJS6R4JRsVXFUaoitSU,628
 utilities/altair.py,sha256=HeZBVUocjkrTNwwKrClppsIqgNFF-ykv05HfZSoHYno,9104
-utilities/asyncio.py,sha256=yfKvAIDCRrWdyQMVZMo4DJQx4nVrXoAcqwhNuF95Ryo,38186
+utilities/asyncio.py,sha256=mHnlSA4KPeDaBRts8Rn4sNA_4urodj7gQzs-_8Z-F7A,37587
 utilities/atomicwrites.py,sha256=geFjn9Pwn-tTrtoGjDDxWli9NqbYfy3gGL6ZBctiqSo,5393
 utilities/atools.py,sha256=-bFGIrwYMFR7xl39j02DZMsO_u5x5_Ph7bRlBUFVYyw,1048
 utilities/cachetools.py,sha256=uBtEv4hD-TuCPX_cQy1lOpLF-QqfwnYGSf0o4Soqydc,2826
@@ -12,7 +12,7 @@ utilities/contextvars.py,sha256=RsSGGrbQqqZ67rOydnM7WWIsM2lIE31UHJLejnHJPWY,505
 utilities/cryptography.py,sha256=_CiK_K6c_-uQuUhsUNjNjTL-nqxAh4_1zTfS11Xe120,972
 utilities/cvxpy.py,sha256=Rv1-fD-XYerosCavRF8Pohop2DBkU3AlFaGTfD8AEAA,13776
 utilities/dataclasses.py,sha256=iiC1wpGXWhaocIikzwBt8bbLWyImoUlOlcDZJGejaIg,33011
-utilities/datetime.py,sha256=uPQdUgJJ9KuF-pogjYRbI9lOK-i5UBzPHexWe4pOVEo,38713
+utilities/datetime.py,sha256=bt9tgxHfjSbl0g5H7W2YSwNm-1HsbKjX5AsYcyEO1mU,30361
 utilities/enum.py,sha256=HoRwVCWzsnH0vpO9ZEcAAIZLMv0Sn2vJxxA4sYMQgDs,5793
 utilities/errors.py,sha256=nC7ZYtxxDBMfrTHtT_MByBfup_wfGQFRo3eDt-0ZPe8,1045
 utilities/eventkit.py,sha256=6M5Xu1SzN-juk9PqBHwy5dS-ta7T0qA6SMpDsakOJ0E,13039
@@ -28,7 +28,7 @@ utilities/hypothesis.py,sha256=fcklSUIagoDZOTn8ITbkIU7BOZMV4g1Owj0khiz9yBY,43401
 utilities/importlib.py,sha256=mV1xT_O_zt_GnZZ36tl3xOmMaN_3jErDWY54fX39F6Y,429
 utilities/inflect.py,sha256=DbqB5Q9FbRGJ1NbvEiZBirRMxCxgrz91zy5jCO9ZIs0,347
 utilities/ipython.py,sha256=V2oMYHvEKvlNBzxDXdLvKi48oUq2SclRg5xasjaXStw,763
-utilities/iterables.py,sha256=mDqw2_0MUVp-P8FklgcaVTi2TXduH0MxbhTDzzhSBho,44915
+utilities/iterables.py,sha256=cuebB4ivKlZuKm8S3PQIfjavn9h-5mBGmvYq4FpSxFg,43812
 utilities/jupyter.py,sha256=ft5JA7fBxXKzP-L9W8f2-wbF0QeYc_2uLQNFDVk4Z-M,2917
 utilities/libcst.py,sha256=Jto5ppzRzsxn4AD32IS8n0lbgLYXwsVJB6EY8giNZyY,4974
 utilities/lightweight_charts.py,sha256=JrkrAZMo6JID2Eoc9QCc05Y_pK4l2zsApIhmii1z2Ig,2764
@@ -39,20 +39,20 @@ utilities/memory_profiler.py,sha256=tf2C51P2lCujPGvRt2Rfc7VEw5LDXmVPCG3z_AvBmbU,
 utilities/modules.py,sha256=iuvLluJya-hvl1Q25-Jk3dLgx2Es3ck4SjJiEkAlVTs,3195
 utilities/more_itertools.py,sha256=tBbjjKx8_Uv_TCjxhPwrGfAx_jRHtvLIZqXVWAsjzqA,8842
 utilities/numpy.py,sha256=Xn23sA2ZbVNqwUYEgNJD3XBYH6IbCri_WkHSNhg3NkY,26122
-utilities/operator.py,sha256=0M2yZJ0PODH47ogFEnkGMBe_cfxwZR02T_92LZVZvHo,3715
+utilities/operator.py,sha256=DuiWdkmK0D-ddvFqOayDkazTQE1Ysvtl6-UiBN5gns8,3857
 utilities/optuna.py,sha256=loyJGWTzljgdJaoLhP09PT8Jz6o_pwBOwehY33lHkhw,1923
-utilities/orjson.py,sha256=lhBSO-QCm2g-uhUE0hKe-PVhAt1TKRRx0iZLSFPmew4,36970
+utilities/orjson.py,sha256=7CraMEtaJtKTDTTVS4fQU6PeuFcG0V8SGwD3sqZ8kSI,36475
 utilities/os.py,sha256=D_FyyT-6TtqiN9KSS7c9g1fnUtgxmyMtzAjmYLkk46A,3587
-utilities/parse.py,sha256=vsZ2jf_ceSI_Kta9titixufysJaVXh0Whjz1T4awJZw,18938
+utilities/parse.py,sha256=YE2VWYKDm9WYf5wcOWrOxVfM6UWvhkSxSkwdxRQNsA0,17633
 utilities/pathlib.py,sha256=PK41rf1c9Wqv7h8f5R7H3_Lhq_gQZTUJD5tu3gMHVaU,3247
 utilities/period.py,sha256=opqpBevBGSGXbA7NYfRJjtthi1JPxdMaZ7QV3xosnTc,4774
 utilities/pickle.py,sha256=MBT2xZCsv0pH868IXLGKnlcqNx2IRVKYNpRcqiQQqxw,653
 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=-YikdIQFqhFXifyMq5R5z6V6N8y639gAnVVXR4jXxdc,3568
+utilities/pottery.py,sha256=2w3YuoH1KmLaCVqkwSghHTOT8S4xiUskwRHSRqrUEQY,3430
 utilities/pqdm.py,sha256=foRytQybmOQ05pjt5LF7ANyzrIa--4ScDE3T2wd31a4,3118
-utilities/psutil.py,sha256=RtbLKOoIJhqrJmEoHDBVeSD-KPzshtS0FtRXBP9_w2s,3751
+utilities/psutil.py,sha256=ZkwBGfTqAv8hxCyJVVJFctIVQvxhYuB28re3mIPBodc,3761
 utilities/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 utilities/pydantic.py,sha256=aP6OKowg2Md4rgQuQ5qTSF4bTbBuq7WtRb7zS3JSRGY,1841
 utilities/pyinstrument.py,sha256=3MPNDbZW_1Aj1aA1_f-yqPStgmjIFxPwIafYq2dsaTs,853
@@ -62,20 +62,19 @@ utilities/pytest_regressions.py,sha256=YI55B7EtLjhz7zPJZ6NK9bWrxrKCKabWZJe1cwcbA
 utilities/python_dotenv.py,sha256=edXsvHZhZnYeqfMfrsRRpj7_9eJI6uizh3xLx8Q9B3w,3228
 utilities/random.py,sha256=lYdjgxB7GCfU_fwFVl5U-BIM_HV3q6_urL9byjrwDM8,4157
 utilities/re.py,sha256=6qxeV0rQZaBDKWcB7apSBmxtg_XzoGY-EdegTkMn-ZY,4578
-utilities/redis.py,sha256=IceT5EjgrebVkGL8X3M35xlqjI2c7zFbyV1P4dExN4M,36037
+utilities/redis.py,sha256=oA70YoFnSaSqTBzMHrbHkkGIBWU69-97MQEwH_0iX3g,35762
 utilities/reprlib.py,sha256=ssYTcBW-TeRh3fhCJv57sopTZHF5FrPyyUg9yp5XBlo,3953
 utilities/scipy.py,sha256=wZJM7fEgBAkLSYYvSmsg5ac-QuwAI0BGqHVetw1_Hb0,947
 utilities/sentinel.py,sha256=3jIwgpMekWgDAxPDA_hXMP2St43cPhciKN3LWiZ7kv0,1248
 utilities/shelve.py,sha256=HZsMwK4tcIfg3sh0gApx4-yjQnrY4o3V3ZRimvRhoW0,738
-utilities/slack_sdk.py,sha256=ltmzv68aa73CJGqTDvt8L9vDm22YU9iOCo3NCiNd3vA,4347
+utilities/slack_sdk.py,sha256=ZpPzV_oHpSJbontGo2_YVVjKH08PjLjsx5Me-nLy918,4246
 utilities/socket.py,sha256=K77vfREvzoVTrpYKo6MZakol0EYu2q1sWJnnZqL0So0,118
-utilities/sqlalchemy.py,sha256=IuQ7CIVNl29TG6i81K6fam8NmTmPjtA6OiIN4nIM9W8,37935
-utilities/sqlalchemy_polars.py,sha256=hApbjQUY-XgKfAXcun8gDP2lGh5LxrudnCpbG_hrYa0,14968
+utilities/sqlalchemy.py,sha256=sTlxK-Is-XCj7srTRh3Nz68axzIdDYZWLQdTeXnOdkQ,38015
+utilities/sqlalchemy_polars.py,sha256=bDiKqHxOWu0Dj4ZDuGcVgR7ulm7sB90iVNINAKaeaKc,14290
 utilities/statsmodels.py,sha256=koyiBHvpMcSiBfh99wFUfSggLNx7cuAw3rwyfAhoKpQ,3410
 utilities/streamlit.py,sha256=U9PJBaKP1IdSykKhPZhIzSPTZsmLsnwbEPZWzNhJPKk,2955
 utilities/string.py,sha256=XmU-s04qIV_tODnKl2pQiwmHaxzgOqRKU-RyzdrfvSE,620
 utilities/tempfile.py,sha256=VqmZJAhTJ1OaVywFzk5eqROV8iJbW9XQ_QYAV0bpdRo,1384
-utilities/tenacity.py,sha256=1PUvODiBVgeqIh7G5TRt5WWMSqjLYkEqP53itT97WQc,4914
 utilities/text.py,sha256=ymBFlP_cA8OgNnZRVNs7FAh7OG8HxE6YkiLEMZv5g_A,11297
 utilities/threading.py,sha256=GvBOp4CyhHfN90wGXZuA2VKe9fGzMaEa7oCl4f3nnPU,1009
 utilities/timer.py,sha256=VeSl3ot8-f4D1d3HjjSsgKvjxHJGXd_sW4KcTExOR64,2475
@@ -88,11 +87,11 @@ utilities/tzlocal.py,sha256=xbBBzVIUKMk8AkhuIp1qxGRNBioIa5I09dpeoBnIOOU,662
 utilities/uuid.py,sha256=jJTFxz-CWgltqNuzmythB7iEQ-Q1mCwPevUfKthZT3c,611
 utilities/version.py,sha256=ufhJMmI6KPs1-3wBI71aj5wCukd3sP_m11usLe88DNA,5117
 utilities/warnings.py,sha256=un1LvHv70PU-LLv8RxPVmugTzDJkkGXRMZTE2-fTQHw,1771
-utilities/whenever.py,sha256=oO0sgIIv4tvhIYmlZ4RFFfY6P54CPeyJCY9A4XHuyqo,11916
+utilities/whenever.py,sha256=tUNfGB_72YSCxbDKiQ7ulWtISC3AgVBrW1s3XlFPkuw,6883
 utilities/whenever2.py,sha256=Jd_fEavXCWTdMj29L7j-HbeMvrl8e3Ah_nxcLEFAwOU,12026
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=gJPr9l7V8s3Y7TXpCGYEM1S81Rplb9e4MoV9Nvy2VU8,1852
-dycw_utilities-0.131.16.dist-info/METADATA,sha256=X0Yt3BUVE7Y7W884VYvx_9CVhE4c9ENSxbhD4ghFoKU,1585
-dycw_utilities-0.131.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.131.16.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.131.16.dist-info/RECORD,,
+dycw_utilities-0.131.18.dist-info/METADATA,sha256=nfjJB5vwmQGGegP4Ma-fxV0cKOE67JbPGxKLmZ0Rd78,1585
+dycw_utilities-0.131.18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.131.18.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.131.18.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.131.16"
+__version__ = "0.131.18"

utilities/asyncio.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import asyncio
 from asyncio import (
     Event,
     Lock,
@@ -14,7 +15,6 @@ from asyncio import (
     create_subprocess_shell,
     create_task,
     sleep,
-    timeout,
 )
 from collections.abc import Callable, Iterable, Iterator
 from contextlib import (
@@ -43,28 +43,22 @@ from typing import (
 )
 
 from typing_extensions import deprecated
-from whenever import TimeDelta
 
 from utilities.dataclasses import replace_non_sentinel
-from utilities.datetime import (
-    SECOND,
-    datetime_duration_to_float,
-    get_now,
-    round_datetime,
-)
 from utilities.errors import repr_error
 from utilities.functions import ensure_int, ensure_not_none
 from utilities.random import SYSTEM_RANDOM
 from utilities.sentinel import Sentinel, sentinel
 from utilities.types import (
+    DateTimeRoundUnit,
     MaybeCallableEvent,
     MaybeType,
     THashable,
     TSupportsRichComparison,
 )
+from utilities.whenever2 import SECOND, get_now
 
 if TYPE_CHECKING:
-    import datetime as dt
     from asyncio import _CoroutineLike
     from asyncio.subprocess import Process
     from collections import deque
@@ -73,7 +67,7 @@ if TYPE_CHECKING:
     from random import Random
     from types import TracebackType
 
-    from utilities.types import Duration
+    from whenever import TimeDelta, ZonedDateTime
 
 
 _T = TypeVar("_T")
@@ -244,7 +238,7 @@ class EnhancedTaskGroup(TaskGroup):
     """Task group with enhanced features."""
 
     _semaphore: Semaphore | None
-    _timeout: Duration | None
+    _timeout: TimeDelta | None
     _error: MaybeType[BaseException]
     _stack: AsyncExitStack
     _timeout_cm: _AsyncGeneratorContextManager[None] | None
@@ -254,7 +248,7 @@ class EnhancedTaskGroup(TaskGroup):
         self,
         *,
         max_tasks: int | None = None,
-        timeout: Duration | None = None,
+        timeout: TimeDelta | None = None,
         error: MaybeType[BaseException] = TimeoutError,
     ) -> None:
         super().__init__()
@@ -306,7 +300,7 @@ class EnhancedTaskGroup(TaskGroup):
             return await coroutine
 
     async def _wrap_with_timeout(self, coroutine: _CoroutineLike[_T], /) -> _T:
-        async with timeout_dur(duration=self._timeout, error=self._error):
+        async with timeout_td(self._timeout, error=self._error):
             return await coroutine
 
 
@@ -331,15 +325,13 @@ class Looper(Generic[_T]):
     """A looper of a core coroutine, handling errors."""
 
     auto_start: bool = field(default=False, repr=False)
-    freq: Duration = field(default=SECOND, repr=False)
-    backoff: Duration = field(default=10 * SECOND, repr=False)
+    freq: TimeDelta = field(default=SECOND, repr=False)
+    backoff: TimeDelta = field(default=10 * SECOND, repr=False)
     empty_upon_exit: bool = field(default=False, repr=False)
     logger: str | None = field(default=None, repr=False)
-    timeout: Duration | None = field(default=None, repr=False)
+    timeout: TimeDelta | None = field(default=None, repr=False)
     # settings
-    _backoff: float = field(init=False, repr=False)
     _debug: bool = field(default=False, repr=False)
-    _freq: float = field(init=False, repr=False)
     # counts
     _entries: int = field(default=0, init=False, repr=False)
     _core_attempts: int = field(default=0, init=False, repr=False)
@@ -379,8 +371,6 @@ class Looper(Generic[_T]):
     _task: Task[None] | None = field(default=None, init=False, repr=False, hash=False)
 
     def __post_init__(self) -> None:
-        self._backoff = datetime_duration_to_float(self.backoff)
-        self._freq = datetime_duration_to_float(self.freq)
         self._logger = getLogger(name=self.logger)
         self._logger.setLevel(DEBUG)
 
@@ -448,7 +438,7 @@ class Looper(Generic[_T]):
 
     async def _apply_back_off(self) -> None:
         """Apply a back off period."""
-        await sleep(self._backoff)
+        await sleep_td(self.backoff)
         self._is_pending_back_off.clear()
 
     async def core(self) -> None:
@@ -541,10 +531,10 @@ class Looper(Generic[_T]):
         *,
         auto_start: bool | Sentinel = sentinel,
         empty_upon_exit: bool | Sentinel = sentinel,
-        freq: Duration | Sentinel = sentinel,
-        backoff: Duration | Sentinel = sentinel,
+        freq: TimeDelta | Sentinel = sentinel,
+        backoff: TimeDelta | Sentinel = sentinel,
         logger: str | None | Sentinel = sentinel,
-        timeout: Duration | None | Sentinel = sentinel,
+        timeout: TimeDelta | None | Sentinel = sentinel,
         _debug: bool | Sentinel = sentinel,
         **kwargs: Any,
     ) -> Self:
@@ -670,7 +660,7 @@ class Looper(Generic[_T]):
     async def run_looper(self) -> None:
         """Run the looper."""
         try:
-            async with timeout_dur(duration=self.timeout):
+            async with timeout_td(self.timeout):
                 while True:
                     if self._is_stopped.is_set():
                         _ = self._debug and self._logger.debug("%s: stopped", self)
@@ -705,7 +695,7 @@ class Looper(Generic[_T]):
                         else:
                             async with self._lock:
                                 self._core_successes += 1
-                            await sleep(self._freq)
+                            await sleep_td(self.freq)
         except RuntimeError as error:  # pragma: no cover
             if error.args[0] == "generator didn't stop after athrow()":
                 return
@@ -718,7 +708,7 @@ class Looper(Generic[_T]):
         while not self.empty():
             await self.core()
             if not self.empty():
-                await sleep(self._freq)
+                await sleep_td(self.freq)
 
     @property
     def stats(self) -> _LooperStats:
@@ -962,47 +952,41 @@ def put_items_nowait(items: Iterable[_T], queue: Queue[_T], /) -> None:
 ##
 
 
-async def sleep_dur(*, duration: Duration | TimeDelta | None = None) -> None:
-    """Sleep which accepts durations."""
-    if duration is None:
+async def sleep_max(
+    sleep: TimeDelta | None = None, /, *, random: Random = SYSTEM_RANDOM
+) -> None:
+    """Sleep which accepts deltas."""
+    if sleep is None:
         return
-    if isinstance(duration, TimeDelta):
-        await sleep(duration.in_seconds())
-    else:
-        await sleep(datetime_duration_to_float(duration))
+    await asyncio.sleep(random.uniform(0.0, sleep.in_seconds()))
 
 
 ##
 
 
-async def sleep_max_dur(
-    *, duration: Duration | None = None, random: Random = SYSTEM_RANDOM
+async def sleep_rounded(
+    *, unit: DateTimeRoundUnit = "second", increment: int = 1
 ) -> None:
-    """Sleep which accepts max durations."""
-    if duration is None:
-        return
-    await sleep(random.uniform(0.0, datetime_duration_to_float(duration)))
+    """Sleep until a rounded time."""
+    await sleep_until(get_now().round(unit, increment=increment, mode="ceil"))
 
 
 ##
 
 
-async def sleep_until(datetime: dt.datetime, /) -> None:
-    """Sleep until a given time."""
-    await sleep_dur(duration=datetime - get_now())
+async def sleep_td(delta: TimeDelta | None = None, /) -> None:
+    """Sleep which accepts deltas."""
+    if delta is None:
+        return
+    await sleep(delta.in_seconds())
 
 
 ##
 
 
-async def sleep_until_rounded(
-    duration: Duration, /, *, rel_tol: float | None = None, abs_tol: float | None = None
-) -> None:
-    """Sleep until a rounded time; accepts durations."""
-    datetime = round_datetime(
-        get_now(), duration, mode="ceil", rel_tol=rel_tol, abs_tol=abs_tol
-    )
-    await sleep_until(datetime)
+async def sleep_until(datetime: ZonedDateTime, /) -> None:
+    """Sleep until a given time."""
+    await sleep_td(datetime - get_now())
 
 
 ##
@@ -1059,13 +1043,16 @@ async def _stream_one(
 
 
 @asynccontextmanager
-async def timeout_dur(
-    *, duration: Duration | None = None, error: MaybeType[BaseException] = TimeoutError
+async def timeout_td(
+    timeout: TimeDelta | None = None,
+    /,
+    *,
+    error: MaybeType[BaseException] = TimeoutError,
 ) -> AsyncIterator[None]:
-    """Timeout context manager which accepts durations."""
-    delay = None if duration is None else datetime_duration_to_float(duration)
+    """Timeout context manager which accepts deltas."""
+    timeout_use = None if timeout is None else timeout.in_seconds()
     try:
-        async with timeout(delay):
+        async with asyncio.timeout(timeout_use):
             yield
     except TimeoutError:
         raise error from None
@@ -1084,10 +1071,10 @@ __all__ = [
     "get_items_nowait",
     "put_items",
     "put_items_nowait",
-    "sleep_dur",
-    "sleep_max_dur",
+    "sleep_max",
+    "sleep_rounded",
+    "sleep_td",
     "sleep_until",
-    "sleep_until_rounded",
     "stream_command",
-    "timeout_dur",
+    "timeout_td",
 ]