Skip to content

Commit 2dc160c

Browse files
committed
dedupe logs
1 parent 05abe0f commit 2dc160c

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/codegen/shared/logging/get_logger.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,10 @@
33
import colorlog
44

55

6-
def get_logger(name: str) -> logging.Logger:
6+
def get_logger(name: str, level: int = logging.INFO) -> logging.Logger:
7+
# Force configure the root logger with a NullHandler to prevent duplicate logs
8+
logging.basicConfig(handlers=[logging.NullHandler()], force=True)
9+
710
formatter = colorlog.ColoredFormatter(
811
"%(white)s%(asctime)s - %(name)s - %(log_color)s%(levelname)s%(reset)s%(white)s - %(message_log_color)s%(message)s",
912
log_colors={
@@ -31,4 +34,8 @@ def get_logger(name: str) -> logging.Logger:
3134
handler = colorlog.StreamHandler()
3235
handler.setFormatter(formatter)
3336
logger.addHandler(handler)
37+
# Ensure the logger propagates to the root logger
38+
logger.propagate = False
39+
# Set the level on the logger itself
40+
logger.setLevel(level)
3441
return logger

0 commit comments

Comments
 (0)