diff options
Diffstat (limited to 'gdpr-obfuscator/__init__.py')
| -rw-r--r-- | gdpr-obfuscator/__init__.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/gdpr-obfuscator/__init__.py b/gdpr-obfuscator/__init__.py new file mode 100644 index 0000000..2ee3b96 --- /dev/null +++ b/gdpr-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) |