aboutsummaryrefslogtreecommitdiffstats
path: root/cli.py
diff options
context:
space:
mode:
Diffstat (limited to 'cli.py')
-rw-r--r--cli.py26
1 files changed, 26 insertions, 0 deletions
diff --git a/cli.py b/cli.py
new file mode 100644
index 0000000..19e64d8
--- /dev/null
+++ b/cli.py
@@ -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()
git.ajschof.me — hosted by ajschofield — powered by cgit