#!/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()