blob: 7f118d0f1ff1633f0cd4e77acaeab3581b0856d9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
from .read import FileHandler
from .obfuscate import obfuscate_data
from .utils import Utilities
class Obfuscator:
def __init__(self):
self.reader = FileHandler()
self.utils = Utilities()
def process_s3(self, input: str) -> bytes:
path, pii_fields = self.utils.process_json_input(input)
obfuscated_data = obfuscate_data(self.reader.read_s3(path), pii_fields)
return self.utils.create_byte_stream(obfuscated_data)
def process_local(self, input: str) -> bytes:
path, pii_fields = self.utils.process_json_input(input)
obfuscated_data = obfuscate_data(self.reader.read_local(path), pii_fields)
return self.utils.create_byte_stream(obfuscated_data)
|
