diff options
Diffstat (limited to 'gdpr_obfuscator/__init__.py')
| -rw-r--r-- | gdpr_obfuscator/__init__.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/gdpr_obfuscator/__init__.py b/gdpr_obfuscator/__init__.py index 43c1cc0..38680e8 100644 --- a/gdpr_obfuscator/__init__.py +++ b/gdpr_obfuscator/__init__.py @@ -1,5 +1,5 @@ from .read import DataReader -from .obfuscate import obfuscate +from .obfuscate import obfuscate_data from typing import List, Dict @@ -7,8 +7,8 @@ class Obfuscator: def __init__(self): self.reader = DataReader() - def import_s3(self, path: str, pii_fields: List[str]) -> bytes: - return self.reader.read_s3(path) + def process_s3(self, path: str, pii_fields: List[str]) -> bytes: + return obfuscate_data(self.reader.read_s3(path), pii_fields) - def import_local(self, path: str, pii_fields: List[str]) -> bytes: - return self.reader.read_local(path) + def process_local(self, path: str, pii_fields: List[str]) -> bytes: + return obfuscate_data(self.reader.read_local(path), pii_fields) |