diff options
Diffstat (limited to 'obfuscator/obfuscate.py')
| -rw-r--r-- | obfuscator/obfuscate.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/obfuscator/obfuscate.py b/obfuscator/obfuscate.py new file mode 100644 index 0000000..ac0bd21 --- /dev/null +++ b/obfuscator/obfuscate.py @@ -0,0 +1,17 @@ +from typing import List, Dict +from obfuscator.logger import get_logger + +logger = get_logger("Obfuscator") + + +def obfuscate( + data: List[Dict[str, str]], pii_fields: List[str] +) -> List[Dict[str, str]]: + if not data: + logger.info("No valid data was provided to obfuscate") + return [] + + return [ + {k: ("***" if k in pii_fields else v) for k, v in record.items()} + for record in data + ] |
