#!/usr/bin/env python

from Bio.GenBank import FeatureParser, Iterator
from BioSQL import BioSeqDatabase
import sys, os

if len(sys.argv) < 3:
    raise SystemExit('Usage: load <database-name> <genbank-files>')

dbname = sys.argv[1]
files = ["'%s'" % arg for arg in sys.argv[2:]]
prog = os.path.join(os.path.dirname(sys.argv[0]), 'gbkhdr')
pipe = os.popen(prog + ' ' + ' '.join(files), 'r')

db = BioSeqDatabase.open_database(
    driver='MySQLdb', user='', passwd='', host='localhost', db=dbname)

try:
    subdb = db[dbname]
    print 'Adding to database "%s".' % dbname
except KeyError:
    subdb = db.new_database(dbname)
    print 'Created new database "%s".' % dbname
subdb.load(Iterator(pipe, FeatureParser()))
