#!/usr/bin/env python

from invirt import database
import sys

def usage():
    print >>sys.stderr, "Usage: %s [create|drop]" %(sys.argv[0],)
    sys.exit(-1)

if len(sys.argv) == 1:
    usage()

def create():
    database.meta.create_all()
    session = database.session()
    session.begin()
    if not database.Type.query.get('linux-hvm'):
        session.save(database.Type(type_id='linux-hvm', description='HVM',
                                   hvm='t', apic='t', acpi='t', pae='t'))
    if not database.Type.query.get('linux'):
        session.save(database.Type(type_id='linux', description='ParaVM',
                                   hvm='f', apic='f', acpi='f', pae='f'))
    session.commit()

database.connect()

if sys.argv[1] == "create":
    create()
elif sys.argv[1] == "drop":
    database.meta.drop_all()
else:
    usage()