add args

Author: Honglei-Qiu

sqlite/GraphNet.db

@@ -3,7 +3,12 @@ work under /GraphNet/
 mkdir -p sqlite/logs
 
 # migrate database
+# Use default database path
 python ./sqlite/init_db.py
 
+# Specify custom database path
+python ./sqlite/init_db.py --db_path sqlite/GraphNet.db
+
+
 # Add data to database
 bash ./sqlite/graphsample_insert.sh | tee "sqlite/logs/insert_$(date +'%Y-%m-%d-%H%M%S').log"

sqlite/Readme.md

@@ -1,5 +1,6 @@
 import sqlite3
 import re
+import argparse
 from pathlib import Path
 
 
@@ -53,4 +54,12 @@ def migrate(db_path: str = "sqlite/GraphNet.db", migrates_dir: str = "sqlite/mig
 
 
 if __name__ == "__main__":
-    migrate()
+    parser = argparse.ArgumentParser(description="GraphNet database migration tool")
+    parser.add_argument(
+        "--db_path",
+        type=str,
+        default="sqlite/GraphNet.db",
+        help="Database file path (default: sqlite/GraphNet.db)",
+    )
+    args = parser.parse_args()
+    migrate(args.db_path)