X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=scripts%2Fml-initdb.py;h=4dc71b5f9cb24ee5d223c6e455d7bbaed78f6270;hb=2efaa889e24d1cdf00d38687a8b759dfe6200f2f;hp=ca696bd7bc2ab847ef15eb4045383b6f2ada438d;hpb=fb881e5d6e0de019b19bc2aa007c7783c73d9ce5;p=m_librarian.git

diff --git a/scripts/ml-initdb.py b/scripts/ml-initdb.py
index ca696bd..4dc71b5 100755
--- a/scripts/ml-initdb.py
+++ b/scripts/ml-initdb.py
@@ -1,11 +1,17 @@
 #! /usr/bin/env python
 
-from m_librarian.db import init_db
-from m_librarian.glst import import_data
+import argparse
+from m_librarian.db import open_db, init_db
+from m_librarian.glst import import_glst
 
 if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description='Init')
+    parser.add_argument('-d', '--database', help='database URI')
+    args = parser.parse_args()
+
+    open_db(args.database)
     init_db()
-    old, new = import_data()
+    old, new = import_glst()
     if old:
         print "Imported %d genres (ignored %d existing)" % (new, old)
     else: