From 9fa5e1f691ef6b2d651789ee475136a701f063a9 Mon Sep 17 00:00:00 2001
From: Alex Schofield <git@ajschof.me>
Date: Wed, 19 Feb 2025 16:06:54 +0000
Subject: add initial code for __init__.py entry point

---
 obfuscator/__init__.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/obfuscator/__init__.py b/obfuscator/__init__.py
index e69de29..726c365 100644
--- a/obfuscator/__init__.py
+++ b/obfuscator/__init__.py
@@ -0,0 +1,15 @@
+from read import DataReader
+from write import DataWriter
+from utils import Utilities
+from obfuscate import obfuscate
+from types import List
+
+
+def main(s3_source: str, pii_fields: List[str], log_level: str = "INFO") -> bytes:
+    reader = DataReader(log_level)
+    writer = DataWriter()
+    utilities = Utilities()
+    source = utilities.get_source(s3_source)
+    data = reader.read_string(source)
+    obfuscated_data = obfuscate(data, pii_fields)
+    return writer.create_byte_stream(obfuscated_data)
-- 
cgit v1.2.3