scripts/ml-initdb.py

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