From de575144e1ae9587de1c3675623a125d12c2d608 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Fri, 21 Feb 2025 03:41:56 +0000 Subject: fix cli.py by updating imports and with new logic --- cli.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) (limited to 'cli.py') diff --git a/cli.py b/cli.py index ef77314..3b64338 100644 --- a/cli.py +++ b/cli.py @@ -1,6 +1,5 @@ import argparse -from gdpr_obfuscator.read import DataReader -from gdpr_obfuscator.obfuscate import obfuscate +from gdpr_obfuscator import Obfuscator def main(): @@ -28,14 +27,12 @@ def main(): args = parser.parse_args() - reader = DataReader() + obfuscator = Obfuscator() if args.local and not args.s3: - data = reader.read_local(args.local) + obfuscated_data = obfuscator.process_local(args.local, args.pii) else: - data = reader.read_s3(args.s3) - - obfuscated_data = obfuscate(data, args.pii) + obfuscated_data = obfuscator.process_s3(args.s3, args.pii) print(obfuscated_data) -- cgit v1.2.3