X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=scripts%2Fml-initdb.py;h=71d3598e512bdd8475833b8ef44666d81d457696;hb=f5f7e40267b3e58f4999c4b00a7b8802e2021958;hp=4dc71b5f9cb24ee5d223c6e455d7bbaed78f6270;hpb=46d1f9fa96d1bc61523fa98647ea1ba948fbc159;p=m_librarian.git

diff --git a/scripts/ml-initdb.py b/scripts/ml-initdb.py
index 4dc71b5..71d3598 100755
--- a/scripts/ml-initdb.py
+++ b/scripts/ml-initdb.py
@@ -1,18 +1,24 @@
 #! /usr/bin/env python
 
 import argparse
+from m_librarian.config import get_config
 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')
+    parser.add_argument('-C', '--config', help='configuration file')
+    parser.add_argument('-D', '--database', help='database URI')
     args = parser.parse_args()
 
+    if args.config:
+        get_config(args.config)  # Get and cache config file
+
     open_db(args.database)
     init_db()
-    old, new = import_glst()
-    if old:
-        print "Imported %d genres (ignored %d existing)" % (new, old)
+    count_old, count_new = import_glst()
+    if count_old:
+        print("Imported %d genres (ignored %d existing)" % (
+            count_new, count_old))
     else:
-        print "Imported %d genres" % new
+        print("Imported %d genres" % count_new)