From e476403c59e922512d06ef5489a07fba87615998 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Wed, 26 Feb 2025 16:13:36 +0000 Subject: add ability to read path & pii fields from json strings as input --- gdpr_obfuscator/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'gdpr_obfuscator/__init__.py') diff --git a/gdpr_obfuscator/__init__.py b/gdpr_obfuscator/__init__.py index 9e2d837..7466153 100644 --- a/gdpr_obfuscator/__init__.py +++ b/gdpr_obfuscator/__init__.py @@ -9,11 +9,13 @@ class Obfuscator: self.reader = DataReader() self.utils = Utilities() - def process_s3(self, path: str, pii_fields: List[str]) -> bytes: + def process_s3(self, input: str) -> bytes: + path, pii_fields = self.utils.process_json_input(input) obfuscated_data = obfuscate_data(self.reader.read_s3(path), pii_fields) return self.utils.create_byte_stream(obfuscated_data) - def process_local(self, path: str, pii_fields: List[str]) -> bytes: + def process_local(self, input: str) -> bytes: + path, pii_fields = self.utils.process_json_input(input) obfuscated_data = obfuscate_data(self.reader.read_local(path), pii_fields) return self.utils.create_byte_stream(obfuscated_data) -- cgit v1.2.3