diff options
Diffstat (limited to 'obfuscator')
| -rw-r--r-- | obfuscator/__init__.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/obfuscator/__init__.py b/obfuscator/__init__.py index e69de29..2ee3b96 100644 --- a/obfuscator/__init__.py +++ b/obfuscator/__init__.py @@ -0,0 +1,12 @@ +from obfuscator.read import DataReader +from obfuscator.write import DataWriter +from obfuscator.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) |