From 97b345545b950615ebd1df77fd0c1a8a4300495a Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Mon, 17 Feb 2025 12:50:36 +0000 Subject: use list/dictionary comprehension to obfuscate pii data --- obfuscator/obfuscate.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/obfuscator/obfuscate.py b/obfuscator/obfuscate.py index 6c0414e..c9c116c 100644 --- a/obfuscator/obfuscate.py +++ b/obfuscator/obfuscate.py @@ -4,4 +4,12 @@ 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]]: - pass \ No newline at end of file + 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 + ] + \ No newline at end of file -- cgit v1.2.3