From f12d646e3fd4a9db3f2d9058a25b29ec9757c1f6 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Wed, 19 Feb 2025 16:17:22 +0000 Subject: let read_s3 handle URI parsing instead of main() in __init__.py --- obfuscator/__init__.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) (limited to 'obfuscator') 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) -- cgit v1.2.3