Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
from typing import List, Tuple
from src.annotations import (
Annotation,
EmailAnnotation,
)
from src.string_replacements import replace_and_update
from src.replacers.interface import ReplacerInterface
import random
import string
def random_char(char_num):
return ''.join(random.choice(string.ascii_letters) for _ in range(char_num))
def random_email():
return random_char(7)+"@gmail.com"
class EmailReplacer(ReplacerInterface):
def __init__(self):
pass
def replace(self, text: str, detections: List[Tuple[int, int, Annotation]]) -> Tuple[str, List[Tuple[int, int, Annotation]]]:
replacements = []
not_processed = []
already_replaced = dict()
for item in detections:
start, end, detection = item
if isinstance(detection, EmailAnnotation):
if text[start:end] not in already_replaced:
already_replaced[text[start:end]] = random_email()
replacements.append((start, end, already_replaced[text[start:end]]))
else:
not_processed.append(item)
return replace_and_update(text, replacements, not_processed)