From 72ced380933e7d728a41cdfdc2017eb9dd11103d Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Thu, 20 Feb 2025 23:52:47 +0000 Subject: move logger into utils.py --- gdpr_obfuscator/logger.py | 34 ---------------------------------- 1 file changed, 34 deletions(-) (limited to 'gdpr_obfuscator/logger.py') diff --git a/gdpr_obfuscator/logger.py b/gdpr_obfuscator/logger.py index fe97210..8b13789 100644 --- a/gdpr_obfuscator/logger.py +++ b/gdpr_obfuscator/logger.py @@ -1,35 +1 @@ -import logging -from enum import Enum - -class LogLevel(Enum): - DEBUG = logging.DEBUG - INFO = logging.INFO - WARNING = logging.WARNING - ERROR = logging.ERROR - CRITICAL = logging.CRITICAL - - -def get_logger(name: str, level: LogLevel = LogLevel.INFO) -> logging.Logger: - if isinstance(level, str): - try: - level = LogLevel[level.upper()] - except KeyError: - raise ValueError( - f"Invalid log level '{level}'. Choose from: {', '.join(l.name for l in LogLevel)}" - ) - - logger = logging.getLogger(name) - - if logger.hasHandlers(): - logger.handlers.clear() - - handler = logging.StreamHandler() - logger.setLevel(level.value) - formatting = logging.Formatter( - "[%(asctime)s] - %(levelname)s::%(name)s - %(message)s" - ) - handler.setFormatter(formatting) - logger.addHandler(handler) - - return logger -- cgit v1.2.3