hmr 0.3.3.1__py3-none-any.whl → 0.3.3.3__py3-none-any.whl

{hmr-0.3.3.1.dist-info → hmr-0.3.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hmr
-Version: 0.3.3.1
+Version: 0.3.3.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.3.3.1.dist-info → hmr-0.3.3.3.dist-info}/RECORD

@@ -1,13 +1,13 @@
-hmr-0.3.3.1.dist-info/METADATA,sha256=i7tLdHbmQlSY2AiGNO_3uqmEStVLcsD12rmmzF9O8Yw,260
-hmr-0.3.3.1.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-hmr-0.3.3.1.dist-info/entry_points.txt,sha256=g_T0uJ43WgsdG14kkkdaBQuIL0HO-m1qvtjXMP6d060,59
+hmr-0.3.3.3.dist-info/METADATA,sha256=TWN_VystOGF0zdKckjiayJus6XK7GJ-L03xlibtXs2E,260
+hmr-0.3.3.3.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+hmr-0.3.3.3.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=7gwsIKKrjEahSz9G9oR4s1LdYXQTCIMO0k4UGXGla9Y,3714
 reactivity/hmr/__init__.py,sha256=S5ZIHqCRpevdzWuhS0aCua_S8F0LkK0YNg6IgeTScFQ,177
 reactivity/hmr/api.py,sha256=-0-6Tn0AVkaDs7_qrCCd9TXxRTPDMDB08-UYfepTdec,1644
-reactivity/hmr/core.py,sha256=SCZi7vlSy4IPHQRTVpjlMn3am5L9sYN5zCE1nKw07eM,10163
+reactivity/hmr/core.py,sha256=TSbL1AMiel2eKzk23z1Sb8ruCTx9s5m6blbj0L9RGD4,10252
 reactivity/hmr/hooks.py,sha256=-yLr5ktiyqPb1nDbHsgv6-c_ZkziBjNqCU-0PCfXGYU,592
-reactivity/hmr/utils.py,sha256=zgKjz3RhcUDYLoIqZFRVBcPtPWUJA1YphJycyrQx3tk,1464
+reactivity/hmr/utils.py,sha256=zM7X5I8ywOguOt20uE55INWOGen3FuJZDAvgouu7lI8,1550
 reactivity/primitives.py,sha256=DR2waJbzhVKOioHXMliE4FIsxQUq7DZA0umPrlvchA4,4217
-hmr-0.3.3.1.dist-info/RECORD,,
+hmr-0.3.3.3.dist-info/RECORD,,

reactivity/hmr/core.py

@@ -9,7 +9,8 @@ from inspect import currentframe
 from pathlib import Path
 from site import getsitepackages
 from types import ModuleType, TracebackType
-from typing import Any
+from typing import Any, Self
+from weakref import WeakValueDictionary
 
 from .. import Reactive, batch, memoized_method
 from .hooks import call_post_reload_hooks, call_pre_reload_hooks
@@ -43,6 +44,8 @@ class NamespaceProxy(Reactive[str, Any]):
 
 
 class ReactiveModule(ModuleType):
+    instances: WeakValueDictionary[Path, Self] = WeakValueDictionary()
+
     def __init__(self, file: Path, namespace: dict, name: str, doc: str | None = None):
         super().__init__(name, doc)
         self.__is_initialized = False
@@ -53,6 +56,8 @@ class ReactiveModule(ModuleType):
         self.__namespace_proxy = NamespaceProxy(namespace)
         self.__file = file
 
+        self.instances[file.resolve()] = self
+
     @property
     def file(self):
         if is_called_in_this_file():
@@ -155,7 +160,7 @@ def patch_meta_path(includes: Iterable[str] = (".",), excludes: Iterable[str] =
 
 
 def get_path_module_map():
-    return {module.file.resolve(): module for module in sys.modules.values() if isinstance(module, ReactiveModule)}
+    return {**ReactiveModule.instances}
 
 
 class ErrorFilter:
@@ -307,4 +312,4 @@ def cli():
     SyncReloader(entry).keep_watching_until_interrupt()
 
 
-__version__ = "0.3.3.1"
+__version__ = "0.3.3.3"

reactivity/hmr/utils.py

@@ -1,8 +1,8 @@
-import sys
 from collections import UserDict
 from collections.abc import Callable
 from functools import wraps
-from inspect import getsource
+from inspect import getsource, getsourcefile
+from pathlib import Path
 from types import FunctionType
 
 from .. import create_memo
@@ -28,9 +28,11 @@ def clear_memos():
 
 
 def cache_across_reloads[T](func: Callable[[], T]) -> Callable[[], T]:
-    module = sys.modules[func.__module__]
+    file = getsourcefile(func)
+    assert file is not None
+    module = ReactiveModule.instances.get(Path(file).resolve())
 
-    if not isinstance(module, ReactiveModule):
+    if module is None:
         from functools import cache
 
         return cache(func)  # type: ignore