From 70a16769761450b9c2aa63cda86a30a73bc0217c Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Thu, 20 Feb 2025 18:27:12 +0000 Subject: update pyproject.toml & references with new src folder name --- gdpr_obfuscator/__init__.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 gdpr_obfuscator/__init__.py (limited to 'gdpr_obfuscator/__init__.py') 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) -- cgit v1.2.3