Conditionalize populating the types table in invirt-database-tables

[invirt/packages/invirt-database.git] / invirt-database-tables

diff --git a/invirt-database-tables b/invirt-database-tables
index 0b37bfc..78c7ab4 100755 (executable)
--- a/invirt-database-tables
+++ b/invirt-database-tables
@@ -10,10 +10,22 @@ def usage():
 if len(sys.argv) == 1:
     usage()
 
+def create():
+    database.meta.create_all()
+    session = database.session()
+    session.begin()
+    if not database.Type.query.get('linux-hvm'):
+        session.save(database.Type(type_id='linux-hvm', description='HVM',
+                                   hvm='t', apic='t', acpi='t', pae='t'))
+    if not database.Type.query.get('linux'):
+        session.save(database.Type(type_id='linux', description='ParaVM',
+                                   hvm='f', apic='f', acpi='f', pae='f'))
+    session.commit()
+
 database.connect()
 
 if sys.argv[1] == "create":
-    database.meta.create_all()
+    create()
 elif sys.argv[1] == "drop":
     database.meta.drop_all()
 else: