hmr 0.4.0.1__tar.gz → 0.4.0.3__tar.gz

{hmr-0.4.0.1 → hmr-0.4.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hmr
-Version: 0.4.0.1
+Version: 0.4.0.3
 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.1 → hmr-0.4.0.3}/pyproject.toml

@@ -6,7 +6,7 @@ description = "Hot Module Reload for Python"
 dependencies = [
     "watchfiles>=0.21,<2 ; sys_platform != 'emscripten'",
 ]
-version = "0.4.0.1"
+version = "0.4.0.3"
 
 [project.scripts]
 hmr = "reactivity.hmr:cli"

{hmr-0.4.0.1 → hmr-0.4.0.3}/reactivity/hmr/api.py

@@ -1,16 +1,23 @@
 from .core import AsyncReloader, BaseReloader, SyncReloader
+from .hooks import call_post_reload_hooks, call_pre_reload_hooks
 
 
-def _clean_up(r: BaseReloader):
-    r.entry_module.load.dispose()
-    r.entry_module.load.invalidate()
+class LifecycleMixin(BaseReloader):
+    def run_with_hooks(self):
+        call_pre_reload_hooks()
+        self.run_entry_file()
+        call_post_reload_hooks()
+
+    def clean_up(self):
+        self.entry_module.load.dispose()
+        self.entry_module.load.invalidate()
 
 
-class SyncReloaderAPI(SyncReloader):
+class SyncReloaderAPI(SyncReloader, LifecycleMixin):
     def __enter__(self):
         from threading import Thread
 
-        self.run_entry_file()
+        self.run_with_hooks()
         self.thread = Thread(target=self.start_watching)
         self.thread.start()
         return super()
@@ -18,27 +25,27 @@ class SyncReloaderAPI(SyncReloader):
     def __exit__(self, *_):
         self.stop_watching()
         self.thread.join()
-        _clean_up(self)
+        self.clean_up()
 
     async def __aenter__(self):
         from asyncio import ensure_future, to_thread
 
-        await to_thread(self.run_entry_file)
+        await to_thread(self.run_with_hooks)
         self.future = ensure_future(to_thread(self.start_watching))
         return super()
 
     async def __aexit__(self, *_):
         self.stop_watching()
         await self.future
-        _clean_up(self)
+        self.clean_up()
 
 
-class AsyncReloaderAPI(AsyncReloader):
+class AsyncReloaderAPI(AsyncReloader, LifecycleMixin):
     def __enter__(self):
         from asyncio import run
         from threading import Thread
 
-        self.run_entry_file()
+        self.run_with_hooks()
         self.thread = Thread(target=lambda: run(self.start_watching()))
         self.thread.start()
         return super()
@@ -46,16 +53,16 @@ class AsyncReloaderAPI(AsyncReloader):
     def __exit__(self, *_):
         self.stop_watching()
         self.thread.join()
-        _clean_up(self)
+        self.clean_up()
 
     async def __aenter__(self):
         from asyncio import ensure_future, to_thread
 
-        await to_thread(self.run_entry_file)
+        await to_thread(self.run_with_hooks)
         self.future = ensure_future(self.start_watching())
         return super()
 
     async def __aexit__(self, *_):
         self.stop_watching()
         await self.future
-        _clean_up(self)
+        self.clean_up()

{hmr-0.4.0.1 → hmr-0.4.0.3}/reactivity/hmr/core.py

@@ -339,4 +339,4 @@ def cli():
     SyncReloader(entry).keep_watching_until_interrupt()
 
 
-__version__ = "0.4.0.1"
+__version__ = "0.4.0.3"