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

diff --git a/code/xen-ips b/code/xen-ips
index 3a69a77..58f3078 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=None, mac_addr=randomMAC(), ip=ip, hostname=None)
+    database.session.save(n)
+    database.session.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)