diff options
Diffstat (limited to 'obfuscator/utils.py')
| -rw-r--r-- | obfuscator/utils.py | 21 |
1 files changed, 11 insertions, 10 deletions
diff --git a/obfuscator/utils.py b/obfuscator/utils.py index 2e4211f..77ca1cf 100644 --- a/obfuscator/utils.py +++ b/obfuscator/utils.py @@ -1,15 +1,16 @@ # Utility functions from obfuscator.logger import get_logger -# Create the logger -logger = get_logger("CLI") +class Utilities: + def __init__(self, logger=None): + self.logger = get_logger("UTILITIES", logger) -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 + 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 |