Added nc

[invirt/packages/invirt-web.git] / code / controls.py

diff --git a/code/controls.py b/code/controls.py
index 09367e2..0acca3a 100644 (file)
--- a/code/controls.py
+++ b/code/controls.py
@@ -94,7 +94,7 @@ def bootMachine(machine, cdtype):
         raise CodeError('"%s" on "control %s create %s' 
                         % (err, machine.name, cdtype))
 
         raise CodeError('"%s" on "control %s create %s' 
                         % (err, machine.name, cdtype))
 

-def createVm(username, state, owner, contact, name, description, memory, disksize, machine_type, cdrom, autoinstall):
+def createVm(username, state, owner, contact, name, description, memory, disksize, machine_type, nic_type, cdrom, autoinstall):

     """Create a VM and put it in the database"""
     # put stuff in the table
     session.begin()
     """Create a VM and put it in the database"""
     # put stuff in the table
     session.begin()

@@ -110,7 +110,7 @@ def createVm(username, state, owner, contact, name, description, memory, disksiz
         machine.uuid = uuidToString(randomUUID())
         machine.boot_off_cd = True
         machine.type = machine_type
         machine.uuid = uuidToString(randomUUID())
         machine.boot_off_cd = True
         machine.type = machine_type

-        session.save_or_update(machine)
+        session.add(machine)

         disk = Disk(machine=machine,
                     guest_device_name='hda', size=disksize)
         nic = NIC.query.filter_by(machine_id=None).filter_by(reusable=True).first()
         disk = Disk(machine=machine,
                     guest_device_name='hda', size=disksize)
         nic = NIC.query.filter_by(machine_id=None).filter_by(reusable=True).first()

@@ -119,8 +119,9 @@ def createVm(username, state, owner, contact, name, description, memory, disksiz
                             "Contact %s." % config.contact)
         nic.machine = machine
         nic.hostname = name
                             "Contact %s." % config.contact)
         nic.machine = machine
         nic.hostname = name

-        session.save_or_update(nic)
-        session.save_or_update(disk)
+        nic.nic_type = nic_type
+        session.add(nic)
+        session.add(disk)

         cache_acls.refreshMachine(machine)
         makeDisks(machine)
         session.commit()
         cache_acls.refreshMachine(machine)
         makeDisks(machine)
         session.commit()

@@ -212,7 +213,7 @@ def deleteVM(machine):
         for nic in machine.nics:
             nic.machine_id = None
             nic.hostname = None
         for nic in machine.nics:
             nic.machine_id = None
             nic.hostname = None

-            session.save_or_update(nic)
+            session.add(nic)

         for disk in machine.disks:
             session.delete(disk)
         session.delete(machine)
         for disk in machine.disks:
             session.delete(disk)
         session.delete(machine)