aboutsummaryrefslogtreecommitdiffstats
path: root/cli.py
blob: bb124215efd106ed03f5402256317f62ae892a55 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import argparse
from obfuscator.csv_reader import CSVReader

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:
        reader = CSVReader(args.local)
        data = reader.read_local()
        print(data)
    else:
        pass

if __name__ == "__main__":
    main()
git.ajschof.me — hosted by ajschofield — powered by cgit