From 6ebdc904eb48185c51c9deb4e426d56227646960 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Fri, 21 Feb 2025 00:00:54 +0000 Subject: backup commit - 21st @ 00:00 --- cli.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'cli.py') diff --git a/cli.py b/cli.py index 1a84370..ed00d32 100644 --- a/cli.py +++ b/cli.py @@ -1,11 +1,13 @@ import argparse from gdpr_obfuscator.read import DataReader -from gdpr_obfuscator.write import DataWriter from gdpr_obfuscator.obfuscate import obfuscate -from gdpr_obfuscator.logger import get_logger +from gdpr_obfuscator.utils import Utilities def main(): + + utils = Utilities() + parser = argparse.ArgumentParser( prog="GDPR-Obfuscator", description="Obfuscate sensitive data stored locally or in an AWS environment", @@ -30,7 +32,7 @@ def main(): args = parser.parse_args() log_level = "DEBUG" if args.verbose else "INFO" - logger = get_logger("CLI", log_level) + logger = utils.get_logger("CLI", log_level) reader = DataReader(log_level) @@ -43,9 +45,7 @@ def main(): obfuscated_data = obfuscate(data, args.pii) - writer = DataWriter() - - return writer.create_byte_stream(obfuscated_data) + return obfuscated_data if __name__ == "__main__": -- cgit v1.2.3