hmr 0.2.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

{hmr-0.2.0.dist-info → hmr-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hmr
-Version: 0.2.0
+Version: 0.3.0
 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.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+hmr-0.3.0.dist-info/METADATA,sha256=PcpVod1MtZSUbVP91KWKD9ioN7qTULhzxHR9vK20ZvA,258
+hmr-0.3.0.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+hmr-0.3.0.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/core.py,sha256=NfzgqTdFD7wrO_8ogZTSqMmVIRw4Z_hhk5-ONRlunSI,9993
+reactivity/hmr/hooks.py,sha256=-yLr5ktiyqPb1nDbHsgv6-c_ZkziBjNqCU-0PCfXGYU,592
+reactivity/hmr/utils.py,sha256=zgKjz3RhcUDYLoIqZFRVBcPtPWUJA1YphJycyrQx3tk,1464
+reactivity/primitives.py,sha256=DR2waJbzhVKOioHXMliE4FIsxQUq7DZA0umPrlvchA4,4217
+hmr-0.3.0.dist-info/RECORD,,

reactivity/hmr/__init__.py

@@ -0,0 +1,5 @@
+from .core import cli
+from .hooks import post_reload, pre_reload
+from .utils import cache_across_reloads
+
+__all__ = ("cache_across_reloads", "cli", "post_reload", "pre_reload")

reactivity/{hmr.py → hmr/core.py}

@@ -7,11 +7,12 @@ from importlib.machinery import ModuleSpec
 from importlib.util import spec_from_loader
 from inspect import currentframe
 from pathlib import Path
-from runpy import run_path
+from site import getsitepackages
 from types import ModuleType, TracebackType
 from typing import Any
 
-from . import Reactive, batch, memoized_method
+from .. import Reactive, batch, memoized_method
+from .hooks import call_post_reload_hooks, call_pre_reload_hooks
 
 
 def is_called_in_this_file() -> bool:
@@ -115,7 +116,7 @@ class ReactiveModuleFinder(MetaPathFinder):
     def __init__(self, includes: Iterable[str] = ".", excludes: Iterable[str] = ()):
         super().__init__()
         self.includes = [Path(i).resolve() for i in includes]
-        self.excludes = [Path(e).resolve() for e in excludes]
+        self.excludes = [Path(e).resolve() for e in (*excludes, *getsitepackages())]
 
     def _accept(self, path: Path):
         return path.is_file() and not is_relative_to_any(path, self.excludes) and is_relative_to_any(path, self.includes)
@@ -149,7 +150,7 @@ def patch_module(name_or_module: str | ModuleType):
     return m
 
 
-def patch_meta_path(includes: Iterable[str] = (".",), excludes: Iterable[str] = (".venv",)):
+def patch_meta_path(includes: Iterable[str] = (".",), excludes: Iterable[str] = ()):
     sys.meta_path.insert(0, ReactiveModuleFinder(includes, excludes))
 
 
@@ -187,13 +188,20 @@ class BaseReloader:
         self.includes = includes
         self.excludes = excludes
         patch_meta_path(includes, excludes)
-        self.last_globals = {}
-        self.error_filter = ErrorFilter(__file__, "<frozen runpy>")
+        self.error_filter = ErrorFilter(*(str(i) for i in Path(__file__, "../..").resolve().glob("**/*.py")))
 
     @memoized_method
     def run_entry_file(self):
+        call_pre_reload_hooks()
+
         with self.error_filter:
-            self.last_globals = run_path(self.entry, self.last_globals, "__main__")
+            if not isinstance(module := sys.modules["__main__"], ReactiveModule):
+                namespace = {"__file__": self.entry, "__name__": "__main__"}
+                module = sys.modules["__main__"] = ReactiveModule(Path(self.entry), namespace, "__main__")
+            module.load.invalidate()
+            module.load()
+
+        call_post_reload_hooks()
 
     @property
     def watch_filter(self):
@@ -281,7 +289,7 @@ def cli():
     if not (path := Path(entry)).is_file():
         raise FileNotFoundError(path.resolve())
     sys.path.insert(0, ".")
-    SyncReloader(entry, excludes={".venv"}).keep_watching_until_interrupt()
+    SyncReloader(entry).keep_watching_until_interrupt()
 
 
-__version__ = "0.2.0"
+__version__ = "0.3.0"

reactivity/hmr/hooks.py

@@ -0,0 +1,25 @@
+from collections.abc import Callable
+from typing import Any
+
+pre_reload_hooks: dict[str, Callable[[], Any]] = {}
+post_reload_hooks: dict[str, Callable[[], Any]] = {}
+
+
+def pre_reload[T](func: Callable[[], T]) -> Callable[[], T]:
+    pre_reload_hooks[func.__name__] = func
+    return func
+
+
+def post_reload[T](func: Callable[[], T]) -> Callable[[], T]:
+    post_reload_hooks[func.__name__] = func
+    return func
+
+
+def call_pre_reload_hooks():
+    for func in pre_reload_hooks.values():
+        func()
+
+
+def call_post_reload_hooks():
+    for func in post_reload_hooks.values():
+        func()

reactivity/hmr/utils.py

@@ -0,0 +1,61 @@
+import sys
+from collections import UserDict
+from collections.abc import Callable
+from functools import wraps
+from inspect import getsource
+from types import FunctionType
+
+from .. import create_memo
+from .core import NamespaceProxy, ReactiveModule
+from .hooks import post_reload, pre_reload
+
+memos: dict[str, Callable] = {}
+functions: dict[str, Callable] = {}
+functions_last: set[str] = set()
+
+
+@pre_reload
+def swap():
+    functions_last.update(functions)
+    functions.clear()
+
+
+@post_reload
+def clear_memos():
+    for func in functions_last:
+        if func not in functions and func in memos:
+            del memos[func]
+
+
+def cache_across_reloads[T](func: Callable[[], T]) -> Callable[[], T]:
+    module = sys.modules[func.__module__]
+
+    if not isinstance(module, ReactiveModule):
+        from functools import cache
+
+        return cache(func)  # type: ignore
+
+    source = getsource(func)
+
+    proxy: NamespaceProxy = module._ReactiveModule__namespace_proxy  # type: ignore  # noqa: SLF001
+
+    func = FunctionType(func.__code__, DictProxy(proxy), func.__name__, func.__defaults__, func.__closure__)
+
+    functions[source] = func
+
+    if source in memos and source in functions_last:
+        return memos[source]
+
+    @wraps(func)
+    @create_memo
+    def wrapper():
+        return functions[source]()
+
+    memos[source] = wrapper
+
+    return wrapper
+
+
+class DictProxy(UserDict, dict):  # type: ignore
+    def __init__(self, data):
+        self.data = data

hmr-0.2.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-hmr-0.2.0.dist-info/METADATA,sha256=EOVgm-SYoDvhCdf7X3stdnfoOwt4JiJz4dU-MH5qp4A,258
-hmr-0.2.0.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-hmr-0.2.0.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.py,sha256=CeJSXPeDA-wc_tE_JC0gXfFgeaGsCEvmVbbsCYJtGok,9603
-reactivity/primitives.py,sha256=DR2waJbzhVKOioHXMliE4FIsxQUq7DZA0umPrlvchA4,4217
-hmr-0.2.0.dist-info/RECORD,,