add error handling for when PII field is missing from data

1 files changed, 7 insertions, 0 deletions

diff --git a/gdpr_obfuscator/obfuscate.py b/gdpr_obfuscator/obfuscate.py
index 0a1d74b..722b750 100644
--- a/gdpr_obfuscator/obfuscate.py
+++ b/gdpr_obfuscator/obfuscate.py
@@ -9,6 +9,13 @@ def obfuscate_data(
     sensitive values with a string of asterisks.
     """
 
+    for record in data:
+        missing_pii_fields = [field for field in pii_fields if field not in record]
+        if missing_pii_fields:
+            raise ValueError(
+                f"A record is missing required PII fields: {missing_pii_fields}"
+            )
+
     return [
         {k: ("***" if k in pii_fields else v) for k, v in record.items()}
         for record in data