dbx_patch.patches.post_import_hook_verify
[docs]
module
dbx_patch.patches.post_import_hook_verify
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95 | """PostImportHook Compatibility Check.
This module verifies that PostImportHook.ImportHookFinder doesn't interfere
with editable install imports.
PostImportHook is used to trigger callbacks after modules are imported.
It shouldn't block imports, but we verify it doesn't interfere.
"""
from dbx_patch.base_patch import BaseVerification
from dbx_patch.models import PatchResult
class PostImportHookVerification(BaseVerification):
"""Verification for PostImportHook compatibility.
Verifies that PostImportHook doesn't interfere with editable imports.
"""
def verify(self) -> PatchResult:
"""Verify PostImportHook doesn't interfere with editable imports.
Returns:
PatchResult with verification details
"""
logger = self._get_logger()
if self._is_verified:
if logger:
logger.info("PostImportHook already verified.")
return PatchResult(
success=True,
already_patched=True,
hook_found=True,
)
try:
# Import the PostImportHook module
from dbruntime.PostImportHook import ImportHookFinder # type: ignore[import-not-found]
if logger:
logger.info("Verifying PostImportHook compatibility...")
# PostImportHook works by:
# 1. Checking if module is in _post_import_hooks registry
# 2. If yes, wrapping the loader to trigger callbacks after import
# 3. If no, returning None and letting other finders handle it
# This should not interfere with editable imports because:
# - It only activates if a module is registered for post-import hooks
# - It wraps the loader but doesn't block the import
# - It's designed to be transparent to the import process
self._is_verified = True
if logger:
logger.success("PostImportHook verified - compatible with editable installs!")
with logger.indent():
logger.info("PostImportHook only triggers callbacks, doesn't block imports")
logger.info("No modifications needed")
return PatchResult(
success=True,
already_patched=False,
hook_found=True,
)
except ImportError as e:
if logger:
logger.warning(f"Could not import PostImportHook: {e}")
with logger.indent():
logger.info("This is normal if not running in Databricks environment.")
return PatchResult(
success=False,
already_patched=False,
hook_found=False,
error=str(e),
)
except Exception as e:
if logger:
logger.error(f"Error verifying PostImportHook: {e}") # noqa: TRY400
return PatchResult(
success=False,
already_patched=False,
hook_found=True,
error=str(e),
)
def is_verified(self) -> bool:
"""Check if PostImportHook has been verified.
Returns:
True if verified, False otherwise
"""
return self._is_verified
|