You've already forked DataMate
d59c167da4
* feature: 将抽取动作移到每一个算子中 * feature: 落盘算子改为默认执行 * feature: 优化前端展示 * feature: 使用pyproject管理依赖
53 lines
2.2 KiB
Python
53 lines
2.2 KiB
Python
#!/user/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
Description: 电话号码匿名化
|
|
Create: 2024/12/26 15:43
|
|
"""
|
|
import re
|
|
import time
|
|
from typing import Dict, Any
|
|
|
|
from loguru import logger
|
|
|
|
from datamate.core.base_op import Mapper
|
|
|
|
|
|
class AnonymizedPhoneNumber(Mapper):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.phone_re_compile = self.get_phone_re_compile()
|
|
|
|
@staticmethod
|
|
def get_phone_re_compile():
|
|
"""按照格式粗略匹配电话号码，支持以下格式电话号码
|
|
前缀：（0086）、（86）、(0086)、(86) 、无
|
|
电话号码：第一位1，第二位3-9，后续数字可以为0-9，数字按照3-4-4进行间隔，间隔符为空格、-、无
|
|
固定电话号码：0AX-CXXX-XXXX、0BXX-CXXX-XXXX、0BXX-CXX-XXXX A为1-2、B为3-9、C为2-8、X为0-9
|
|
约束：电话号码前后皆为非数字
|
|
"""
|
|
number_prefix = r'([\(（]?\+?(00)?86[)\）]?[- ]?)?'
|
|
cellphone_pattern = r"1[3-9]\d[- ]?\d{4}[- ]?\d{4}"
|
|
landline_pattern = (r'[(（]?(0?[12]\d)[)）]?[ -]?[2-8]\d{3}[ -]?\d{4}'
|
|
r'|[(（]?(0?[3-9]\d{2})[)）]?[ -]?[2-8]\d{2}\d?[ -]?\d{4}')
|
|
phone_numbers_pattern = rf'(?<=[^\d]){number_prefix}({cellphone_pattern}|{landline_pattern})(?=[^\d])'
|
|
phone_re_compile = re.compile(phone_numbers_pattern)
|
|
return phone_re_compile
|
|
|
|
def execute(self, sample: Dict[str, Any]) -> Dict[str, Any]:
|
|
start = time.time()
|
|
self.read_file_first(sample)
|
|
sample[self.text_key] = self._phone_number_filter(sample[self.text_key])
|
|
logger.info(
|
|
f"fileName: {sample[self.filename_key]}, method: PhoneNumberCleaner costs {time.time() - start:6f} s")
|
|
return sample
|
|
|
|
def _phone_number_filter(self, input_data: str):
|
|
""" 电话号码匿名化"""
|
|
# 正则匹配：电话号码前需匹配不是数字的字符串
|
|
# 为避免处于文章开头和结尾的电话号码不可被识别，需要在输入字符串的前后手动加上字符串
|
|
input_data = ''.join(['【', input_data, '】'])
|
|
input_data = self.phone_re_compile.sub("<tel>", input_data)
|
|
return input_data[1:-1]
|