0.18.0

Author: EcmaXp

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "reloader.py"
-version = "0.17.1"
+version = "0.18.0"
 description = "A simple script reloader"
 license = "MIT"
 authors = [{ name = "EcmaXp", email = "[email protected]" }]

reloader.py

@@ -14,6 +14,7 @@
 import linecache
 import os
 import sys
+import sysconfig
 import threading
 import traceback
 import warnings
@@ -35,7 +36,7 @@
 from watchdog.utils.event_debouncer import EventDebouncer
 
 __author__ = "EcmaXp"
-__version__ = "0.17.1"
+__version__ = "0.18.0"
 __license__ = "MIT"
 __url__ = "https://pypi.org/project/reloader.py/"
 __all__ = [
@@ -300,7 +301,7 @@ def load_lines(self) -> list[str]:
         linecache.updatecache(self.file, self.module.__dict__)
 
         lines = None
-        if self.module.__loader__:
+        if getattr(self.module, "__loader__", None):
             with suppress(OSError):
                 lines = inspect.getsourcelines(self.module)[0][:]
 
@@ -444,7 +445,12 @@ def watch_all(self):
 
 
 class Watcher:
-    def __init__(self, *, debounce_interval: float = DEFAULT_DEBOUNCE_INTERVAL):
+    def __init__(
+        self,
+        *,
+        debounce_interval: float = DEFAULT_DEBOUNCE_INTERVAL,
+        disallowed_paths: set[Path] | None = None,
+    ):
         self.handlers: list[Callable[[list[CodeModule | Path]], None]] = []
         self._code_modules: dict[str, CodeModule | ScriptModule] = {}
         self._resources: set[str] = set()
@@ -463,6 +469,7 @@ def __init__(self, *, debounce_interval: float = DEFAULT_DEBOUNCE_INTERVAL):
             observer=self._watchdog_observer,
         )
         self._sys_modules_watcher = SysModulesWatcher(self._module_callback)
+        self._disallowed_paths = disallowed_paths or set()
 
     def _get_threads(self):
         return (
@@ -504,8 +511,22 @@ def get_code_module(self, module_name: str) -> CodeModule | ScriptModule | None:
 
         raise ValueError(f"Module {module_name!r} not found")
 
+    def _check_module(self, module: ModuleType):
+        return self._is_allowed(module) and self._has_source(module)
+
+    def _is_allowed(self, module: ModuleType):
+        path = getattr(module, "__file__", None)
+        if not path:
+            return True
+
+        path = Path(path).absolute()
+        return not any(
+            path.is_relative_to(disallowed_path)
+            for disallowed_path in self._disallowed_paths
+        )
+
     @staticmethod
-    def _check_module(module: ModuleType):
+    def _has_source(module: ModuleType):
         try:
             inspect.getsource(module)
             return True
@@ -770,6 +791,8 @@ def get_reloader_cls(reloader_cls: Type[T], loop: bool) -> Type[T]:
 parser.add_argument("--clear", "-C", action="store_true")
 parser.add_argument("--debounce", "-d", type=float, default=DEFAULT_DEBOUNCE_INTERVAL)
 parser.add_argument("--no-cwd-python-path", "-P", action="store_true")
+parser.add_argument("--disallowed-path-types", type=str, action="append", default=[])
+parser.add_argument("--allowed-path-types", type=str, action="append", default=[])
 parser.add_argument("--version", action="version", version=f"%(prog)s {__version__}")
 parser.add_argument("script", type=Path, nargs=argparse.OPTIONAL)
 parser.add_argument("argv", type=str, nargs=argparse.REMAINDER)
@@ -786,7 +809,21 @@ def main():
     if not args.no_cwd_python_path:
         sys.path.insert(0, str(Path.cwd().resolve()))
 
-    watcher = Watcher(debounce_interval=args.debounce)
+    disallowed_path_types = set(
+        args.disallowed_path_types or sysconfig.get_path_names()
+    )
+    for allowed_path_type in args.allowed_path_types:
+        disallowed_path_types.remove(allowed_path_type)
+
+    disallowed_paths = {
+        Path(sysconfig.get_path(path_type)).absolute()
+        for path_type in disallowed_path_types
+    }
+
+    watcher = Watcher(
+        debounce_interval=args.debounce,
+        disallowed_paths=disallowed_paths,
+    )
     for path in args.watch:
         watcher.watch_resource(path)