Add option -C|--config file

[m_librarian.git] / scripts / ml-initdb.py

diff --git a/scripts/ml-initdb.py b/scripts/ml-initdb.py
index 79bbe431a1ef1da517c801465b5d8040309c98f1..0d4e73b6467d7e93bff7febe22e9622ad3b8e64b 100755 (executable)
--- a/scripts/ml-initdb.py
+++ b/scripts/ml-initdb.py
@@ -1,6 +1,23 @@
 #! /usr/bin/env python
 
-from m_librarian.db import init_db
+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('-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)
+    else:
+        print "Imported %d genres" % new