init datamate

runtime/ops/mapper/phone_number_cleaner/process.py

@@ -0,0 +1,51 @@
+#!/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()
+        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]