hmr 0.5.1__py3-none-any.whl → 0.5.2__py3-none-any.whl

{hmr-0.5.1.dist-info → hmr-0.5.2.dist-info}/METADATA

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

{hmr-0.5.1.dist-info → hmr-0.5.2.dist-info}/RECORD

@@ -1,14 +1,14 @@
-hmr-0.5.1.dist-info/METADATA,sha256=ad1Eq5-t1QdaHg5SI1N_CIvz_ADraD8FCvPtj5Xvek8,258
-hmr-0.5.1.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
-hmr-0.5.1.dist-info/entry_points.txt,sha256=g_T0uJ43WgsdG14kkkdaBQuIL0HO-m1qvtjXMP6d060,59
+hmr-0.5.2.dist-info/METADATA,sha256=kc7P4_6r3h7ZzS3fLQ99YLceVJcVJrGG0iWLCKi6XbU,258
+hmr-0.5.2.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
+hmr-0.5.2.dist-info/entry_points.txt,sha256=g_T0uJ43WgsdG14kkkdaBQuIL0HO-m1qvtjXMP6d060,59
 reactivity/__init__.py,sha256=pX-RUzkezCC1x4eOWGxNhXbwrbvBLP_3pQuZr9eZz1Y,300
 reactivity/functional.py,sha256=jJLEq88w5ag2cXuD1T118uEjnEg37_mqvWmFkb47KVY,1300
 reactivity/helpers.py,sha256=rcrCXy1mmlei-if_BN4nSgS2IG9kBj-jq1YRwEfcITo,5329
 reactivity/hmr/__init__.py,sha256=S5ZIHqCRpevdzWuhS0aCua_S8F0LkK0YNg6IgeTScFQ,177
 reactivity/hmr/__main__.py,sha256=uIcyjR5gMFIXH_3hS0B3SD00RirVf7GIct-uItx675o,64
-reactivity/hmr/api.py,sha256=Esb1fYiBW0SLxQ0MPXby25ZgIIZhIp-M3b2KiqpffmU,2094
-reactivity/hmr/core.py,sha256=BHtkX5b3MAu-dxiMMuAn-8F6JnTkKZSNbEew7Wk59bc,11571
+reactivity/hmr/api.py,sha256=sKDQV7uKRVoDifTZKrLQGZP-fzgYbdVjfEOC6ki4NKY,2280
+reactivity/hmr/core.py,sha256=kXrPyXi6yxE88-5wcRB_r-ikOH_3u0ISUEx1nXtt-SI,11686
 reactivity/hmr/hooks.py,sha256=jIFpe4CNxfaS9RcR4OIodx_sOZlnJ_IA1T1RtHPXhwU,945
 reactivity/hmr/utils.py,sha256=h9m7iRXlvsLTrHoXV9gEVbhz3XsPK4KgnStYoCAWE5I,1616
 reactivity/primitives.py,sha256=QkiUF8bqLZtyXguWwNJi34HJIc-SzUKqi1M9UgTNuvI,5952
-hmr-0.5.1.dist-info/RECORD,,
+hmr-0.5.2.dist-info/RECORD,,

reactivity/hmr/api.py

@@ -1,9 +1,13 @@
+import sys
+
 from .core import AsyncReloader, BaseReloader, SyncReloader, create_effect
 from .hooks import call_post_reload_hooks, call_pre_reload_hooks
 
 
 class LifecycleMixin(BaseReloader):
     def run_with_hooks(self):
+        self._original_main_module = sys.modules["__main__"]
+        sys.modules["__main__"] = self.entry_module
         call_pre_reload_hooks()
         self.effect = create_effect(self.run_entry_file)
         call_post_reload_hooks()
@@ -12,6 +16,7 @@ class LifecycleMixin(BaseReloader):
         self.effect.dispose()
         self.entry_module.load.dispose()
         self.entry_module.load.invalidate()
+        sys.modules["__main__"] = self._original_main_module
 
 
 class SyncReloaderAPI(SyncReloader, LifecycleMixin):

reactivity/hmr/core.py

@@ -331,11 +331,14 @@ def cli():
     if len(sys.argv) < 2:
         print("\n Usage: hmr <entry file>, just like python <entry file>\n")
         exit(1)
-    entry = sys.argv[-1]
+    sys.argv.pop(0)  # this file itself
+    entry = sys.argv[0]
     if not (path := Path(entry)).is_file():
         raise FileNotFoundError(path.resolve())
     sys.path.insert(0, ".")
-    SyncReloader(entry).keep_watching_until_interrupt()
+    reloader = SyncReloader(entry)
+    sys.modules["__main__"] = reloader.entry_module
+    reloader.keep_watching_until_interrupt()
 
 
-__version__ = "0.5.1"
+__version__ = "0.5.2"