nit logger fix

Author: caroljung-cg

src/codegen/runner/servers/local_daemon.py

@@ -16,15 +16,15 @@
 )
 from codegen.runner.models.codemod import Codemod, CodemodRunResult
 from codegen.runner.sandbox.runner import SandboxRunner
-from codegen.shared.logging.get_logger import get_colored_logger
+from codegen.shared.logging.get_logger import get_logger
 
 # Configure logging at module level
 logging.basicConfig(
     level=logging.INFO,
     format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
     force=True,
 )
-logger = get_colored_logger(__name__)
+logger = get_logger(__name__)
 
 server_info: ServerInfo
 runner: SandboxRunner

src/codegen/sdk/codebase/codebase_context.py

@@ -31,6 +31,7 @@
 from codegen.sdk.typescript.external.ts_declassify.ts_declassify import TSDeclassify
 from codegen.shared.enums.programming_language import ProgrammingLanguage
 from codegen.shared.exceptions.control_flow import StopCodemodException
+from codegen.shared.logging.get_logger import get_logger
 from codegen.shared.performance.stopwatch_utils import stopwatch, stopwatch_with_sentry
 
 if TYPE_CHECKING:
@@ -51,9 +52,7 @@
     from codegen.sdk.core.node_id_factory import NodeId
     from codegen.sdk.core.parser import Parser
 
-import logging
-
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 # src/vs/platform/contextview/browser/contextMenuService.ts is ignored as there is a parsing error with tree-sitter

src/codegen/shared/logging/get_logger.py

@@ -3,7 +3,7 @@
 import colorlog
 
 
-def get_colored_logger(name: str) -> logging.Logger:
+def get_logger(name: str) -> logging.Logger:
     formatter = colorlog.ColoredFormatter(
         "%(white)s%(asctime)s - %(name)s - %(log_color)s%(levelname)s%(reset)s%(white)s - %(message_log_color)s%(message)s",
         log_colors={
@@ -26,9 +26,9 @@ def get_colored_logger(name: str) -> logging.Logger:
     logger = logging.getLogger(name)
     if logger.hasHandlers():
         for h in logger.handlers:
-            h.setFormatter(formatter)
-    else:
-        handler = colorlog.StreamHandler()
-        handler.setFormatter(formatter)
-        logger.addHandler(handler)
+            logger.removeHandler(h)
+
+    handler = colorlog.StreamHandler()
+    handler.setFormatter(formatter)
+    logger.addHandler(handler)
     return logger