xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-05 21:30:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add support for analytics file

Browse Source
This commit is contained in:
Ori Kotek
2024-02-22 13:21:56 +02:00
parent e3dba12fea
commit 4ebac16ff7
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
pr_agent/log/__init__.py
View File

@ -5,6 +5,8 @@ from enum import Enum
from loguru import logger from loguru import logger
from pr_agent.config_loader import get_settings
class LoggingFormat(str, Enum): class LoggingFormat(str, Enum):
CONSOLE = "CONSOLE" CONSOLE = "CONSOLE"
@ -15,6 +17,10 @@ def json_format(record: dict) -> str:
return record["message"] return record["message"]
def analytics_filter(record: dict) -> bool:
return record.get("extra", {}).get("analytics", False)
def setup_logger(level: str = "INFO", fmt: LoggingFormat = LoggingFormat.CONSOLE): def setup_logger(level: str = "INFO", fmt: LoggingFormat = LoggingFormat.CONSOLE):
level: int = logging.getLevelName(level.upper()) level: int = logging.getLevelName(level.upper())
if type(level) is not int: if type(level) is not int:
@ -33,6 +39,17 @@ def setup_logger(level: str = "INFO", fmt: LoggingFormat = LoggingFormat.CONSOLE
logger.remove(None) logger.remove(None)
logger.add(sys.stdout, level=level, colorize=True) logger.add(sys.stdout, level=level, colorize=True)
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,
)
return logger return logger