update cli.py to properly read from local csv files

1 files changed, 4 insertions, 2 deletions

diff --git a/cli.py b/cli.py
index 0bbaf89..bb12421 100644
--- a/cli.py
+++ b/cli.py
@@ -1,5 +1,5 @@
 import argparse
-from obfuscator.csv_reader import read_local, read_s3
+from obfuscator.csv_reader import CSVReader
 
 def main():
     parser = argparse.ArgumentParser(description="gdpr-obfuscator")
@@ -10,7 +10,9 @@ def main():
     args = parser.parse_args()
 
     if args.local and not args.s3:
-        print(read(args.local))
+        reader = CSVReader(args.local)
+        data = reader.read_local()
+        print(data)
     else:
         pass