X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-database.git/blobdiff_plain/f2adfe4c961ec18f2c767c5d48538b456ace4d77..49a5555d50b72a14ee16d70dc9612ab022061592:/sipb-xen-database-tables

diff --git a/sipb-xen-database-tables b/sipb-xen-database-tables
index f054616..0b37bfc 100755
--- a/sipb-xen-database-tables
+++ b/sipb-xen-database-tables
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-from sipb_xen_database import *
+from invirt import database
 import sys
 
 def usage():
@@ -10,9 +10,11 @@ def usage():
 if len(sys.argv) == 1:
     usage()
 
-connect('postgres://sipb-xen@localhost/sipb_xen')
+database.connect()
 
 if sys.argv[1] == "create":
-    meta.create_all()
+    database.meta.create_all()
+elif sys.argv[1] == "drop":
+    database.meta.drop_all()
 else:
-    meta.drop_all()
+    usage()