2023-10-16 14:56:00 +03:00
|
|
|
import json
|
|
|
|
import logging
|
|
|
|
import sys
|
|
|
|
from enum import Enum
|
|
|
|
|
|
|
|
from loguru import logger
|
|
|
|
|
2024-02-22 13:21:56 +02:00
|
|
|
from pr_agent.config_loader import get_settings
|
|
|
|
|
2023-10-16 14:56:00 +03:00
|
|
|
|
|
|
|
class LoggingFormat(str, Enum):
|
|
|
|
CONSOLE = "CONSOLE"
|
|
|
|
JSON = "JSON"
|
|
|
|
|
|
|
|
|
|
|
|
def json_format(record: dict) -> str:
|
|
|
|
return record["message"]
|
|
|
|
|
|
|
|
|
2024-02-22 13:21:56 +02:00
|
|
|
def analytics_filter(record: dict) -> bool:
|
|
|
|
return record.get("extra", {}).get("analytics", False)
|
|
|
|
|
|
|
|
|
2023-10-16 14:56:00 +03:00
|
|
|
def setup_logger(level: str = "INFO", fmt: LoggingFormat = LoggingFormat.CONSOLE):
|
|
|
|
level: int = logging.getLevelName(level.upper())
|
|
|
|
if type(level) is not int:
|
|
|
|
level = logging.INFO
|
|
|
|
|
|
|
|
if fmt == LoggingFormat.JSON:
|
|
|
|
logger.remove(None)
|
|
|
|
logger.add(
|
|
|
|
sys.stdout,
|
|
|
|
level=level,
|
|
|
|
format="{message}",
|
|
|
|
colorize=False,
|
|
|
|
serialize=True,
|
|
|
|
)
|
|
|
|
elif fmt == LoggingFormat.CONSOLE:
|
|
|
|
logger.remove(None)
|
|
|
|
logger.add(sys.stdout, level=level, colorize=True)
|
|
|
|
|
2024-02-22 13:21:56 +02:00
|
|
|
log_file = get_settings().get("CONFIG.ANALYTICS_FILE", "")
|
|
|
|
if log_file:
|
|
|
|
logger.add(
|
|
|
|
log_file,
|
|
|
|
filter=analytics_filter,
|
|
|
|
level=level,
|
|
|
|
format="{message}",
|
|
|
|
colorize=False,
|
|
|
|
serialize=True,
|
|
|
|
)
|
|
|
|
|
2023-10-16 14:56:00 +03:00
|
|
|
return logger
|
|
|
|
|
|
|
|
|
|
|
|
def get_logger(*args, **kwargs):
|
|
|
|
return logger
|