add support for csv input files

1 files changed, 4 insertions, 2 deletions

diff --git a/create_report.py b/create_report.py
index 9d7cb57..04b44b7 100644
--- a/create_report.py
+++ b/create_report.py
@@ -4,11 +4,13 @@ import argparse
 import sys
 
 parser = argparse.ArgumentParser(prog=sys.argv[0])
-parser.add_argument('exe_path', help="executable to parse")
+parser.add_argument('input_path', help="input to parse")
 parser.add_argument('-o', '--out', help="output file (default: report.html)", default="report.html")
+parser.add_argument('-c', '--csv', help="Flag: the input file is a csv and NOT an executable", action='store_true')
+
 args = parser.parse_args()
 
-data, stats = parse_data(args.executable)
+data, stats = parse_data(args.input_path, args.csv)
 
 # Load the template from the current directory
 env = Environment(loader=FileSystemLoader('.'))