Merge invirt-database-server into invirt-database.

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

diff --git a/invirt-database-tables b/invirt-database-tables
new file mode 100755 (executable)
index 0000000..78c7ab4
--- /dev/null
+++ b/invirt-database-tables
@@ -0,0 +1,32 @@
+#!/usr/bin/env python
+
+from invirt import database
+import sys
+
+def usage():
+    print >>sys.stderr, "Usage: %s [create|drop]" %(sys.argv[0],)
+    sys.exit(-1)
+
+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":
+    create()
+elif sys.argv[1] == "drop":
+    database.meta.drop_all()
+else:
+    usage()