configure logging

Author: caroljung-cg

pyproject.toml

@@ -74,6 +74,7 @@ dependencies = [
   "httpx>=0.28.1",
   "docker>=6.1.3",
   "urllib3>=2.0.0",
+  "colorlog>=6.9.0",
 ]
 
 license = { text = "Apache-2.0" }

src/codegen/git/repo_operator/repo_operator.py

@@ -113,16 +113,20 @@ def viz_file_path(self) -> str:
         return os.path.join(self.viz_path, "graph.json")
 
     def _set_bot_email(self, git_cli: GitCLI) -> None:
+        logging.info(f"****** Setting bot email to {CODEGEN_BOT_EMAIL} ******")
         with git_cli.config_writer("repository") as writer:
             if not writer.has_section("user"):
                 writer.add_section("user")
             writer.set("user", "email", CODEGEN_BOT_EMAIL)
+            logging.info(f"****** [DONE] Setting bot email to {CODEGEN_BOT_EMAIL} ******")
 
     def _set_bot_username(self, git_cli: GitCLI) -> None:
+        logging.info(f"****** Setting bot USERNAME to {CODEGEN_BOT_NAME} ******")
         with git_cli.config_writer("repository") as writer:
             if not writer.has_section("user"):
                 writer.add_section("user")
             writer.set("user", "name", CODEGEN_BOT_NAME)
+            logging.info(f"****** [DONE] Setting bot USERNAME to {CODEGEN_BOT_NAME} ******")
 
     def _unset_bot_email(self, git_cli: GitCLI) -> None:
         with git_cli.config_writer("repository") as writer:
@@ -136,6 +140,7 @@ def _unset_bot_username(self, git_cli: GitCLI) -> None:
 
     @cached_property
     def git_cli(self) -> GitCLI:
+        logger.info("**** Initializing git_cli!!!*****")
         git_cli = GitCLI(self.repo_path)
         username = None
         user_level = None

src/codegen/runner/servers/local_daemon.py

@@ -1,4 +1,3 @@
-import logging
 import os
 from contextlib import asynccontextmanager
 
@@ -16,14 +15,9 @@
 from codegen.runner.models.codemod import Codemod, CodemodRunResult
 from codegen.runner.sandbox.runner import SandboxRunner
 from codegen.shared.enums.programming_language import ProgrammingLanguage
+from codegen.shared.logging.get_logger import get_logger
 
-logging.basicConfig(
-    level=logging.INFO,
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-    force=True,
-)
-
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 server_info: ServerInfo
 runner: SandboxRunner

src/codegen/shared/logging/get_logger.py

@@ -0,0 +1,33 @@
+import logging
+
+import colorlog
+
+handler = colorlog.StreamHandler()
+handler.setFormatter(
+    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={
+            "DEBUG": "cyan",
+            "INFO": "green",
+            "WARNING": "yellow",
+            "ERROR": "red",
+            "CRITICAL": "red,bg_white",
+        },
+        secondary_log_colors={
+            "message": {
+                "DEBUG": "cyan",
+                "INFO": "blue",
+                "WARNING": "yellow",
+                "ERROR": "red",
+                "CRITICAL": "red,bg_white",
+            }
+        },
+    )
+)
+
+
+def get_logger(name: str, level: int = logging.INFO) -> logging.Logger:
+    logger = logging.getLogger(name)
+    logger.addHandler(handler)
+    logger.setLevel(level)
+    return logger