hmr 0.4.0.4__py3-none-any.whl → 0.4.1__py3-none-any.whl

{hmr-0.4.0.4.dist-info → hmr-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hmr
-Version: 0.4.0.4
+Version: 0.4.1
 Summary: Hot Module Reload for Python
 Project-URL: repository, https://github.com/promplate/pyth-on-line/tree/reactivity
 Requires-Python: >=3.12

{hmr-0.4.0.4.dist-info → hmr-0.4.1.dist-info}/RECORD

@@ -1,14 +1,14 @@
-hmr-0.4.0.4.dist-info/METADATA,sha256=LUEgpV4DByNJUiBfSXLxwdbI0W1GcxaZlemWOA6BX7M,260
-hmr-0.4.0.4.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
-hmr-0.4.0.4.dist-info/entry_points.txt,sha256=g_T0uJ43WgsdG14kkkdaBQuIL0HO-m1qvtjXMP6d060,59
+hmr-0.4.1.dist-info/METADATA,sha256=Zx1F2ZUK7Els9nM6w76ceL693CD7JAPUvf7uBVUS2LQ,258
+hmr-0.4.1.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
+hmr-0.4.1.dist-info/entry_points.txt,sha256=g_T0uJ43WgsdG14kkkdaBQuIL0HO-m1qvtjXMP6d060,59
 reactivity/__init__.py,sha256=pX-RUzkezCC1x4eOWGxNhXbwrbvBLP_3pQuZr9eZz1Y,300
 reactivity/functional.py,sha256=U06vshcVhZ0sb218gcmHtEhfgTNAGtQ7zyvPz2w5qKM,1292
 reactivity/helpers.py,sha256=1KCpre2HTFZrngEKkI2HwSFMkCmsUCq2aPEbp0y3kKg,5140
 reactivity/hmr/__init__.py,sha256=S5ZIHqCRpevdzWuhS0aCua_S8F0LkK0YNg6IgeTScFQ,177
 reactivity/hmr/__main__.py,sha256=uIcyjR5gMFIXH_3hS0B3SD00RirVf7GIct-uItx675o,64
-reactivity/hmr/api.py,sha256=7_NitUF3MA8lepS3p_uTbPxZhFIGIXGxPu6iyRZP7o0,1916
-reactivity/hmr/core.py,sha256=CB0QRElWIRMZGe4iDVJrH9B7vKM4R57_hOVh-EDRfqI,11679
-reactivity/hmr/hooks.py,sha256=-yLr5ktiyqPb1nDbHsgv6-c_ZkziBjNqCU-0PCfXGYU,592
+reactivity/hmr/api.py,sha256=Esb1fYiBW0SLxQ0MPXby25ZgIIZhIp-M3b2KiqpffmU,2094
+reactivity/hmr/core.py,sha256=HxPngNSKcw5OrIiokqJtbUfAw2Fph-1G3ToJIDP4Pf4,11677
+reactivity/hmr/hooks.py,sha256=jIFpe4CNxfaS9RcR4OIodx_sOZlnJ_IA1T1RtHPXhwU,945
 reactivity/hmr/utils.py,sha256=-PO-LMP4sc3IP-Bn_baq2w9IFWBZ3zGesgRn5wR6bS0,1555
 reactivity/primitives.py,sha256=mB6cbHKDqtilOfgaEhshtRWJq9s0nPEKqRK0hfCoyFE,5671
-hmr-0.4.0.4.dist-info/RECORD,,
+hmr-0.4.1.dist-info/RECORD,,

reactivity/hmr/api.py

@@ -29,10 +29,11 @@ class SyncReloaderAPI(SyncReloader, LifecycleMixin):
         self.clean_up()
 
     async def __aenter__(self):
-        from asyncio import ensure_future, to_thread
+        from asyncio import ensure_future, sleep, to_thread
 
         await to_thread(self.run_with_hooks)
         self.future = ensure_future(to_thread(self.start_watching))
+        await sleep(0)
         return super()
 
     async def __aexit__(self, *_):
@@ -44,11 +45,19 @@ class SyncReloaderAPI(SyncReloader, LifecycleMixin):
 class AsyncReloaderAPI(AsyncReloader, LifecycleMixin):
     def __enter__(self):
         from asyncio import run
-        from threading import Thread
+        from threading import Event, Thread
 
         self.run_with_hooks()
-        self.thread = Thread(target=lambda: run(self.start_watching()))
+
+        e = Event()
+
+        async def task():
+            e.set()
+            await self.start_watching()
+
+        self.thread = Thread(target=lambda: run(task()))
         self.thread.start()
+        e.wait()
         return super()
 
     def __exit__(self, *_):
@@ -57,10 +66,11 @@ class AsyncReloaderAPI(AsyncReloader, LifecycleMixin):
         self.clean_up()
 
     async def __aenter__(self):
-        from asyncio import ensure_future, to_thread
+        from asyncio import ensure_future, sleep, to_thread
 
         await to_thread(self.run_with_hooks)
         self.future = ensure_future(self.start_watching())
+        await sleep(0)
         return super()
 
     async def __aexit__(self, *_):

reactivity/hmr/core.py

@@ -340,4 +340,4 @@ def cli():
     SyncReloader(entry).keep_watching_until_interrupt()
 
 
-__version__ = "0.4.0.4"
+__version__ = "0.4.1"

reactivity/hmr/hooks.py

@@ -1,4 +1,5 @@
 from collections.abc import Callable
+from contextlib import contextmanager
 from typing import Any
 
 pre_reload_hooks: dict[str, Callable[[], Any]] = {}
@@ -15,6 +16,24 @@ def post_reload[T](func: Callable[[], T]) -> Callable[[], T]:
     return func
 
 
+@contextmanager
+def use_pre_reload(func):
+    pre_reload(func)
+    try:
+        yield func
+    finally:
+        pre_reload_hooks.pop(func.__name__, None)
+
+
+@contextmanager
+def use_post_reload(func):
+    post_reload(func)
+    try:
+        yield func
+    finally:
+        post_reload_hooks.pop(func.__name__, None)
+
+
 def call_pre_reload_hooks():
     for func in pre_reload_hooks.values():
         func()