aboutsummaryrefslogtreecommitdiffstats
path: root/gdpr_obfuscator/__init__.py
blob: f688063638bf954bc60fe62aa1930015d457b38c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
from .read import DataReader
from .write import DataWriter
from .obfuscate import obfuscate
from typing import List


def main(s3_source: str, pii_fields: List[str], log_level: str = "INFO") -> bytes:
    reader = DataReader(log_level)
    writer = DataWriter()
    data = reader.read_s3(s3_source)
    obfuscated_data = obfuscate(data, pii_fields)
    return writer.create_byte_stream(obfuscated_data)
git.ajschof.me — hosted by ajschofield — powered by cgit