Avoid duplicate logging to stdout (#185)

1kastner · Jul 14, 2023 · d76b797 · d76b797
1 parent 3079e08
commit d76b797
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/conflowgen/logging/logging.py b/conflowgen/logging/logging.py
@@ -59,10 +59,15 @@ def setup_logger(
     logger = logging.getLogger("conflowgen")
     logger.setLevel(logging.DEBUG)
 
-    flow_handler = logging.StreamHandler(stream=sys.stdout)
-    flow_handler.setLevel(logging.DEBUG)
-    flow_handler.setFormatter(formatter)
-    logger.addHandler(flow_handler)
+    stream_handlers = [handler for handler in logger.handlers if isinstance(handler, logging.StreamHandler)]
+    if any(handler.stream == sys.stdout for handler in stream_handlers):
+        logger.warning("Duplicate StreamHandler streaming to sys.stdout detected. "
+                       "Skipping adding another StreamHandler.")
+    else:
+        stream_handler = logging.StreamHandler(stream=sys.stdout)
+        stream_handler.setLevel(logging.DEBUG)
+        stream_handler.setFormatter(formatter)
+        logger.addHandler(stream_handler)
 
     if not os.path.isdir(logging_directory):
         logger.debug(f"Creating log directory at {logging_directory}")