From dd4fcebceccb750424f13e8217740907e1e9f8c9 Mon Sep 17 00:00:00 2001
From: Alex Schofield <git@ajschof.me>
Date: Sun, 2 Mar 2025 18:23:01 +0000
Subject: add error handling for when PII field is missing from data

---
 gdpr_obfuscator/obfuscate.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'gdpr_obfuscator/obfuscate.py')

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
-- 
cgit v1.2.3