Generalize export format flag input

Author: theotherjimmy

tools/memap.py

@@ -486,6 +486,15 @@ def parse_toolchain(string) :
         else:
             return newstring
 
+    export_formats = ["json", "csv-ci", "table"]
+    @staticmethod
+    def parse_export_format(string):
+        newstring = string.lower().replace("_","-")
+        if newstring not in MemapParser.export_formats:
+            raise (argparse.ArgumentTypeError("{} is not a supported export format. Supported export formats are {}".format(string, ", ".join(MemapParser.export_formats))))
+        else :
+            return newstring
+
 def main():
 
     version = '0.3.10'
@@ -500,8 +509,8 @@ def main():
 
     parser.add_argument('-o', '--output', help='output file name', required=False)
 
-    parser.add_argument('-e', '--export', dest='export', required=False,\
-                        help="export format (examples: 'json', 'csv-ci', 'table': default)")
+    parser.add_argument('-e', '--export', dest='export', required=False, default='table', type=MemapParser.parse_export_format,\
+                        help="export format (examples: %s: default)" % ", ".join(MemapParser.export_formats))
 
     parser.add_argument('-v', '--version', action='version', version=version)
 
@@ -521,10 +530,6 @@ def main():
         if memap.parse(args.file, args.toolchain) is False:
             sys.exit(0)
 
-    # default export format is table
-    if not args.export:
-        args.export = 'table'
-
     # Write output in file
     if args.output != None:
         memap.generate_output(args.export, args.output)