add function for main process log

jerry wang · jerry wang · commit ddda9e01d2c7 · 2023-06-17T17:41:55.000+08:00
diff --git a/OTHERS/MultiprocessingLogger/logger.py b/OTHERS/MultiprocessingLogger/logger.py
@@ -15,6 +15,7 @@ def __init__(self) -> None:
             target=self._process_target,
         )
         self.process.start()
+        self.redirect_main_process_log_to_queue()
 
     def get_log_file_path(self) -> str:
         # TODO: make this configurable
@@ -23,6 +24,22 @@ def get_log_file_path(self) -> str:
         log_file_path = os.path.join(log_file_dir, 'multip.log')
         return log_file_path
 
+    def redirect_main_process_log_to_queue(self):
+
+        main_process_id = os.getpid()
+
+        def filter(record: logging.LogRecord):
+            if record.process == main_process_id:
+                self.queue.put_nowait(record)
+            return None
+
+        root = logging.getLogger()
+        root.setLevel(logging.INFO)
+
+        handler = logging.Handler()
+        handler.addFilter(filter)
+        root.addHandler(handler)
+
     def init_logger_configure(self):
         
         root = logging.getLogger()
