X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=scripts%2Fml-import.py;h=066c5ccb6bf51bb1069c8487dc7e07571761b7aa;hb=46d1f9fa96d1bc61523fa98647ea1ba948fbc159;hp=0273ff67972e122b492947d5798b79ec837b0feb;hpb=e53f480b62166cc8028f1d6e4fda307ad7af2233;p=m_librarian.git diff --git a/scripts/ml-import.py b/scripts/ml-import.py index 0273ff6..066c5cc 100755 --- a/scripts/ml-import.py +++ b/scripts/ml-import.py @@ -1,14 +1,19 @@ #! /usr/bin/env python -import sys -from m_librarian.db import init_db, update_counters -from m_librarian.glst import import_data +import argparse +from m_librarian.db import open_db, init_db, update_counters +from m_librarian.glst import import_glst from m_librarian.inp import import_inpx if __name__ == '__main__': - if len(sys.argv) != 2: - sys.exit("Usage: %s file.inpx" % sys.argv[0]) + parser = argparse.ArgumentParser(description='Import') + parser.add_argument('-d', '--database', help='database URI') + parser.add_argument('inpx', nargs='+', help='INPX files to import') + args = parser.parse_args() + + open_db(args.database) init_db() - import_data() - import_inpx(sys.argv[1]) + import_glst() + for inpx in args.inpx: + import_inpx(inpx) update_counters()