diff options
Diffstat (limited to 'cli.py')
| -rw-r--r-- | cli.py | 26 |
1 files changed, 26 insertions, 0 deletions
@@ -0,0 +1,26 @@ +import argparse +from obfuscator.csv_reader import CSVReader +from obfuscator.logger import get_logger + +logger = get_logger("CLI") + + +def main(): + parser = argparse.ArgumentParser(description="gdpr-obfuscator") + # 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: + logger.debug("User chose to read CSV from local path") + reader = CSVReader() + data = reader.read_local(args.local) + print(data) + else: + logger.debug("User chose to read CSV from S3") + + +if __name__ == "__main__": + main() |
