From a16fd48f1e87e9b101250aba532ec81de813e260 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Fri, 14 Feb 2025 12:24:40 +0000 Subject: require only --local or --s3 to be chosen by user in cli.py --- cli.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) (limited to 'cli.py') diff --git a/cli.py b/cli.py index 1911f5c..81114d2 100644 --- a/cli.py +++ b/cli.py @@ -3,9 +3,16 @@ from obfuscator.csv_reader import read def main(): parser = argparse.ArgumentParser(description="gdpr-obfuscator") - parser.add_argument("--local", help="Path to local CSV file") - parser.add_argument("--s3", help="Path to S3 object for CSV file") + # Require user to either choose a local file or an S3 object + loc = parser.add_mutually_exclusive_group(required=True) + loc.add_argument("--local") + loc.add_argument("--s3") args = parser.parse_args() + if args.local and not args.s3: + print(read(args.local)) + else: + pass + if __name__ == "__main__": main() -- cgit v1.2.3