You've already forked DataMate
d59c167da4
* feature: 将抽取动作移到每一个算子中 * feature: 落盘算子改为默认执行 * feature: 优化前端展示 * feature: 使用pyproject管理依赖
76 lines
3.0 KiB
Python
76 lines
3.0 KiB
Python
#!/user/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
Description: 身份证号码匿名化插件
|
|
Create: 2024/12/26 15:43
|
|
"""
|
|
import ipaddress
|
|
import re
|
|
import time
|
|
from typing import Dict, Any
|
|
|
|
from loguru import logger
|
|
|
|
from datamate.core.base_op import Mapper
|
|
|
|
|
|
class AnonymizedIpAddress(Mapper):
|
|
def __init__(self, *args, **kwargs):
|
|
# IP地址校验
|
|
# X.X.X.X与四级目录格式相同，避免误清洗，该格式的IP地址必须匹配 IP/IP地址等字样
|
|
super().__init__(*args, **kwargs)
|
|
self.ipv4_1_and_prefix_pattern = r'ip(地址| address|v4)?( |:|：)*(?<![\.\d])'
|
|
# X.X.X.X
|
|
self.ipv4_pattern = r'(?<![\.\d])\d\.\d\.\d\.\d(?![\.\d])'
|
|
self.ipv4_re_compile = re.compile(r"(?<![\d.])(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})(?![.\d])")
|
|
self.ipv6_re_compile = re.compile(r"(?<![0-9a-fA-F:])(([0-9a-fA-F]{0,4}:)+[0-9a-fA-F]{0,4})(?![0-9a-fA-F:])")
|
|
|
|
@staticmethod
|
|
def verify_ip_address(ip):
|
|
"""验证字符串是否为合法ip地址"""
|
|
try:
|
|
ipaddress.ip_address(ip)
|
|
except ValueError:
|
|
return False
|
|
return True
|
|
|
|
def execute(self, sample: Dict[str, Any]) -> Dict[str, Any]:
|
|
start = time.time()
|
|
self.read_file_first(sample)
|
|
sample[self.text_key] = self._ip_address_filter(sample[self.text_key])
|
|
logger.info(f"fileName: {sample[self.filename_key]}, method: IPAddressCleaner costs {time.time() - start:6f} s")
|
|
return sample
|
|
|
|
def filter_ipv4(self, ipv4, line):
|
|
"""ipv4地址匿名化"""
|
|
if not self.verify_ip_address(ipv4):
|
|
return line
|
|
ipv4_format = ipv4.replace(".", "\\.")
|
|
# 非单字节ip地址直接匿名化
|
|
if not re.search(self.ipv4_pattern, "【" + ipv4 + "】"):
|
|
line = re.compile(r"(?<![\d.])" + ipv4_format + r"(?![.\d])").sub("<ip>", line)
|
|
elif re.search(self.ipv4_1_and_prefix_pattern + ipv4_format + r"(?![.\d])", line, re.IGNORECASE):
|
|
# 单字节ip地址需搜索关键字眼，有关键字眼则段落中单字节ip地址匿名化
|
|
line = re.compile(self.ipv4_pattern).sub("<ip>", line)
|
|
return line
|
|
|
|
def _ip_address_filter(self, input_data: str):
|
|
""" IPv4、IPv6地址匿名化"""
|
|
lines = input_data.split("\n")
|
|
line_list = []
|
|
for line in lines:
|
|
# 为防止IP地址处于段落开头或结尾不能被匹配，需要在字符串首尾加占位符
|
|
line = ''.join(['【', line, '】'])
|
|
ipv4_groups = self.ipv4_re_compile.findall(line)
|
|
for ipv4 in ipv4_groups:
|
|
line = self.filter_ipv4(ipv4, line)
|
|
ipv6_groups = self.ipv6_re_compile.findall(line)
|
|
for group in ipv6_groups:
|
|
ipv6 = group[0]
|
|
if ipv6 and self.verify_ip_address(ipv6):
|
|
line = re.compile(r"(?<![0-9a-fA-F:])" + ipv6 + "(?![0-9a-fA-F:])").sub("<ip>", line)
|
|
line_list.append(line[1:-1])
|
|
text = "\n".join([line.strip() for line in line_list])
|
|
return text
|