diff options
| -rw-r--r-- | obfuscator/__init__.py | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/obfuscator/__init__.py b/obfuscator/__init__.py index 98e309f..2ee3b96 100644 --- a/obfuscator/__init__.py +++ b/obfuscator/__init__.py @@ -1,6 +1,5 @@ from obfuscator.read import DataReader from obfuscator.write import DataWriter -from obfuscator.utils import Utilities from obfuscator.obfuscate import obfuscate from typing import List @@ -8,8 +7,6 @@ from typing import List def main(s3_source: str, pii_fields: List[str], log_level: str = "INFO") -> bytes: reader = DataReader(log_level) writer = DataWriter() - utilities = Utilities() - source = utilities.get_source(s3_source) - data = reader.read_string(source) + data = reader.read_s3(s3_source) obfuscated_data = obfuscate(data, pii_fields) return writer.create_byte_stream(obfuscated_data) |