iter: bind internal context for logging

maxray/__init__.py

@@ -12,27 +12,12 @@
 
 import os
 
-from loguru import logger
+from loguru import logger as _logger
+from .logging import logger
 
 if not os.environ.get("MAXRAY_LOG_LEVEL"):
     # Avoid logspam for users of the library
-    logger.disable("maxray")
-
-
-def _set_logging(enabled: bool):
-    if enabled:
-        logger.enable("maxray")
-    else:
-        logger.disable("maxray")
-
-
-@contextmanager
-def _with_logging():
-    try:
-        _set_logging(True)
-        yield
-    finally:
-        _set_logging(False)
+    _logger.disable("maxray")
 
 
 def transform(writer):

maxray/function_store.py

@@ -13,7 +13,7 @@
 from pathlib import Path
 from typing import Any, Optional
 
-from loguru import logger
+from .logging import logger
 
 
 method_wrapper = type(object().__init__)

maxray/logging.py

@@ -0,0 +1,3 @@
+from loguru import logger as _logger
+
+logger = _logger.bind(maxray_internal=True)

maxray/nodes.py

@@ -8,7 +8,7 @@
 
 from typing import Any, Callable, Optional
 
-from loguru import logger
+from .logging import logger
 
 
 @dataclass
@@ -21,7 +21,6 @@ class FnContext:
     line_offset: int
     "Line number in `source_file` at which the function definition begins."
 
-    call_count: ContextVar[int]
     compile_id: str
 
     def __repr__(self):