feat(logger): updated the logger utility to be more colorful

Author: MemerGamer

src/core/logger_utility.py

@@ -1,34 +1,48 @@
 from loguru import logger
-from rich.logging import RichHandler
+from rich.traceback import Traceback
+from rich.console import Console
+import io
 import sys
 
 
-# Configure Loguru logger
+def rich_formatter(record):
+    """Rich formatter for loguru logger"""
+    format_ = "<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>\n"
+    if record["exception"] is not None:
+        output = io.StringIO()
+        console = Console(file=output, force_terminal=True)
+        traceback = Traceback.from_exception(*record["exception"])
+        console.print(traceback)
+        record["extra"]["rich_exception"] = output.getvalue()
+        format_ += "{extra[rich_exception]}"
+    return format_
+
+
 def setup_logger():
+    """Setup the logger configuration for the application"""
+
     # Remove default logger configuration
     logger.remove()
 
     # Add a handler for console logging with rich formatting
-    logger.add(
-        sys.stderr, format="{time} {level} {message}", level="INFO", colorize=True
-    )
-    logger.configure(
-        handlers=[
-            {
-                "sink": RichHandler(markup=True),
-                "format": " <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>",
-            }
-        ]
-    )
+    logger.add(sys.stderr, format=rich_formatter, level="INFO", colorize=True)
 
     # Add a handler for file logging
     logger.add("logs/app_{time}.log", rotation="1 week", level="DEBUG")
 
 
 setup_logger()
 
-# Example logging
-logger.info("This is an informational message.")
-logger.debug("This is a debug message.")
-logger.error("This is an error message.")
-logger.warning("This is a warning message.")
+# Example log messages:
+# @logger.catch
+# def divide(a, b):
+#     a / b
+
+
+# divide(1, 0)
+
+# logger.info("This is an info message")
+# logger.debug("This is a debug message")
+# logger.warning("This is a warning message")
+# logger.error("This is an error message")
+# logger.critical("This is a critical message")