update sipb-xen-database-tables

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

diff --git a/sipb-xen-database-tables b/sipb-xen-database-tables
index f054616..0b37bfc 100755 (executable)
--- 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()