From 70a16769761450b9c2aa63cda86a30a73bc0217c Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Thu, 20 Feb 2025 18:27:12 +0000 Subject: update pyproject.toml & references with new src folder name --- gdpr_obfuscator/obfuscate.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 gdpr_obfuscator/obfuscate.py (limited to 'gdpr_obfuscator/obfuscate.py') diff --git a/gdpr_obfuscator/obfuscate.py b/gdpr_obfuscator/obfuscate.py new file mode 100644 index 0000000..cd12b6d --- /dev/null +++ b/gdpr_obfuscator/obfuscate.py @@ -0,0 +1,26 @@ +from typing import List, Dict +from obfuscator.logger import get_logger + +logger = get_logger("OBFUSCATE") + + +def obfuscate( + data: List[Dict[str, str]], pii_fields: List[str] +) -> List[Dict[str, str]]: + """ + A function to obfuscate PII fields in a list of dictionaries, replacing + sensitive values with a string of asterisks. + """ + if not data: + logger.error( + "Invalid or empty data was provided to obfuscate. Returning empty list." + ) + return [] + if not pii_fields: + logger.error("No PII fields provided to obfuscate. Returning data unchanged.") + return data + + return [ + {k: ("***" if k in pii_fields else v) for k, v in record.items()} + for record in data + ] -- cgit v1.2.3