X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-database.git/blobdiff_plain/8aab6dbab69832cd8705aa463d091b731f1fa19a..d6af243d1320e0b258a088ff33e262b41594ad44:/sipb-xen-database-tables?ds=sidebyside

diff --git a/sipb-xen-database-tables b/sipb-xen-database-tables
index 439e5e1..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,11 +10,11 @@ def usage():
 if len(sys.argv) == 1:
     usage()
 
-connect('postgres://sipb-xen@/sipb_xen')
+database.connect()
 
 if sys.argv[1] == "create":
-    meta.create_all()
+    database.meta.create_all()
 elif sys.argv[1] == "drop":
-    meta.drop_all()
+    database.meta.drop_all()
 else:
     usage()