Fix circular dependency in logging.py

Author: kirkbrauer

packages/jumpstarter/jumpstarter/exporter/logging.py

@@ -2,15 +2,12 @@
 from collections import deque
 from contextlib import contextmanager
 from threading import RLock
-from typing import TYPE_CHECKING
 
 from jumpstarter_protocol import jumpstarter_pb2
 
+from .logging_protocol import LoggerRegistration
 from jumpstarter.common import LogSource
 
-if TYPE_CHECKING:
-    from .session import Session
-
 
 class LogHandler(logging.Handler):
     def __init__(self, queue: deque, source: LogSource = LogSource.UNSPECIFIED):
@@ -69,7 +66,9 @@ def context_log_source(self, logger_name: str, source: LogSource):
             self.remove_child_handler(logger_name)
 
 
-def get_logger(name: str, source: LogSource = LogSource.SYSTEM, session: "Session" = None) -> logging.Logger:
+def get_logger(
+    name: str, source: LogSource = LogSource.SYSTEM, session: LoggerRegistration | None = None
+) -> logging.Logger:
     """
     Get a logger with automatic LogSource mapping.
 

packages/jumpstarter/jumpstarter/exporter/logging_protocol.py

@@ -0,0 +1,22 @@
+"""Protocol for logger registration to avoid circular dependencies."""
+
+from typing import Protocol
+
+from jumpstarter.common import LogSource
+
+
+class LoggerRegistration(Protocol):
+    """Protocol for objects that can register logger sources.
+
+    This protocol defines the interface for objects that can associate
+    logger names with log sources, enabling proper routing of log messages.
+    """
+
+    def add_logger_source(self, logger_name: str, source: LogSource) -> None:
+        """Register a logger name with its corresponding log source.
+
+        Args:
+            logger_name: Name of the logger to register
+            source: The log source category for this logger
+        """
+        ...