aboutsummaryrefslogtreecommitdiffstats
path: root/gdpr_obfuscator/__init__.py
blob: 9e2d83733e2723c48ab6ddbb3765ae61a7d5e1f7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from .read import DataReader
from .obfuscate import obfuscate_data
from .utils import Utilities
from typing import List


class Obfuscator:
    def __init__(self):
        self.reader = DataReader()
        self.utils = Utilities()

    def process_s3(self, path: str, pii_fields: List[str]) -> bytes:
        obfuscated_data = obfuscate_data(self.reader.read_s3(path), pii_fields)
        return self.utils.create_byte_stream(obfuscated_data)

    def process_local(self, path: str, pii_fields: List[str]) -> bytes:
        obfuscated_data = obfuscate_data(self.reader.read_local(path), pii_fields)
        return self.utils.create_byte_stream(obfuscated_data)


class ExportTool:
    def __init__(self):
        pass
git.ajschof.me — hosted by ajschofield — powered by cgit