backup commit - 21st @ 00:00

3 files changed, 8 insertions, 7 deletions

diff --git a/gdpr_obfuscator/__init__.py b/gdpr_obfuscator/__init__.py
index edfc665..69854cb 100644
--- a/gdpr_obfuscator/__init__.py
+++ b/gdpr_obfuscator/__init__.py
@@ -1,17 +1,18 @@
 from .read import DataReader
 from .obfuscate import obfuscate
-from .logger import get_logger
 from typing import List, Dict
+from .utils import Utilities
 
-class ImportData:
+
+class Obfuscator:
     def __init__(self, verbosity: bool = False):
         self.verbosity = verbosity
         self.log_level = "DEBUG" if verbosity else "INFO"
-        self.logger = get_logger("ImportData", self.log_level)
+        self.logger = Utilities.get_logger("ImportData", self.log_level)
         self.reader = DataReader()
 
     def import_s3(self, path: str, pii_fields: List[str]) -> bytes:
         return self.reader.read_s3(path)
 
     def import_local(self, path: str, pii_fields: List[str]) -> bytes:
-        return self.reader.read_local(path)
\ No newline at end of file
+        return self.reader.read_local(path)
diff --git a/gdpr_obfuscator/obfuscate.py b/gdpr_obfuscator/obfuscate.py
index ac8acaf..6439542 100644
--- a/gdpr_obfuscator/obfuscate.py
+++ b/gdpr_obfuscator/obfuscate.py
@@ -1,7 +1,7 @@
 from typing import List, Dict
-from .logger import get_logger
+from .utils import Utilities
 
-logger = get_logger("OBFUSCATE")
+logger = Utilities.get_logger("OBFUSCATE")
 
 
 def obfuscate(
diff --git a/gdpr_obfuscator/utils.py b/gdpr_obfuscator/utils.py
index fe0ef42..9283d9e 100644
--- a/gdpr_obfuscator/utils.py
+++ b/gdpr_obfuscator/utils.py
@@ -1,7 +1,7 @@
 import csv
 import io
 import logging
-import Enum
+from enum import Enum
 from typing import List, Dict