X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-database.git/blobdiff_plain/8ff8bfe4ba94dd1ed0a89a06d528c1b48072f2dd..5d9dbc6ac22ce03205de8ce2246a1774526e84c4:/invirt-database-tables

diff --git a/invirt-database-tables b/invirt-database-tables
new file mode 100755
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()