diff options
| -rw-r--r-- | gdpr_obfuscator/obfuscate.py | 7 |
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 |