Conditionalize populating the types table in invirt-database-tables
[invirt/packages/invirt-database.git] / 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     if not database.Type.query.get('linux-hvm'):
18         session.save(database.Type(type_id='linux-hvm', description='HVM',
19                                    hvm='t', apic='t', acpi='t', pae='t'))
20     if not database.Type.query.get('linux'):
21         session.save(database.Type(type_id='linux', description='ParaVM',
22                                    hvm='f', apic='f', acpi='f', pae='f'))
23     session.commit()
24
25 database.connect()
26
27 if sys.argv[1] == "create":
28     create()
29 elif sys.argv[1] == "drop":
30     database.meta.drop_all()
31 else:
32     usage()