darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-02-05 10:53:28 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix email masking algorithm

Browse Source
This commit is contained in:
Kenneth Kehl
2024-06-13 10:17:47 -07:00
parent f6f99b6124
commit 4e8e1acb5b
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

@@ -155,7 +155,9 @@ class PIIFilter(logging.Filter):
) # ['alice@google.com', 'bob@abc.com']
for email in emails:
# 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
def filter(self, record):

2
tests/notifications_utils/test_logging.py
View File

@@ -62,5 +62,5 @@ def test_pii_filter():
args=None,
)
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