mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-02-19 02:54:28 -05:00
ef7cfbb860
Restructuring the Repo to make it clear the difference between classic autogpt and the autogpt platform: * Move the "classic" projects `autogpt`, `forge`, `frontend`, and `benchmark` into a `classic` folder * Also rename `autogpt` to `original_autogpt` for absolute clarity * Rename `rnd/` to `autogpt_platform/` * `rnd/autogpt_builder` -> `autogpt_platform/frontend` * `rnd/autogpt_server` -> `autogpt_platform/backend` * Adjust any paths accordingly
96 lines
3.4 KiB
Python
96 lines
3.4 KiB
Python
import logging
|
|
|
|
from colorama import Fore, Style
|
|
from google.cloud.logging_v2.handlers import CloudLoggingFilter, StructuredLogHandler
|
|
|
|
from .utils import remove_color_codes
|
|
|
|
|
|
class FancyConsoleFormatter(logging.Formatter):
|
|
"""
|
|
A custom logging formatter designed for console output.
|
|
|
|
This formatter enhances the standard logging output with color coding. The color
|
|
coding is based on the level of the log message, making it easier to distinguish
|
|
between different types of messages in the console output.
|
|
|
|
The color for each level is defined in the LEVEL_COLOR_MAP class attribute.
|
|
"""
|
|
|
|
# level -> (level & text color, title color)
|
|
LEVEL_COLOR_MAP = {
|
|
logging.DEBUG: Fore.LIGHTBLACK_EX,
|
|
logging.INFO: Fore.BLUE,
|
|
logging.WARNING: Fore.YELLOW,
|
|
logging.ERROR: Fore.RED,
|
|
logging.CRITICAL: Fore.RED + Style.BRIGHT,
|
|
}
|
|
|
|
def format(self, record: logging.LogRecord) -> str:
|
|
# Make sure `msg` is a string
|
|
if not hasattr(record, "msg"):
|
|
record.msg = ""
|
|
elif not type(record.msg) is str:
|
|
record.msg = str(record.msg)
|
|
|
|
# Determine default color based on error level
|
|
level_color = ""
|
|
if record.levelno in self.LEVEL_COLOR_MAP:
|
|
level_color = self.LEVEL_COLOR_MAP[record.levelno]
|
|
record.levelname = f"{level_color}{record.levelname}{Style.RESET_ALL}"
|
|
|
|
# Determine color for message
|
|
color = getattr(record, "color", level_color)
|
|
color_is_specified = hasattr(record, "color")
|
|
|
|
# Don't color INFO messages unless the color is explicitly specified.
|
|
if color and (record.levelno != logging.INFO or color_is_specified):
|
|
record.msg = f"{color}{record.msg}{Style.RESET_ALL}"
|
|
|
|
return super().format(record)
|
|
|
|
|
|
class ForgeFormatter(FancyConsoleFormatter):
|
|
def __init__(self, *args, no_color: bool = False, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.no_color = no_color
|
|
|
|
def format(self, record: logging.LogRecord) -> str:
|
|
# Make sure `msg` is a string
|
|
if not hasattr(record, "msg"):
|
|
record.msg = ""
|
|
elif not type(record.msg) is str:
|
|
record.msg = str(record.msg)
|
|
|
|
# Strip color from the message to prevent color spoofing
|
|
if record.msg and not getattr(record, "preserve_color", False):
|
|
record.msg = remove_color_codes(record.msg)
|
|
|
|
# Determine color for title
|
|
title = getattr(record, "title", "")
|
|
title_color = getattr(record, "title_color", "") or self.LEVEL_COLOR_MAP.get(
|
|
record.levelno, ""
|
|
)
|
|
if title and title_color:
|
|
title = f"{title_color + Style.BRIGHT}{title}{Style.RESET_ALL}"
|
|
# Make sure record.title is set, and padded with a space if not empty
|
|
record.title = f"{title} " if title else ""
|
|
|
|
if self.no_color:
|
|
return remove_color_codes(super().format(record))
|
|
else:
|
|
return super().format(record)
|
|
|
|
|
|
class StructuredLoggingFormatter(StructuredLogHandler, logging.Formatter):
|
|
def __init__(self):
|
|
# Set up CloudLoggingFilter to add diagnostic info to the log records
|
|
self.cloud_logging_filter = CloudLoggingFilter()
|
|
|
|
# Init StructuredLogHandler
|
|
super().__init__()
|
|
|
|
def format(self, record: logging.LogRecord) -> str:
|
|
self.cloud_logging_filter.filter(record)
|
|
return super().format(record)
|