hmr 0.0.2.1__tar.gz → 0.0.3__tar.gz

{hmr-0.0.2.1 → hmr-0.0.3}/PKG-INFO

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

@@ -6,7 +6,7 @@ description = "Hot Module Reload for Python"
 dependencies = [
     "watchfiles>=0.21,<2",
 ]
-version = "0.0.2.1"
+version = "0.0.3"
 
 [project.scripts]
 hmr = "reactivity.hmr:cli"

{hmr-0.0.2.1 → hmr-0.0.3}/reactivity/hmr.py

@@ -1,5 +1,5 @@
 import sys
-from collections.abc import Iterable, Sequence
+from collections.abc import Iterable, MutableMapping, Sequence
 from contextlib import suppress
 from functools import cached_property
 from importlib.abc import Loader, MetaPathFinder
@@ -9,6 +9,7 @@ from inspect import currentframe
 from pathlib import Path
 from runpy import run_path
 from types import ModuleType
+from typing import Any
 
 from . import Reactive, batch, create_effect, memoized_method
 
@@ -26,11 +27,29 @@ def is_called_in_this_file() -> bool:
     return frame.f_globals.get("__file__") == __file__
 
 
+class NamespaceProxy(Reactive[str, Any]):
+    def __init__(self, initial: MutableMapping, check_equality=True):
+        super().__init__(initial, check_equality)
+        self._original = initial
+
+    def __setitem__(self, key, value):
+        self._original[key] = value
+        return super().__setitem__(key, value)
+
+    def __delitem__(self, key):
+        del self._original[key]
+        return super().__delitem__(key)
+
+
 class ReactiveModule(ModuleType):
     def __init__(self, file: Path, namespace: dict, name: str, doc: str | None = None):
         super().__init__(name, doc)
+        self.__is_initialized = False
+        self.__dict__.update(namespace)
+        self.__is_initialized = True
+
         self.__namespace = namespace
-        self.__namespace_proxy = Reactive(namespace)
+        self.__namespace_proxy = NamespaceProxy(namespace)
         self.__file = file
 
     @property
@@ -42,9 +61,18 @@ class ReactiveModule(ModuleType):
     @property
     def load(self):
         if is_called_in_this_file():
-            return lambda: exec(self.__file.read_text("utf-8"), self.__namespace, self.__namespace_proxy)
+            code = compile(self.__file.read_text("utf-8"), str(self.__file), "exec", dont_inherit=True)
+            return lambda: exec(code, self.__namespace, self.__namespace_proxy)
         raise AttributeError("load")
 
+    def __dir__(self):
+        return iter(self.__namespace_proxy)
+
+    def __getattribute__(self, name: str):
+        if name == "__dict__" and self.__is_initialized:
+            return self.__namespace
+        return super().__getattribute__(name)
+
     def __getattr__(self, name: str):
         try:
             return self.__namespace_proxy[name]
@@ -64,11 +92,10 @@ class ReactiveModuleLoader(Loader):
         self._is_package = is_package
 
     def create_module(self, spec: ModuleSpec):
-        namespace = {}
+        namespace = {"__file__": str(self._file), "__spec__": spec, "__loader__": self}
         if self._is_package:
             assert self._file.name == "__init__.py"
             namespace["__path__"] = [str(self._file.parent.parent)]
-            namespace["__package__"] = spec.name
         return ReactiveModule(self._file, namespace, spec.name)
 
     def exec_module(self, module: ModuleType):
@@ -100,7 +127,7 @@ class ReactiveModuleFinder(MetaPathFinder):
                 return spec_from_loader(fullname, ReactiveModuleLoader(file), origin=str(file))
             file = directory / f"{fullname.replace('.', '/')}/__init__.py"
             if file.is_file() and all(not file.is_relative_to(exclude) for exclude in self.excludes):
-                return spec_from_loader(fullname, ReactiveModuleLoader(file, is_package=True), origin=str(file))
+                return spec_from_loader(fullname, ReactiveModuleLoader(file, is_package=True), origin=str(file), is_package=True)
 
 
 def patch_module(name_or_module: str | ModuleType):
@@ -219,4 +246,4 @@ def cli():
     SyncReloader(entry, excludes={".venv"}).keep_watching_until_interrupt()
 
 
-__version__ = "0.0.2.1"
+__version__ = "0.0.3"