|
|
| import re
|
|
|
| def mask_pii(text):
|
| entities = []
|
| original_text = text
|
|
|
|
|
| patterns = {
|
| "full_name": r"(?:(?:Mr|Ms|Mrs|Dr)\\.?\s)?[A-Z][a-z]+(?:\s[A-Z][a-z]+)+",
|
| "email": r"[\\w\\.-]+@[\\w\\.-]+",
|
| "phone_number": r"(\\+91[-\\s]?)?[6-9]\\d{9}",
|
| "dob": r"(\\d{2}[/-]\\d{2}[/-]\\d{4})",
|
| "aadhar_num": r"\\d{4}\\s\\d{4}\\s\\d{4}",
|
| "credit_debit_no": r"\\d{4}[-\\s]\\d{4}[-\\s]\\d{4}[-\\s]\\d{4}",
|
| "cvv_no": r"\\b\\d{3}\\b",
|
| "expiry_no": r"(0[1-9]|1[0-2])/\\d{2}"
|
| }
|
|
|
| masked_text = text
|
|
|
| for entity, pattern in patterns.items():
|
| for match in re.finditer(pattern, original_text):
|
| start, end = match.span()
|
| matched_text = match.group()
|
|
|
| entities.append({
|
| "position": [start, end],
|
| "classification": entity,
|
| "entity": matched_text
|
| })
|
|
|
| masked_text = masked_text.replace(matched_text, f"[{entity}]", 1)
|
|
|
| return masked_text, entities
|
|
|
| def unmask_pii(masked_text, entities):
|
| unmasked_text = masked_text
|
| for ent in entities:
|
| unmasked_text = unmasked_text.replace(f"[{ent['classification']}]", ent['entity'], 1)
|
| return unmasked_text
|
|
|
