Merge pull request #2 from ajschofield/feat/obfuscator

merge obfuscator into stable

1 files changed, 17 insertions, 0 deletions

diff --git a/obfuscator/obfuscate.py b/obfuscator/obfuscate.py
new file mode 100644
index 0000000..ac0bd21
--- /dev/null
+++ b/obfuscator/obfuscate.py
@@ -0,0 +1,17 @@
+from typing import List, Dict
+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]]:
+    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
+    ]