use list/dictionary comprehension to obfuscate pii data

1 files changed, 9 insertions, 1 deletions

diff --git a/obfuscator/obfuscate.py b/obfuscator/obfuscate.py
index 6c0414e..c9c116c 100644
--- a/obfuscator/obfuscate.py
+++ b/obfuscator/obfuscate.py
@@ -4,4 +4,12 @@ from obfuscator.logger import get_logger
 logger = get_logger("Obfuscator")
 
 def obfuscate(data: List[Dict[str, str]], pii_fields: List[str]) -> List[Dict[str,str]]:
-    pass
\ No newline at end of file
+    if not data:
+        logger.info("No valid data was provided to obfuscate")
+        return []
+    
+    return [
+        {k: ("***" if k in pii_fields else v) for k, v in record.items()}
+        for record in data
+    ]
+    
\ No newline at end of file