Fixed logging output routing and added colorized logging (#205, #206)

- Routed info/warning messages to stdout, errors to stderr
- Added ANSI color codes for different log levels
- Implemented ColoredFormatter with specified color scheme

Author: SaurabhCodesAI

scripts/shared.py

@@ -1,6 +1,7 @@
 # Standard library
 import logging
 import os
+import sys
 from datetime import datetime, timezone
 
 # Third-party
@@ -113,12 +114,50 @@ def paths_update(logger, paths, old_quarter, new_quarter):
     return paths
 
 
+class ColoredFormatter(logging.Formatter):
+    """Adds colors to log messages."""
+
+    COLORS = {
+        logging.DEBUG: "\033[90m",  # bright black
+        logging.INFO: "\033[37m",  # white
+        logging.WARNING: "\033[103m",  # bright yellow
+        logging.ERROR: "\033[33m",  # yellow
+        logging.CRITICAL: "\033[31m",  # red
+    }
+    RESET = "\033[0m"
+
+    def format(self, record):
+        message = super().format(record)
+        color = self.COLORS.get(record.levelno, "")
+        if color:
+            return f"{color}{message}{self.RESET}"
+        return message
+
+
 def setup(current_file):
     # Set up logging
-    logging.basicConfig(
-        level=logging.INFO,
-        format="%(asctime)s - %(levelname)s - %(module)s - %(message)s",
+    root = logging.getLogger()
+    root.handlers.clear()
+    root.setLevel(logging.INFO)
+
+    formatter = ColoredFormatter(
+        "%(asctime)s - %(levelname)s - %(module)s - %(message)s"
     )
+
+    # Info/warning to stdout
+    stdout_handler = logging.StreamHandler(sys.stdout)
+    stdout_handler.setLevel(logging.DEBUG)
+    stdout_handler.setFormatter(formatter)
+    stdout_handler.addFilter(lambda r: r.levelno < logging.ERROR)
+
+    # Errors to stderr
+    stderr_handler = logging.StreamHandler(sys.stderr)
+    stderr_handler.setLevel(logging.ERROR)
+    stderr_handler.setFormatter(formatter)
+
+    root.addHandler(stdout_handler)
+    root.addHandler(stderr_handler)
+
     logger = logging.getLogger(__name__)
 
     # Datetime
@@ -220,11 +259,11 @@ def update_readme(
         entry_start_index = lines.index(entry_start_line)
         entry_end_index = lines.index(entry_end_line)
         # Include any trailing empty/whitespace-only lines
-        while entry_end_index + 1 < len(lines):
-            if not lines[entry_end_index + 1].strip():
-                entry_end_index += 1
-            else:
-                break
+        while (
+            entry_end_index + 1 < len(lines)
+            and not lines[entry_end_index + 1].strip()
+        ):
+            entry_end_index += 1
     # Initalize variables of entry is not present
     else:
         entry_start_index = None