diff options
Diffstat (limited to 'stdout_filterer.py')
-rw-r--r-- | stdout_filterer.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/stdout_filterer.py b/stdout_filterer.py new file mode 100644 index 0000000..c4cf845 --- /dev/null +++ b/stdout_filterer.py @@ -0,0 +1,15 @@ +import re + + +class RedactPhoneNumbers: + def __init__(self, stream): + self.stream = stream + + def write(self, text): + # Regular expression to match phone numbers + phone_regex = r"(\+\d{1,2}\s?)?1?\-?\.?\s?\(?\d{3}\)?[\s.-]?\d{3}[\s.-]?\d{4}" + redacted_text = re.sub(phone_regex, "****", text) + self.stream.write(redacted_text) + + def flush(self): + self.stream.flush() |