update pyproject.toml & references with new src folder name

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)