bdd-trace 0.5.3__py3-none-any.whl → 0.5.4__py3-none-any.whl

bdd_trace/log.py

@@ -28,10 +28,14 @@ def setup_logging(
     root_logger.handlers = []
 
     _log_queue = Queue(1000)
+    log_process_handlers = extra_handlers or []
+
     for module in modules:
-        _setup_module_logger(level, module)
+        origin_handlers = _setup_module_logger(level, module)
+        for handler in origin_handlers:
+            handler.addFilter(ModulePrefixFilter([module]))
+            log_process_handlers.append(handler)
 
-    log_process_handlers = extra_handlers or []
     if enable_console_log:
         format = trace_id_format if profile != Profile.NO_TRACE else no_trace_id_format
         console_handler = _create_console_handler(modules, format)
@@ -52,26 +56,28 @@ def stop_logging() -> None:
     _log_queue = None
 
 
-def _create_console_handler(modules: list[str], format: str) -> Handler:
-    class ModulePrefixFilter:
-        def __init__(self, module_prefixes: list[str]):
-            self.module_prefixes: set[str] = set(module_prefixes)
-            self.module_prefixes_with_dot: tuple[str, ...] = tuple(f"{prefix}." for prefix in module_prefixes)
+class ModulePrefixFilter:
+    def __init__(self, module_prefixes: list[str]):
+        self.module_prefixes: set[str] = set(module_prefixes)
+        self.module_prefixes_with_dot: tuple[str, ...] = tuple(f"{prefix}." for prefix in module_prefixes)
+
+    def filter(self, record: LogRecord) -> bool:
+        return record.name in self.module_prefixes or record.name.startswith(self.module_prefixes_with_dot)
 
-        def filter(self, record: LogRecord) -> bool:
-            return record.name in self.module_prefixes or record.name.startswith(self.module_prefixes_with_dot)
 
+class SingleLineFormatter(Formatter):
+    def formatMessage(self, record: LogRecord) -> str:
+        return super().formatMessage(record).replace("\n", " ").replace("\r", "")
+
+
+def _create_console_handler(modules: list[str], format: str) -> Handler:
     handler = StreamHandler(sys.stderr)
     handler.setFormatter(Formatter(format))
     handler.addFilter(ModulePrefixFilter(modules))
     return handler
 
 
-def _setup_module_logger(level: str | int, module: str) -> None:
-    class SingleLineFormatter(Formatter):
-        def formatMessage(self, record: LogRecord) -> str:
-            return super().formatMessage(record).replace("\n", " ").replace("\r", "")
-
+def _setup_module_logger(level: str | int, module: str) -> list[Handler]:
     logger = getLogger(module)
     logger.setLevel(level)
     logger.propagate = False
@@ -81,4 +87,6 @@ def _setup_module_logger(level: str | int, module: str) -> None:
         raise RuntimeError("log queue is not initialized")
     handler = QueueHandler(queue)
     handler.setFormatter(SingleLineFormatter())
+    origin_handlers = logger.handlers
     logger.handlers = [handler]
+    return origin_handlers or []

{bdd_trace-0.5.3.dist-info → bdd_trace-0.5.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: bdd-trace
-Version: 0.5.3
+Version: 0.5.4
 Summary: BDD项目日志和追踪SDK
 Author: 蔡涛
 Author-email: 蔡涛 <caitao@zhejianglab.cn>

bdd_trace-0.5.4.dist-info/RECORD

@@ -0,0 +1,7 @@
+bdd_trace/__init__.py,sha256=X4Nx6VppPfXXXEe5oC_foryalbOk5f9HLOi50Nat8mY,154
+bdd_trace/log.py,sha256=HqkUiomCJlR3OsbHMSOYFe4mt5WitYcFhRqBmK9pBPM,3136
+bdd_trace/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bdd_trace/trace.py,sha256=wmzwkcHtV7CslLFKzpIFjSv-55WsqX28uPfsgYDvumA,3784
+bdd_trace-0.5.4.dist-info/WHEEL,sha256=Jb20R3Ili4n9P1fcwuLup21eQ5r9WXhs4_qy7VTrgPI,79
+bdd_trace-0.5.4.dist-info/METADATA,sha256=IuseK2ZM3uoiF4UMT2EXohHzYngc0gq8-L62CC6MTj4,1585
+bdd_trace-0.5.4.dist-info/RECORD,,

bdd_trace-0.5.3.dist-info/RECORD

@@ -1,7 +0,0 @@
-bdd_trace/__init__.py,sha256=X4Nx6VppPfXXXEe5oC_foryalbOk5f9HLOi50Nat8mY,154
-bdd_trace/log.py,sha256=CmcTloY8F6DxxxvsJr0nuuHVk4Wkhr7udvKTVF4_xZk,2922
-bdd_trace/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bdd_trace/trace.py,sha256=wmzwkcHtV7CslLFKzpIFjSv-55WsqX28uPfsgYDvumA,3784
-bdd_trace-0.5.3.dist-info/WHEEL,sha256=Jb20R3Ili4n9P1fcwuLup21eQ5r9WXhs4_qy7VTrgPI,79
-bdd_trace-0.5.3.dist-info/METADATA,sha256=CSTmHsSBwYcZKPLdsK7ECN4E35JM1f2X4Hn9edIcuNU,1585
-bdd_trace-0.5.3.dist-info/RECORD,,