scripts/ml-initdb.py

   1 #! /usr/bin/env python
   2
   3 import argparse
   4
   5 from m_librarian.config import get_config
   6 from m_librarian.db import open_db, init_db
   7 from m_librarian.glst import import_glst
   8
   9 if __name__ == '__main__':
  10     parser = argparse.ArgumentParser(description='Init database')
  11     parser.add_argument('-C', '--config', help='configuration file')
  12     parser.add_argument('-D', '--database', help='database URI')
  13     args = parser.parse_args()
  14
  15     if args.config:
  16         get_config(args.config)  # Get and cache config file
  17
  18     open_db(args.database)
  19     init_db()
  20     count_old, count_new = import_glst()
  21     if count_old:
  22         print("Imported %d genres (ignored %d existing)" % (
  23             count_new, count_old))
  24     else:
  25         print("Imported %d genres" % count_new)