dycw-utilities 0.129.2__py3-none-any.whl → 0.129.3__py3-none-any.whl

{dycw_utilities-0.129.2.dist-info → dycw_utilities-0.129.3.dist-info}/METADATA

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

{dycw_utilities-0.129.2.dist-info → dycw_utilities-0.129.3.dist-info}/RECORD

@@ -1,6 +1,6 @@
-utilities/__init__.py,sha256=xCbJwDm5euQ0FhR3M7TD-JZyt1BghWGAEn2rH1dZO9k,60
+utilities/__init__.py,sha256=xvWWvcsivl7tOOnGKv9OKQ27M1f_gdDFBYwkYVERk-4,60
 utilities/altair.py,sha256=Gpja-flOo-Db0PIPJLJsgzAlXWoKUjPU1qY-DQ829ek,9156
-utilities/asyncio.py,sha256=mP67R3RwY6cTCgiNlxA9h-o3Y6xv5uGzKKB08C49uKI,37001
+utilities/asyncio.py,sha256=3n5EIcSq2xtEF1i4oR0oY2JmBq3NyugeHKFK39Mt22s,37987
 utilities/atomicwrites.py,sha256=geFjn9Pwn-tTrtoGjDDxWli9NqbYfy3gGL6ZBctiqSo,5393
 utilities/atools.py,sha256=IYMuFSFGSKyuQmqD6v5IUtDlz8PPw0Sr87Cub_gRU3M,1168
 utilities/cachetools.py,sha256=C1zqOg7BYz0IfQFK8e3qaDDgEZxDpo47F15RTfJM37Q,2910
@@ -89,7 +89,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.129.2.dist-info/METADATA,sha256=2XZ9vn3lnybj_7rhJwuTFSU8SVBRLkroYURNVY1_9sc,12803
-dycw_utilities-0.129.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.129.2.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.129.2.dist-info/RECORD,,
+dycw_utilities-0.129.3.dist-info/METADATA,sha256=tSQGMjW1iN-3qoQebYNCtjnI6yzQARSXvw0ThdEKy2A,12803
+dycw_utilities-0.129.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.129.3.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.129.3.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.129.2"
+__version__ = "0.129.3"

utilities/asyncio.py

@@ -353,6 +353,7 @@ class Looper(Generic[_T]):
     _is_entered: Event = field(default_factory=Event, init=False, repr=False)
     _is_initialized: Event = field(default_factory=Event, init=False, repr=False)
     _is_initializing: Event = field(default_factory=Event, init=False, repr=False)
+    _is_pending_back_off: Event = field(default_factory=Event, init=False, repr=False)
     _is_pending_restart: Event = field(default_factory=Event, init=False, repr=False)
     _is_pending_stop: Event = field(default_factory=Event, init=False, repr=False)
     _is_pending_stop_when_empty: Event = field(
@@ -439,6 +440,11 @@ class Looper(Generic[_T]):
     def __len__(self) -> int:
         return self._queue.qsize()
 
+    async def _apply_back_off(self) -> None:
+        """Apply a back off period."""
+        await sleep(self._backoff)
+        self._is_pending_back_off.clear()
+
     async def core(self) -> None:
         """Core part of running the looper."""
 
@@ -492,7 +498,7 @@ class Looper(Generic[_T]):
                                 repr_error(error),
                                 self.backoff,
                             )
-                            await sleep(self._backoff)
+                            await self._apply_back_off()
                         case _ as never:
                             assert_never(never)
                 else:
@@ -549,6 +555,21 @@ class Looper(Generic[_T]):
             **kwargs,
         )
 
+    def request_back_off(self) -> None:
+        """Request the looper to back off."""
+        match self._is_pending_back_off.is_set():
+            case True:
+                _ = self._debug and self._logger.debug(
+                    "%s: already requested back off", self
+                )
+            case False:
+                _ = self._debug and self._logger.debug(
+                    "%s: requesting back off...", self
+                )
+                self._is_pending_back_off.set()
+            case _ as never:
+                assert_never(never)
+
     def request_restart(self) -> None:
         """Request the looper to restart."""
         match self._is_pending_restart.is_set():
@@ -563,6 +584,7 @@ class Looper(Generic[_T]):
                 self._is_pending_restart.set()
             case _ as never:
                 assert_never(never)
+        self.request_back_off()
 
     def request_stop(self) -> None:
         """Request the looper to stop."""
@@ -602,9 +624,7 @@ class Looper(Generic[_T]):
         initialization = await self.initialize(skip_sleep_if_failure=True)
         match tear_down, initialization:
             case None, None:
-                _ = self._debug and self._logger.debug(
-                    "%s: finished restarting; sleeping for %s...", self, self.backoff
-                )
+                _ = self._debug and self._logger.debug("%s: finished restarting", self)
                 async with self._lock:
                     self._restart_successes += 1
             case Exception(), None:
@@ -616,6 +636,7 @@ class Looper(Generic[_T]):
                     repr_error(tear_down),
                     self.backoff,
                 )
+                await self._apply_back_off()
             case None, Exception():
                 async with self._lock:
                     self._restart_failures += 1
@@ -625,6 +646,7 @@ class Looper(Generic[_T]):
                     repr_error(initialization),
                     self.backoff,
                 )
+                await self._apply_back_off()
             case Exception(), Exception():
                 async with self._lock:
                     self._restart_failures += 1
@@ -635,9 +657,9 @@ class Looper(Generic[_T]):
                     repr_error(initialization),
                     self.backoff,
                 )
+                await self._apply_back_off()
             case _ as never:
                 assert_never(never)
-        await sleep(self._backoff)
 
     async def run_looper(self) -> None:
         """Run the looper."""
@@ -651,6 +673,8 @@ class Looper(Generic[_T]):
                         self._is_pending_stop_when_empty.is_set() and self.empty()
                     ):
                         await self.stop()
+                    elif self._is_pending_back_off.is_set():
+                        await self._apply_back_off()
                     elif self._is_pending_restart.is_set():
                         await self.restart()
                     elif not self._is_initialized.is_set():
@@ -758,7 +782,7 @@ class Looper(Generic[_T]):
                                 repr_error(error),
                                 self.backoff,
                             )
-                            await sleep(self._backoff)
+                            await self._apply_back_off()
                         case _ as never:
                             assert_never(never)
                 else: