workaround/fix annoying logging issues

1 files changed, 15 insertions, 12 deletions

diff --git a/obfuscator/utils.py b/obfuscator/utils.py
index 1d1c3fe..81eb04a 100644
--- a/obfuscator/utils.py
+++ b/obfuscator/utils.py
@@ -1,15 +1,18 @@
 # Utility functions
 from obfuscator.logger import get_logger
 
-# Create the logger
-logger = get_logger("UTILS")
-
-
-def get_s3_path(uri):
-    parts = uri.replace("s3://", "").split("/")
-    logger.debug(f"Parts: {parts}")
-    bucket = parts.pop(0)
-    logger.debug(f"Bucket: {bucket}")
-    key = "/".join(parts)
-    logger.debug(f"Key: {key}")
-    return bucket, key
+
+class Utilities:
+
+    def __init__(self, logger=None):
+        # Create the logger
+        self.logger = get_logger("UTILITIES", logger)
+
+    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