X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=scripts%2Fml-initdb.py;h=6535d4559827ac37b7b4ffa36925f39604ff03b5;hb=114ba9ea6f4aa1af0ff361d75be9f2434360678d;hp=4445fc6f951ddac6dac3e4cfd7f995cbb554205a;hpb=af6ee17b251b1ae074e1eca5d80c7a599ee88536;p=m_librarian.git

diff --git a/scripts/ml-initdb.py b/scripts/ml-initdb.py
index 4445fc6..6535d45 100755
--- a/scripts/ml-initdb.py
+++ b/scripts/ml-initdb.py
@@ -1,13 +1,24 @@
 #! /usr/bin/env python
 
+from __future__ import print_function
+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__':
-    open_db()
+    parser = argparse.ArgumentParser(description='Init')
+    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)
+        print("Imported %d genres (ignored %d existing)" % (new, old))
     else:
-        print "Imported %d genres" % new
+        print("Imported %d genres" % new)