diff options
Diffstat (limited to 'obfuscator')
| -rw-r--r-- | obfuscator/__init__.py | 15 |
1 files changed, 15 insertions, 0 deletions
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) |