update cli.py to transform cli parameters into a json string

1 files 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)