dycw-utilities 0.119.0__py3-none-any.whl → 0.121.0__py3-none-any.whl

{dycw_utilities-0.119.0.dist-info → dycw_utilities-0.121.0.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.119.0
+Version: 0.121.0
 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.19; extra == 'test'
+Requires-Dist: hypothesis<6.132,>=6.131.20; 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'
@@ -80,7 +80,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.19; extra == 'zzz-test-hypothesis'
+Requires-Dist: hypothesis<6.132,>=6.131.20; 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.119.0.dist-info → dycw_utilities-0.121.0.dist-info}/RECORD

@@ -1,7 +1,7 @@
-utilities/__init__.py,sha256=ayFWZ1qOKB0Lqd63WRIXSBg4vZDal1xvPPVJr5LEPEA,60
+utilities/__init__.py,sha256=FKqdZTwX6QRMMK9ehPxm3uM8OyIJA4Mgay0d8pgY6Nw,60
 utilities/altair.py,sha256=Gpja-flOo-Db0PIPJLJsgzAlXWoKUjPU1qY-DQ829ek,9156
 utilities/astor.py,sha256=xuDUkjq0-b6fhtwjhbnebzbqQZAjMSHR1IIS5uOodVg,777
-utilities/asyncio.py,sha256=HGX79AKzpQbbDBW3paxAXrhWYeudcJjiO1ETU40d_-8,18463
+utilities/asyncio.py,sha256=gn8dxBsNzFfFzrNpImF731LAmgLwEQM3uiwKuUPlg08,18822
 utilities/atomicwrites.py,sha256=geFjn9Pwn-tTrtoGjDDxWli9NqbYfy3gGL6ZBctiqSo,5393
 utilities/atools.py,sha256=IYMuFSFGSKyuQmqD6v5IUtDlz8PPw0Sr87Cub_gRU3M,1168
 utilities/cachetools.py,sha256=C1zqOg7BYz0IfQFK8e3qaDDgEZxDpo47F15RTfJM37Q,2910
@@ -59,15 +59,15 @@ 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=OHw3J2dBA5QssDluKXAG1zIAK2mJJTd6uBuf_1YQuAE,26646
+utilities/redis.py,sha256=i5G3k-EIi2JSf9arLIu0YW6viGsiYtxAoynbb4KvJek,26628
 utilities/reprlib.py,sha256=Re9bk3n-kC__9DxQmRlevqFA86pE6TtVfWjUgpbVOv0,1849
 utilities/rich.py,sha256=t50MwwVBsoOLxzmeVFSVpjno4OW6Ufum32skXbV8-Bs,1911
 utilities/scipy.py,sha256=X6ROnHwiUhAmPhM0jkfEh0-Fd9iRvwiqtCQMOLmOQF8,945
 utilities/sentinel.py,sha256=3jIwgpMekWgDAxPDA_hXMP2St43cPhciKN3LWiZ7kv0,1248
 utilities/shelve.py,sha256=HZsMwK4tcIfg3sh0gApx4-yjQnrY4o3V3ZRimvRhoW0,738
-utilities/slack_sdk.py,sha256=NLHmWYK6wc5bz4CGImugXceaToasNBLSqA5sd5ld2r4,3307
+utilities/slack_sdk.py,sha256=A2f7-DYOngRoUP6ZdLIaUQ6Lfzgru5Xp3U3k5JfEkQE,3301
 utilities/socket.py,sha256=K77vfREvzoVTrpYKo6MZakol0EYu2q1sWJnnZqL0So0,118
-utilities/sqlalchemy.py,sha256=09stMwvmI68zlk-DSy9GDk5_YxcMddLh87RPC8Bs4yY,35469
+utilities/sqlalchemy.py,sha256=iwNJyR0Dx47l81aHEq5P6eMzgduVxxGH531OVUp2k_A,35451
 utilities/sqlalchemy_polars.py,sha256=wjJpoUo-yO9E2ujpG_06vV5r2OdvBiQ4yvV6wKCa2Tk,15605
 utilities/statsmodels.py,sha256=koyiBHvpMcSiBfh99wFUfSggLNx7cuAw3rwyfAhoKpQ,3410
 utilities/streamlit.py,sha256=U9PJBaKP1IdSykKhPZhIzSPTZsmLsnwbEPZWzNhJPKk,2955
@@ -78,7 +78,7 @@ utilities/text.py,sha256=Fo12N4aA7k2rnb4W4vH9iiDh88Q5_nvRssTkfNsvVM8,10965
 utilities/threading.py,sha256=GvBOp4CyhHfN90wGXZuA2VKe9fGzMaEa7oCl4f3nnPU,1009
 utilities/timer.py,sha256=Rkc49KSpHuC8s7vUxGO9DU55U9I6yDKnchsQqrUCVBs,4075
 utilities/traceback.py,sha256=p9WATV-e4_5AW6SvyRBiU-MY8XnEFcKgrFNUoFzalXI,27521
-utilities/types.py,sha256=FhE1b8v_s_IlmXucwY7jAMWAq9cfpzyKssQwgwb3jnM,18267
+utilities/types.py,sha256=2f1DqTZTMRlpCPWPd9-rh_uwmRPv9UdBoi_Bfv7Ccmo,18374
 utilities/typing.py,sha256=H6ysJkI830aRwLsMKz0SZIw4cpcsm7d6KhQOwr-SDh0,13817
 utilities/tzdata.py,sha256=yCf70NICwAeazN3_JcXhWvRqCy06XJNQ42j7r6gw3HY,1217
 utilities/tzlocal.py,sha256=3upDNFBvGh1l9njmLR2z2S6K6VxQSb7QizYGUbAH3JU,960
@@ -88,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.119.0.dist-info/METADATA,sha256=XG8mqNyIqDEemm7CPwNSAdYJS_z_rsP8SqlYoT-DEAY,12943
-dycw_utilities-0.119.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.119.0.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.119.0.dist-info/RECORD,,
+dycw_utilities-0.121.0.dist-info/METADATA,sha256=YvumB9cX9CE9qNSb60yruXQ-NlHPyugeA91GzOJLPUM,12943
+dycw_utilities-0.121.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.121.0.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.121.0.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.119.0"
+__version__ = "0.121.0"

utilities/asyncio.py

@@ -26,7 +26,6 @@ from typing import (
     TYPE_CHECKING,
     Any,
     Generic,
-    Literal,
     NoReturn,
     TextIO,
     TypeVar,
@@ -49,6 +48,7 @@ from utilities.reprlib import get_repr
 from utilities.sentinel import Sentinel, sentinel
 from utilities.types import (
     Coroutine1,
+    DurationOrEveryDuration,
     MaybeCallableEvent,
     MaybeType,
     THashable,
@@ -121,17 +121,14 @@ class EnhancedTaskGroup(TaskGroup):
 ##
 
 
-type _DurationOrEvery = Duration | tuple[Literal["every"], Duration]
-
-
 @dataclass(kw_only=True, unsafe_hash=True)
 class InfiniteLooper(ABC, Generic[THashable]):
     """An infinite loop which can throw exceptions by setting events."""
 
-    sleep_core: _DurationOrEvery = SECOND
-    sleep_restart: _DurationOrEvery = MINUTE
+    sleep_core: DurationOrEveryDuration = SECOND
+    sleep_restart: DurationOrEveryDuration = MINUTE
     logger: str | None = None
-    _events: Mapping[THashable, Event] = field(
+    _events: Mapping[THashable | None, Event] = field(
         default_factory=dict, init=False, repr=False, hash=False
     )
 
@@ -229,7 +226,7 @@ class InfiniteLooper(ABC, Generic[THashable]):
             msgs.append(f"Sleeping {self._sleep_restart_desc}...")
             getLogger(name=self.logger).error("\n".join(msgs))
 
-    def _raise_error(self, event: THashable, /) -> NoReturn:
+    def _raise_error(self, event: THashable | None, /) -> NoReturn:
         """Raise the error corresponding to given event."""
         mapping = dict(self._yield_events_and_exceptions())
         error = mapping.get(event, InfiniteLooperError)
@@ -241,7 +238,7 @@ class InfiniteLooper(ABC, Generic[THashable]):
             event: Event() for event, _ in self._yield_events_and_exceptions()
         }
 
-    async def _run_sleep(self, sleep: _DurationOrEvery, /) -> None:
+    async def _run_sleep(self, sleep: DurationOrEveryDuration, /) -> None:
         """Sleep until the next part of the loop."""
         match sleep:
             case int() | float() | dt.timedelta() as duration:
@@ -264,12 +261,12 @@ class InfiniteLooper(ABC, Generic[THashable]):
             case _ as never:
                 assert_never(never)
 
-    def _set_event(self, event: THashable, /) -> None:
+    def _set_event(self, *, event: THashable | None = None) -> None:
         """Set the given event."""
         try:
             event_obj = self._events[event]
         except KeyError:
-            raise InfiniteLooperError(looper=self, event=event) from None
+            raise _InfiniteLooperNoSuchEventError(looper=self, event=event) from None
         event_obj.set()
 
     def _yield_coroutines(self) -> Iterator[Callable[[], Coroutine1[None]]]:
@@ -278,14 +275,18 @@ class InfiniteLooper(ABC, Generic[THashable]):
 
     def _yield_events_and_exceptions(
         self,
-    ) -> Iterator[tuple[THashable, MaybeType[BaseException]]]:
+    ) -> Iterator[tuple[THashable | None, MaybeType[BaseException]]]:
         """Yield the events & exceptions."""
-        yield from []
+        yield (None, _InfiniteLooperDefaultEventError)
 
 
 @dataclass(kw_only=True, slots=True)
 class InfiniteLooperError(Exception):
     looper: InfiniteLooper[Any]
+
+
+@dataclass(kw_only=True, slots=True)
+class _InfiniteLooperNoSuchEventError(InfiniteLooperError):
     event: Hashable
 
     @override
@@ -293,6 +294,13 @@ class InfiniteLooperError(Exception):
         return f"{get_class_name(self.looper)!r} does not have an event {self.event!r}"
 
 
+@dataclass(kw_only=True, slots=True)
+class _InfiniteLooperDefaultEventError(InfiniteLooperError):
+    @override
+    def __str__(self) -> str:
+        return f"{get_class_name(self.looper)!r} default event error"
+
+
 ##
 
 

utilities/redis.py

@@ -589,7 +589,7 @@ async def publish(
 
 
 @dataclass(kw_only=True)
-class PublisherIQL(InfiniteQueueLooper[None, tuple[str, EncodableT]]):
+class Publisher(InfiniteQueueLooper[None, tuple[str, EncodableT]]):
     """Publish a set of messages to Redis."""
 
     redis: Redis
@@ -612,12 +612,12 @@ class PublisherIQL(InfiniteQueueLooper[None, tuple[str, EncodableT]]):
     def _yield_events_and_exceptions(
         self,
     ) -> Iterator[tuple[None, MaybeType[BaseException]]]:
-        yield (None, PublisherIQLError)  # skipif-ci-and-not-linux
+        yield (None, PublisherError)  # skipif-ci-and-not-linux
 
 
 @dataclass(kw_only=True)
-class PublisherIQLError(Exception):
-    publisher: PublisherIQL
+class PublisherError(Exception):
+    publisher: Publisher
 
     @override
     def __str__(self) -> str:
@@ -812,8 +812,8 @@ _ = _TestRedis
 
 
 __all__ = [
-    "PublisherIQL",
-    "PublisherIQLError",
+    "Publisher",
+    "PublisherError",
     "RedisHashMapKey",
     "RedisKey",
     "publish",

utilities/slack_sdk.py

@@ -35,7 +35,7 @@ async def _send_adapter(url: str, text: str, /) -> None:
 
 
 @dataclass(init=False, unsafe_hash=True)
-class SlackHandlerIQL(Handler, InfiniteQueueLooper[None, str]):
+class SlackHandler(Handler, InfiniteQueueLooper[None, str]):
     """Handler for sending messages to Slack."""
 
     @override
@@ -110,4 +110,4 @@ def _get_client(url: str, /, *, timeout: Duration = _TIMEOUT) -> AsyncWebhookCli
     return AsyncWebhookClient(url, timeout=timeout_use)
 
 
-__all__ = ["SendToSlackError", "SlackHandlerIQL", "send_to_slack"]
+__all__ = ["SendToSlackError", "SlackHandler", "send_to_slack"]

utilities/sqlalchemy.py

@@ -609,7 +609,7 @@ class TablenameMixin:
 
 
 @dataclass(kw_only=True)
-class UpserterIQL(InfiniteQueueLooper[None, _InsertItem]):
+class Upserter(InfiniteQueueLooper[None, _InsertItem]):
     """Upsert a set of items to a database."""
 
     engine: AsyncEngine
@@ -641,12 +641,12 @@ class UpserterIQL(InfiniteQueueLooper[None, _InsertItem]):
     def _yield_events_and_exceptions(
         self,
     ) -> Iterator[tuple[None, MaybeType[BaseException]]]:
-        yield (None, UpserterIQLError)
+        yield (None, UpserterError)
 
 
 @dataclass(kw_only=True)
-class UpserterIQLError(Exception):
-    upserter: UpserterIQL
+class UpserterError(Exception):
+    upserter: Upserter
 
     @override
     def __str__(self) -> str:
@@ -1108,8 +1108,8 @@ __all__ = [
     "InsertItemsError",
     "TablenameMixin",
     "UpsertItemsError",
-    "UpserterIQL",
-    "UpserterIQLError",
+    "Upserter",
+    "UpserterError",
     "check_engine",
     "columnwise_max",
     "columnwise_min",

utilities/types.py

@@ -92,6 +92,7 @@ type DateTimeLike = MaybeStr[dt.datetime]
 type DateOrDateTime = dt.date | dt.datetime
 type Duration = Number | dt.timedelta
 type DurationLike = MaybeStr[Duration]
+type DurationOrEveryDuration = Duration | tuple[Literal["every"], Duration]
 type MaybeCallableDate = MaybeCallable[dt.date]
 type MaybeCallableDateTime = MaybeCallable[dt.datetime]
 type TimeLike = MaybeStr[dt.time]
@@ -270,6 +271,7 @@ __all__ = [
     "DateTimeLike",
     "Duration",
     "DurationLike",
+    "DurationOrEveryDuration",
     "EnumLike",
     "ExcInfo",
     "IterableHashable",