feat: added flags for mismatch_loader (#4246)

Signed-off-by: Meet Soni <[email protected]>

Author: inosmeet

cve_bin_tool/mismatch_loader.py

@@ -1,3 +1,4 @@
+import argparse
 import itertools
 import os
 import sqlite3
@@ -66,14 +67,46 @@ def setup_sqlite(data_dir: str, db_file: str) -> bool:
     return True
 
 
-def main():
+def setup_args():
     """
-    Run the SQLite loader utility
+    Setup command line arguments
     """
+
     db_path = DISK_LOCATION_DEFAULT / DBNAME
     parent_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
     data_dir = os.path.join(parent_dir, "data")
-    if not setup_sqlite(data_dir, db_path):
+
+    parser = argparse.ArgumentParser(description="mismatch loader")
+    parser.add_argument(
+        "--dir",
+        metavar="DATA_DIR",
+        type=str,
+        default=data_dir,
+        help="Data folder location",
+    )
+    parser.add_argument(
+        "--database",
+        "-db",
+        type=str,
+        default=db_path,
+        help="SQLite DB file location",
+    )
+    args = parser.parse_args()
+    return args
+
+
+def main():
+    """
+    Run the SQLite loader utility
+    """
+    args = setup_args()
+
+    if not os.path.exists(args.dir) or not os.path.isdir(args.dir):
+        LOGGER.error(
+            f"Specified data directory does not exist or is not a folder: {args.dir}"
+        )
+        exit(1)
+    if not setup_sqlite(args.dir, args.database):
         exit(1)
     exit(0)