diff options
Diffstat (limited to 'obfuscator/obfuscate.py')
| -rw-r--r-- | obfuscator/obfuscate.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/obfuscator/obfuscate.py b/obfuscator/obfuscate.py index 3f589cb..cd12b6d 100644 --- a/obfuscator/obfuscate.py +++ b/obfuscator/obfuscate.py @@ -1,8 +1,7 @@ from typing import List, Dict from obfuscator.logger import get_logger -# Create the logger -logger = get_logger("Obfuscator") +logger = get_logger("OBFUSCATE") def obfuscate( @@ -12,14 +11,15 @@ def obfuscate( A function to obfuscate PII fields in a list of dictionaries, replacing sensitive values with a string of asterisks. """ - # If no data is provided, log a message and return an empty list if not data: - logger.info("No valid data was provided to obfuscate") + 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 - # Obfuscate the PII fields in each record using a list/dict comprehension - # This code is good but makes debugging a bit tricky. I may consider - # breaking it down into a for loop. return [ {k: ("***" if k in pii_fields else v) for k, v in record.items()} for record in data |