X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-web.git/blobdiff_plain/fdda3852e46b9fb9bde623d6dbd4d55752fcbf74..2be0969e153ace507ec07b27d156b50e54e00f94:/code/controls.py diff --git a/code/controls.py b/code/controls.py index 2a86a89..a5f1693 100644 --- a/code/controls.py +++ b/code/controls.py @@ -47,12 +47,16 @@ def lvinstall(machine, autoinstall): memsize = machine.memory swapsize = getswap(disksize, memsize) imagesize = disksize - swapsize - ip = machine.nics[0].ip - remctl('control', machine.name, 'install', - 'dist=%s' % autoinstall.distribution, - 'mirror=%s' % autoinstall.mirror, - 'arch=%s' % autoinstall.arch, - 'imagesize=%s' % imagesize) + + installer_options = ['dist=%s' % autoinstall.distribution, + 'mirror=%s' % autoinstall.mirror, + 'arch=%s' % autoinstall.arch, + 'imagesize=%s' % imagesize] + if autoinstall.preseed: + installer_options += ['preseed=http://'+config.web.hostname+'/static/preseed/'+autoinstall.autoinstall_id+'.preseed'] + + remctl('control', machine.name, 'install', + *installer_options) def lvcopy(machine_orig_name, machine, rootpw): """Copy a golden image onto a machine's disk""" @@ -106,17 +110,17 @@ def createVm(username, state, owner, contact, name, description, memory, disksiz 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() + nic = NIC.query.filter_by(machine_id=None).filter_by(reusable=True).first() if not nic: #No IPs left! raise CodeError("No IP addresses left! " - "Contact %s." % config.web.errormail) + "Contact %s." % config.contact) nic.machine = machine nic.hostname = name - session.save_or_update(nic) - session.save_or_update(disk) + session.add(nic) + session.add(disk) cache_acls.refreshMachine(machine) makeDisks(machine) session.commit() @@ -208,7 +212,7 @@ def deleteVM(machine): 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) @@ -222,7 +226,7 @@ def commandResult(username, state, command_name, machine_id, fields): machine = validation.Validate(username, state, machine_id=machine_id).machine action = command_name cdrom = fields.get('cdrom') or None - if cdrom is not None and not CDROM.query().filter_by(cdrom_id=cdrom).one(): + if cdrom is not None and not CDROM.query.filter_by(cdrom_id=cdrom).one(): raise CodeError("Invalid cdrom type '%s'" % cdrom) if action not in "reboot create destroy shutdown delete".split(" "): raise CodeError("Invalid action '%s'" % action)