invirt-database-tables

   1 #!/usr/bin/env python
   2
   3 from invirt import database
   4 import sys
   5
   6 def usage():
   7     print >>sys.stderr, "Usage: %s [create|drop]" %(sys.argv[0],)
   8     sys.exit(-1)
   9
  10 if len(sys.argv) == 1:
  11     usage()
  12
  13 def create():
  14     database.meta.create_all()
  15     session = database.session()
  16     session.begin()
  17     session.save(database.Type(type_id='linux-hvm', description='HVM',
  18                                hvm='t', apic='t', acpi='t', pae='t'))
  19     session.save(database.Type(type_id='linux', description='ParaVM',
  20                                hvm='f', apic='f', acpi='f', pae='f'))
  21     session.commit()
  22
  23 database.connect()
  24
  25 if sys.argv[1] == "create":
  26     create()
  27 elif sys.argv[1] == "drop":
  28     database.meta.drop_all()
  29 else:
  30     usage()