Update logger

Author: twsl

src/dbx_patch/patches/autoreload_hook_patch.py

@@ -5,8 +5,17 @@
 and only allows imports from specific paths (like /Workspace). We need to add
 editable install paths to this allowlist.
 
-Additionally, we patch builtins.__import__ itself to ensure editable paths work
-even when called through the autoreload hook's original import reference.
+Environment Variables:
+    DBX_PATCH_DEBUG_IMPORTS: Set to '1', 'true', or 'yes' to enable extremely verbose
+        import tracing by patching builtins.__import__. This logs every single import
+        that happens in the Python process and should only be used for debugging
+        import-related issues with editable installs.
+
+Example:
+            import os
+            os.environ['DBX_PATCH_DEBUG_IMPORTS'] = '1'
+            from dbx_patch import patch_dbx
+            patch_dbx()
 """
 
 import builtins
@@ -103,13 +112,21 @@ def patch(self) -> PatchResult:
         """
         logger = self._get_logger()
 
-        # Patch builtins.__import__ for debug logging if in debug mode
-        if logger and logger._logger.isEnabledFor(logging.DEBUG) and not self._import_patch_applied:
-            logger.info("Debug logging enabled - patching builtins.__import__ for debug logging...")
+        # Patch builtins.__import__ for debug logging ONLY if explicitly enabled via env var
+        # This is extremely verbose and should only be used for debugging import issues
+        import os
+
+        if (
+            os.environ.get("DBX_PATCH_DEBUG_IMPORTS", "").lower() in ("1", "true", "yes")
+            and not self._import_patch_applied
+        ):
+            if logger:
+                logger.info("DBX_PATCH_DEBUG_IMPORTS enabled - patching builtins.__import__ for import tracing...")
             self._original_builtins_import = builtins.__import__
             builtins.__import__ = self._patched_builtins_import  # type: ignore[assignment]
             self._import_patch_applied = True
-            logger.info("builtins.__import__ patched for debug logging")
+            if logger:
+                logger.info("builtins.__import__ patched for import tracing (this will be very verbose!)")
 
         if self._is_applied:
             if logger:

src/dbx_patch/utils/logger.py

@@ -18,6 +18,10 @@ class PatchLogger:
     Uses composition with logging.Logger and only logs when:
     - DBX_PATCH_ENABLED env var is set to true
     - DBX_PATCH_LOG_LEVEL env var is set to appropriate level (DEBUG, INFO, WARNING, ERROR, CRITICAL)
+
+    Environment Variables:
+        DBX_PATCH_ENABLED: Set to '1', 'true', or 'yes' to enable logging
+        DBX_PATCH_LOG_LEVEL: Set to DEBUG, INFO, WARNING, ERROR, or CRITICAL (default: ERROR)
     """
 
     def __init__(self, name: str = "dbx-patch") -> None:
@@ -47,10 +51,15 @@ def __init__(self, name: str = "dbx-patch") -> None:
             # Add console handler
             handler = logging.StreamHandler(sys.stdout)
             handler.setLevel(level)
-            formatter = logging.Formatter("%(message)s")
-            handler.setFormatter(formatter)
+
+            if self._logger.hasHandlers():
+                self._logger.handlers.clear()
+
             self._logger.addHandler(handler)
 
+            # Prevent propagation to root logger to avoid duplicate messages
+            self._logger.propagate = False
+
         self._indent_level = 0
         self._indent_char = "  "