X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-web.git/blobdiff_plain/e37dd15bbf8683bab8454ea0eec8e9968ce982b2..874a30391973642d3ff350f977c1ac705644fbb0:/code/xen-ips

diff --git a/code/xen-ips b/code/xen-ips
index 3a69a77..0af6956 100755
--- a/code/xen-ips
+++ b/code/xen-ips
@@ -1,6 +1,6 @@
 #!/usr/bin/python
 import random
-from sipb_xen_database import *
+from invirt import database
 import sys
 
 # stolen directly from xend/server/netif.py
@@ -37,9 +37,9 @@ def usage():
     print >> sys.stderr, "USAGE: " + sys.argv[0] + " <ip>"
 
 def addip(ip):
-    n = NIC(machine_id=None, mac_addr=randomMAC(), ip=ip, hostname=None)
-    ctx.current.save(n)
-    ctx.current.flush()
+    n = database.NIC(machine_id=None, mac_addr=randomMAC(), ip=ip, hostname=None)
+    database.ctx.current.save(n)
+    database.ctx.current.flush()
 
 
 if __name__ == '__main__':
@@ -48,5 +48,5 @@ if __name__ == '__main__':
     else:
         usage()
         sys.exit(1)
-    connect('postgres://sipb-xen@sipb-xen-dev/sipb_xen')
+    database.connect()
     addip(ip)