diff options
Diffstat (limited to 'gdpr_obfuscator/utils.py')
| -rw-r--r-- | gdpr_obfuscator/utils.py | 31 |
1 files changed, 1 insertions, 30 deletions
diff --git a/gdpr_obfuscator/utils.py b/gdpr_obfuscator/utils.py index 9283d9e..c8aadb2 100644 --- a/gdpr_obfuscator/utils.py +++ b/gdpr_obfuscator/utils.py @@ -1,50 +1,21 @@ import csv import io -import logging from enum import Enum from typing import List, Dict class Utilities: - class LogLevel(Enum): - DEBUG = logging.DEBUG - INFO = logging.INFO - WARNING = logging.WARNING - ERROR = logging.ERROR - CRITICAL = logging.CRITICAL - - @staticmethod - def get_logger(name: str, level: "Utilities.LogLevel" = None) -> logging.Logger: - level = level or Utilities.LogLevel.INFO - logger = logging.getLogger(name) - if logger.hasHandlers(): - logger.handlers.clear() - - handler = logging.StreamHandler() - logger.setLevel(level.value) - formatter = logging.Formatter( - "[%(asctime)s] - %(levelname)s::%(name)s - %(message)s" - ) - handler.setFormatter(formatter) - logger.addHandler(handler) - return logger def __init__(self, logger=None): - self.logger = self.get_logger(__name__, logger) + pass def get_s3_path(self, uri): parts = uri.replace("s3://", "").split("/") - self.logger.debug(f"Parts: {parts}") bucket = parts.pop(0) - self.logger.debug(f"Bucket: {bucket}") key = "/".join(parts) - self.logger.debug(f"Key: {key}") return bucket, key def create_byte_stream(self, data: List[Dict[str, str]]) -> bytes: - if not data: - self.logger.error("Invalid or empty data was provided to write") - return b"" output = io.StringIO() |