From 38c8b002278c5388d5f624a285023ee0260e86db Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Wed, 26 Feb 2025 16:44:37 +0000 Subject: update cli.py to transform cli parameters into a json string --- cli.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/cli.py b/cli.py index 6c71c51..ac68a9a 100644 --- a/cli.py +++ b/cli.py @@ -1,4 +1,5 @@ import argparse +import json from gdpr_obfuscator import Obfuscator # This is a simple CLI for demonstration and doesn't undergo the same level @@ -31,10 +32,17 @@ def main(): obfuscator = Obfuscator() + json_input = json.dumps( + { + "file_path": args.local if args.local else args.s3, + "pii_fields": args.pii, + } + ) + if args.local and not args.s3: - obfuscated_data = obfuscator.process_local(args.local, args.pii) + obfuscated_data = obfuscator.process_local(json_input) else: - obfuscated_data = obfuscator.process_s3(args.s3, args.pii) + obfuscated_data = obfuscator.process_s3(json_input) print(obfuscated_data) -- cgit v1.2.3