update funcion/method names & add missing parameters

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)