log refactor

Author: qwes5s5

fastdeploy/worker/worker_process.py

@@ -27,36 +27,12 @@
 
 import numpy as np
 
-_original_getLogger = logging.getLogger
+from tools.logger_patch import intercept_paddle_loggers
 
-
-def _patched_getLogger(name=None):
-    """Intercept paddle-related logger creation and configure format immediately"""
-    logger = _original_getLogger(name)
-    if name and str(name).startswith("paddle"):
-        # Configure paddle logger immediately
-        formatter = logging.Formatter(
-            "%(levelname)-8s %(asctime)s %(process)-5s %(filename)s[line:%(lineno)d] %(message)s",
-            datefmt="%Y-%m-%d %H:%M:%S",
-        )
-        logger.setLevel(logging.INFO)
-        for handler in logger.handlers[:]:
-            logger.removeHandler(handler)
-        stream_handler = logging.StreamHandler()
-        stream_handler.setFormatter(formatter)
-        logger.addHandler(stream_handler)
-        logger.propagate = False
-    return logger
-
-
-logging.getLogger = _patched_getLogger
-
-import paddle
-import paddle.distributed as dist
-from paddle.distributed import fleet
-
-# Restore original getLogger
-logging.getLogger = _original_getLogger
+with intercept_paddle_loggers():
+    import paddle
+    import paddle.distributed as dist
+    from paddle.distributed import fleet
 
 from fastdeploy import envs
 from fastdeploy.config import (
@@ -104,7 +80,6 @@ def _patched_getLogger(name=None):
 def _configure_third_party_loggers():
     """Unify third-party library (paddleformers, paddle) logger formats.
     Should be called after these libraries are imported."""
-    import logging
 
     # Create standard format (without color)
     formatter = logging.Formatter(

tools/__init__.py

@@ -0,0 +1,46 @@
+"""
+# Copyright (c) 2025  PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""
+
+import logging
+from contextlib import contextmanager
+
+
+@contextmanager
+def intercept_paddle_loggers():
+    """Intercept and configure paddle loggers during import."""
+    _original = logging.getLogger
+
+    def _patched(name=None):
+        logger = _original(name)
+        if name and str(name).startswith("paddle"):
+            formatter = logging.Formatter(
+                "%(levelname)-8s %(asctime)s %(process)-5s %(filename)s[line:%(lineno)d] %(message)s",
+                datefmt="%Y-%m-%d %H:%M:%S",
+            )
+            logger.setLevel(logging.INFO)
+            for handler in logger.handlers[:]:
+                logger.removeHandler(handler)
+            stream_handler = logging.StreamHandler()
+            stream_handler.setFormatter(formatter)
+            logger.addHandler(stream_handler)
+            logger.propagate = False
+        return logger
+
+    logging.getLogger = _patched
+    try:
+        yield
+    finally:
+        logging.getLogger = _original