diff options
Diffstat (limited to 'gdpr_obfuscator')
| -rw-r--r-- | gdpr_obfuscator/__init__.py | 9 | ||||
| -rw-r--r-- | gdpr_obfuscator/obfuscate.py | 4 | ||||
| -rw-r--r-- | gdpr_obfuscator/utils.py | 2 |
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 |