Update type hints

Author: twsl

notebooks/examples.ipynb

@@ -454,15 +454,7 @@
     "7. **Refresh** - Detect newly installed packages\n",
     "8. **Development Workflow** - Complete dev setup with autoreload\n",
     "9. **Debugging** - Diagnostic information\n",
-    "10. **Cleanup** - Remove patches\n",
-    "\n",
-    "### Next Steps\n",
-    "\n",
-    "- See **QUICKSTART.md** for more copy-paste examples\n",
-    "- See **README.md** for complete documentation\n",
-    "- See **INSTALL.md** for installation instructions\n",
-    "\n",
-    "**Happy coding! 🚀**\n"
+    "10. **Cleanup** - Remove patches\n"
    ]
   }
  ],

src/dbx_patch/patches/python_path_hook_patch.py

@@ -10,12 +10,13 @@
 
 from collections.abc import Callable
 import sys
+from typing import Any
 
 from dbx_patch.models import PatchResult
 from dbx_patch.utils.logger import get_logger
 
 _PATCH_APPLIED = False
-_ORIGINAL_HANDLE_SYS_PATH = None
+_ORIGINAL_HANDLE_SYS_PATH: Callable[..., None] | None = None
 _EDITABLE_PATHS: set[str] = set()
 
 
@@ -41,7 +42,7 @@ def refresh_editable_paths() -> int:
     return len(_EDITABLE_PATHS)
 
 
-def create_patched_handle_sys_path(original_method) -> Callable[..., None]:
+def create_patched_handle_sys_path(original_method: Callable[..., None]) -> Callable[..., None]:
     """Create a patched version of PythonPathHook._handle_sys_path_maybe_updated.
 
     That preserves editable install paths.
@@ -53,7 +54,7 @@ def create_patched_handle_sys_path(original_method) -> Callable[..., None]:
         Patched method that preserves editable paths
     """
 
-    def patched_handle_sys_path_maybe_updated(self) -> None:
+    def patched_handle_sys_path_maybe_updated(self: Any) -> None:
         # Call original method first
         original_method(self)
 
@@ -111,6 +112,16 @@ def patch_python_path_hook(verbose: bool = True) -> PatchResult:
         # Save original method
         _ORIGINAL_HANDLE_SYS_PATH = PythonPathHook._handle_sys_path_maybe_updated
 
+        # Type narrowing check
+        if _ORIGINAL_HANDLE_SYS_PATH is None:
+            if logger:
+                logger.error("Failed to save original method")
+            return PatchResult(
+                success=False,
+                already_patched=False,
+                hook_found=True,
+            )
+
         # Create and apply patch
         patched_method = create_patched_handle_sys_path(_ORIGINAL_HANDLE_SYS_PATH)
         PythonPathHook._handle_sys_path_maybe_updated = patched_method

src/dbx_patch/patches/sys_path_init_patch.py

@@ -8,15 +8,16 @@
 """
 
 from collections.abc import Callable
+from typing import Any
 
 from dbx_patch.models import PatchResult
 from dbx_patch.utils.logger import get_logger
 
 _PATCH_APPLIED = False
-_ORIGINAL_PATCH_SYS_PATH = None
+_ORIGINAL_PATCH_SYS_PATH: Callable[[], None] | None = None
 
 
-def create_patched_patch_sys_path(original_function) -> Callable[[], None]:
+def create_patched_patch_sys_path(original_function: Callable[[], None]) -> Callable[[], None]:
     """Create a patched version of patch_sys_path.
 
     With_developer_paths that also processes .pth files for editable installs.
@@ -88,6 +89,16 @@ def patch_sys_path_init(verbose: bool = True) -> PatchResult:
         # Save original function
         _ORIGINAL_PATCH_SYS_PATH = sys_path_init.patch_sys_path_with_developer_paths
 
+        # Type narrowing check
+        if _ORIGINAL_PATCH_SYS_PATH is None:
+            if logger:
+                logger.error("Failed to save original function")
+            return PatchResult(
+                success=False,
+                already_patched=False,
+                function_found=True,
+            )
+
         # Create and apply patch
         patched_function = create_patched_patch_sys_path(_ORIGINAL_PATCH_SYS_PATH)
         sys_path_init.patch_sys_path_with_developer_paths = patched_function

src/dbx_patch/patches/wsfs_import_hook_patch.py

@@ -14,12 +14,13 @@
 
 from collections.abc import Callable
 import inspect
+from typing import Any
 
 from dbx_patch.models import PatchResult
 from dbx_patch.utils.logger import get_logger
 
 _PATCH_APPLIED = False
-_ORIGINAL_IS_USER_IMPORT = None
+_ORIGINAL_IS_USER_IMPORT: Callable[..., bool] | None = None
 _EDITABLE_PATHS: set[str] = set()
 
 
@@ -47,7 +48,7 @@ def refresh_editable_paths() -> int:
     return len(_EDITABLE_PATHS)
 
 
-def create_patched_is_user_import(original_method) -> Callable[..., bool]:
+def create_patched_is_user_import(original_method: Callable[..., bool]) -> Callable[..., bool]:
     """Create a patched version of WsfsImportHook.__is_user_import that allows editable installs.
 
     Args:
@@ -57,7 +58,7 @@ def create_patched_is_user_import(original_method) -> Callable[..., bool]:
         Patched method that includes editable path checking
     """
 
-    def patched_is_user_import(self) -> bool:
+    def patched_is_user_import(self: Any) -> bool:
         try:
             f = inspect.currentframe()
             num_items_processed = 0
@@ -139,6 +140,19 @@ def patch_wsfs_import_hook(verbose: bool = True) -> PatchResult:
         # Save original method
         _ORIGINAL_IS_USER_IMPORT = WsfsImportHook._WsfsImportHook__is_user_import
 
+        # Type narrowing check
+        if _ORIGINAL_IS_USER_IMPORT is None:
+            if logger:
+                logger.error("Failed to save original method")
+            return PatchResult(
+                success=False,
+                already_patched=False,
+                editable_paths_count=0,
+                editable_paths=[],
+                hook_found=True,
+                error="Failed to save original method",
+            )
+
         # Create and apply patch
         patched_method = create_patched_is_user_import(_ORIGINAL_IS_USER_IMPORT)
         WsfsImportHook._WsfsImportHook__is_user_import = patched_method