scripts/invirt-database-tables

   1 #!/usr/bin/python3
   2
   3 import argparse
   4
   5 from invirt import database
   6
   7
   8 def create():
   9     database.meta.create_all()
  10     session = database.session()
  11     session.begin()
  12     if not database.Type.query.get('linux-hvm'):
  13         session.add(database.Type(type_id='linux-hvm', description='HVM',
  14                                   hvm='t', apic='t', acpi='t', pae='t'))
  15     if not database.Type.query.get('linux'):
  16         session.add(database.Type(type_id='linux', description='ParaVM',
  17                                   hvm='f', apic='f', acpi='f', pae='f'))
  18     session.commit()
  19
  20 def main():
  21     parser = argparse.ArgumentParser(description='Modify the invirt database')
  22     parser.add_argument('action', choices=['create', 'drop'])
  23
  24     args = parser.parse_args()
  25
  26     database.connect()
  27
  28     if args.action == 'create':
  29         create()
  30     elif args.action == 'drop':
  31         database.meta.drop_all()
  32
  33
  34 if __name__ == '__main__':
  35     main()