From 6e17fcd7a99f6405dad105d83381f80e5951fc83 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Fri, 21 Feb 2025 02:19:24 +0000 Subject: update funcion/method names & add missing parameters --- examples/local.py | 6 ++++-- gdpr_obfuscator/__init__.py | 10 +++++----- gdpr_obfuscator/obfuscate.py | 2 +- 3 files changed, 10 insertions(+), 8 deletions(-) diff --git a/examples/local.py b/examples/local.py index d41dfa3..660146d 100644 --- a/examples/local.py +++ b/examples/local.py @@ -2,8 +2,10 @@ from gdpr_obfuscator import Obfuscator obfuscator = Obfuscator() + def main(): - return obfuscator.local(location="../mock_data.csv", pii_fields=["name", "email"]) + print(obfuscator.process_local(path="mock_data.csv", pii_fields=["name", "email"])) + if __name__ == "__main__": - main() \ No newline at end of file + main() 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) diff --git a/gdpr_obfuscator/obfuscate.py b/gdpr_obfuscator/obfuscate.py index 1b6cf0c..0a1d74b 100644 --- a/gdpr_obfuscator/obfuscate.py +++ b/gdpr_obfuscator/obfuscate.py @@ -1,7 +1,7 @@ from typing import List, Dict -def obfuscate( +def obfuscate_data( data: List[Dict[str, str]], pii_fields: List[str] ) -> List[Dict[str, str]]: """ -- cgit v1.2.3