73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
import logging
|
|
from loki_logger_handler.loki_logger_handler import LokiLoggerHandler
|
|
from uuid import uuid4, UUID
|
|
|
|
LOG_CONSOLE = 1
|
|
LOG_FILE = 2
|
|
LOG_LOKI = 4
|
|
run_id: UUID
|
|
|
|
# Quick non-intrusive debug check for sk.devops logger
|
|
def setup(name: str, handlers: int) -> logging.Logger:
|
|
global run_id
|
|
# Generate a unique run ID.
|
|
run_id = uuid4()
|
|
|
|
logger = logging.getLogger(name)
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
fmt = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s [%(url)]')
|
|
|
|
if handlers & LOG_CONSOLE:
|
|
console = logging.StreamHandler()
|
|
console.setLevel(logging.INFO)
|
|
console.setFormatter(fmt)
|
|
logger.addHandler(console)
|
|
|
|
if handlers & LOG_FILE:
|
|
file = logging.FileHandler('devops_debug.log')
|
|
file.setLevel(logging.DEBUG)
|
|
file.setFormatter(fmt)
|
|
logger.addHandler(file)
|
|
|
|
if handlers & LOG_LOKI:
|
|
# Create an instance of the custom handler
|
|
custom_handler = LokiLoggerHandler(
|
|
url="https://loki.koszewscy.waw.pl/loki/api/v1/push",
|
|
labels={"application": "docs-harverster"},
|
|
label_keys={"http_method": "http_method"},
|
|
timeout=10,
|
|
)
|
|
logger.addHandler(custom_handler)
|
|
|
|
return logger
|
|
|
|
def debug(logger: logging.Logger, message: str, **kwargs):
|
|
global run_id
|
|
logger.debug(message, extra={"run_id": str(run_id), **kwargs})
|
|
|
|
def info(logger: logging.Logger, message: str, **kwargs):
|
|
global run_id
|
|
logger.info(message, extra={"run_id": str(run_id), **kwargs})
|
|
|
|
def warning(logger: logging.Logger, message: str, **kwargs):
|
|
global run_id
|
|
logger.warning(message, extra={"run_id": str(run_id), **kwargs})
|
|
|
|
def error(logger: logging.Logger, message: str, **kwargs):
|
|
global run_id
|
|
logger.error(message, extra={"run_id": str(run_id), **kwargs})
|
|
|
|
def critical(logger: logging.Logger, message: str, **kwargs):
|
|
global run_id
|
|
logger.critical(message, extra={"run_id": str(run_id), **kwargs})
|
|
|
|
def log_entity_creation(logger: logging.Logger, entity_class: type, entity_key: str):
|
|
global run_id
|
|
logger.debug(f'Created new "{entity_class.__name__}" object with key: "{entity_key}"',
|
|
extra={
|
|
"entity_class": entity_class.__name__,
|
|
"entity_key": entity_key,
|
|
"run_id": str(run_id)
|
|
})
|