#!/usr/bin/python3

import argparse

from invirt import database


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

def main():
    parser = argparse.ArgumentParser(description='Modify the invirt database')
    parser.add_argument('action', choices=['create', 'drop'])

    args = parser.parse_args()

    database.connect()

    if args.action == 'create':
        create()
    elif args.action == 'drop':
        database.meta.drop_all()


if __name__ == '__main__':
    main()