X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=scripts%2Fml-import.py;h=066c5ccb6bf51bb1069c8487dc7e07571761b7aa;hb=2aec5f3920d5ba9f7add67a1980e4321f5b7689b;hp=530e4a05651f3a3b34cbc5cc57353e7d1a75d0ee;hpb=af6ee17b251b1ae074e1eca5d80c7a599ee88536;p=m_librarian.git

diff --git a/scripts/ml-import.py b/scripts/ml-import.py
index 530e4a0..066c5cc 100755
--- a/scripts/ml-import.py
+++ b/scripts/ml-import.py
@@ -1,16 +1,19 @@
 #! /usr/bin/env python
 
-import sys
+import argparse
 from m_librarian.db import open_db, init_db, update_counters
 from m_librarian.glst import import_glst
 from m_librarian.inp import import_inpx
 
 if __name__ == '__main__':
-    if len(sys.argv) < 2:
-        sys.exit("Usage: %s file.inpx..." % sys.argv[0])
-    open_db()
+    parser = argparse.ArgumentParser(description='Import')
+    parser.add_argument('-d', '--database', help='database URI')
+    parser.add_argument('inpx', nargs='+', help='INPX files to import')
+    args = parser.parse_args()
+
+    open_db(args.database)
     init_db()
     import_glst()
-    for inpx in sys.argv[1:]:
+    for inpx in args.inpx:
         import_inpx(inpx)
     update_counters()