darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 01:41:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

improve email masking

Browse Source
This commit is contained in:
Kenneth Kehl
2024-06-13 10:27:04 -07:00
parent 2a005076d0
commit 1c697202f3
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
notifications_utils/logging.py
View File

@@ -138,7 +138,9 @@ class PIIFilter(logging.Filter):
) # ['alice@google.com', 'bob@abc.com'] ) # ['alice@google.com', 'bob@abc.com']
for email in emails: for email in emails:
# do something with each found email string # do something with each found email string
msg = msg.replace(email, f"XXXXX{email[-10:]}") email_parts = email.split("@")
masked_email = f"{email_parts[0][0:3]}XXX@{email_parts[1][0:7]}XXX"
msg = msg.replace(email, masked_email)
return msg return msg
def filter(self, record): def filter(self, record):

2
tests/notifications_utils/test_logging.py
View File

@@ -62,5 +62,5 @@ def test_pii_filter():
args=None, args=None,
) )
pii_filter = logging.PIIFilter() pii_filter = logging.PIIFilter()
clean_msg = "phone1: 1XXXXX55555, phone2: 1XXXXX55554, email1: XXXXXe@fake.gov, email2: XXXXX2.fake.gov" clean_msg = "phone1: 1XXXXX55555, phone2: 1XXXXX55554, email1: fakXXX@fake.goXXX, email2: fakXXX@fake2.fXXX"
assert pii_filter.filter(record).msg == clean_msg assert pii_filter.filter(record).msg == clean_msg