def download(caller_id, description, name, path, disk_dev, disk_controller): """ Downloads Image with given path and saves it with specified name and description. @cmview_admin_cm @param_post{description,string} @param_post{name,string} @param_post{path,string} HTTP or FTP path to IsoImage @param_post{disk_dev} @param_post{disk_controller} """ # size value is taken try: connection = urllib.urlopen(path) size = int(connection.info()["Content-Length"]) except IOError: log.exception('Cannot find image') raise CMException('image_not_found') except KeyError: log.exception(caller_id, 'Cannot calculate size') raise CMException('image_calculate_size') user = User.get(caller_id) image = IsoImage.create(name=name, description=description, user=user, disk_dev=disk_dev, disk_controller=disk_controller) try: image.save() except Exception, e: log.error(caller_id, "Unable to save image to DB: %s" % str(e)) raise CMException('image_create')
def edit(caller_id, iso_image_id, name, description, disk_controller): """ Sets Image's new attributes. Those should be get by src.cm.manager.image.get_by_id(). @cmview_admin_cm @parameter{system_image_id,string} new Image name @parameter{name,string} new Image name @parameter{description,string} new Image description @parameter{disk_controller} new Image controller optional @parameter{video_device} new video device optional @parameter{network_device} new network device optional @parameter{platform} optional """ image = IsoImage.admin_get(iso_image_id) if image.state != image_states['ok']: raise CMException('image_edit') image.name = name image.description = description image.disk_controller = disk_controller try: image.save() except: raise CMException('image_edit')
def attach(caller_id, iso_image_id, vm_id): # vm_id, img_id, destination='usb', check=True/False """ Attaches specified IsoImage to specified VM. It makes possible booting any operating system on created VM. @cmview_user @param_post{iso_image_id,int} id of block device (should be IsoImage type) @param_post{vm_id,int} id of the VM which IsoImage should be attached to @response{None} """ vm = VM.get(caller_id, vm_id) disk = IsoImage.get(caller_id, iso_image_id) # Check if disk is already attached to a vm if disk.vm: raise CMException('image_attached') disk.attach(vm) try: disk.save() except: raise CMException('iso_image_attach')
def download(caller_id, description, name, path, disk_dev, disk_controller): """ Downloads image depending on the \c data parameter. @cmview_admin_cm @parameter{description,string} @parameter{name,string} @parameter{path,string} HTTP or FTP path to image to download @parameter{type,image_types} type of image, automatically set, type is in the URL requested @response{None} """ # size value is taken try: connection = urllib.urlopen(path) size = int(connection.info()["Content-Length"]) except IOError: log.exception('Cannot find image') raise CMException('image_not_found') except KeyError: log.exception(caller_id, 'Cannot calculate size') raise CMException('image_calculate_size') user = User.get(caller_id) image = IsoImage.create(name=name, description=description, user=user, disk_dev=disk_dev, disk_controller=disk_controller) try: image.save() except Exception, e: log.error(caller_id, "Unable to save image to DB: %s" % str(e)) raise CMException('image_create')
def download(caller_id, name, description, path, disk_controller): """ Downloads specified IsoImage and saves it with specified name and description. @cmview_user @param_post{name,string} @param_post{description,string} @param_post{path,string} HTTP or FTP path to IsoImage to download @param_post{disk_controller} """ user = User.get(caller_id) if not any([path.startswith('http://'), path.startswith('https://'), path.startswith('ftp://')]): path = 'http://' + path.strip() # size value is taken try: connection = urllib.urlopen(path) size = int(connection.info()["Content-Length"]) except IOError: log.exception('Cannot find image') raise CMException('image_not_found') except KeyError: log.exception(caller_id, 'Cannot calculate size') raise CMException('image_calculate_size') user.check_storage(size / (1024 * 1024)) image = IsoImage.create(user=user, description=description, name=name, disk_controller=disk_controller, disk_dev=1) try: image.save() except Exception, e: log.error(caller_id, "Unable to save image to DB: %s" % str(e)) raise CMException('image_create')
def get_by_id(caller_id, iso_image_id): """ @cmview_admin_cm @param_post{iso_image_id,int} id of the IsoImage to get @response{dict} IsoImage.dict property for requested IsoImage """ return IsoImage.admin_get(iso_image_id).dict
def copy(caller_id, src_image_id, dest_user_id): """ Copies selected IsoImage to User's IsoImages pool. @cmview_admin_cm @param_post{src_image_id,int} @param_post{dest_user_id,int} """ src_image = IsoImage.admin_get(src_image_id) dest_user = User.get(dest_user_id) dest_image = IsoImage.create(name=src_image.name, description=src_image.description, user=dest_user, disk_controller=src_image.disk_controller, disk_dev=src_image.disk_dev) try: dest_image.save() except Exception, e: log.error(caller_id, "Unable to commit: %s" % str(e)) raise CMException('image_create')
def get_by_id(caller_id, iso_image_id): """ @cmview_admin_cm @parameter{image_id,int} id of the Image to get @parameter{type,image_types} type of image, automatically set, type is in the URL requested @response{dict} extended information about specified Image """ return IsoImage.admin_get(iso_image_id).dict
def get_by_id(caller_id, iso_image_id): """ Returns requested IsoImage provided it belongs to caller. @cmview_user @param_post{iso_image_id,int} id of the IsoImage to get @response{dict} IsoImage.dict property of the requested IsoImage. """ return IsoImage.get(caller_id, iso_image_id).dict
def delete(caller_id, iso_image_id): """ Deletes given Image @cmview_admin_cm @parameter{system_image_id} id of the Image to delete @parameter{type,image_types} type of image, automatically set, type is in the URL requested """ image = IsoImage.admin_get(iso_image_id) image.check_attached() image.state = image_states['locked'] image.save()
def delete(caller_id, iso_image_id): """ Sets IsoImage state to 'locked'. @cmview_admin_cm @param_post{iso_image_id} id of the IsoImage to delete @todo Should delete IsoImage and set its state to 'deleted'. """ image = IsoImage.admin_get(iso_image_id) image.check_attached() image.state = image_states['locked'] image.save()
def delete(caller_id, iso_image_id): """ Deletes specified IsoImage. @cmview_user @param_post{iso_image_ids} id of the IsoImage to delete """ image = IsoImage.get(caller_id, iso_image_id) if image.state != image_states['ok']: raise CMException('image_delete') try: subprocess.call(['rm', image.path]) except Exception, e: raise CMException('image_delete')
def detach(caller_id, iso_image_id, vm_id): """ Detaches specified IsoImage from specified VM. @cmview_user @param_post{iso_image_id,int} id of the IsoImage to detach @param_post{vm_id,int} id of the VM from which IsoImage should be detached @response{None} """ vm = VM.get(caller_id, vm_id) disk = IsoImage.get(caller_id, iso_image_id) disk.detach(vm) try: disk.save() except: raise CMException('iso_image_attach')
def download(caller_id, name, description, path, disk_controller): """ Downloads specified IsoImage and saves it with specified name and description. @cmview_user @param_post{name,string} @param_post{description,string} @param_post{path,string} HTTP or FTP path to IsoImage to download @param_post{disk_controller} """ user = User.get(caller_id) if not any([ path.startswith('http://'), path.startswith('https://'), path.startswith('ftp://') ]): path = 'http://' + path.strip() # size value is taken try: connection = urllib.urlopen(path) size = int(connection.info()["Content-Length"]) except IOError: log.exception('Cannot find image') raise CMException('image_not_found') except KeyError: log.exception(caller_id, 'Cannot calculate size') raise CMException('image_calculate_size') user.check_storage(size / (1024 * 1024)) image = IsoImage.create(user=user, description=description, name=name, disk_controller=disk_controller, disk_dev=1) try: image.save() except Exception, e: log.error(caller_id, "Unable to save image to DB: %s" % str(e)) raise CMException('image_create')
def edit(caller_id, iso_image_id, name, description, disk_controller): """ Updates IsoImage's attributes. @cmview_user @param_post{iso_image_id,int} id of the Image to edit @param_post{name,string} new name @param_post{description,string} new description @param_post{disk_controller} new controller (optional) """ image = IsoImage.get(caller_id, iso_image_id) if image.state != image_states['ok']: raise CMException('image_edit') image.name = name image.description = description image.disk_controller = disk_controller try: image.save() except: raise CMException('image_edit')
def edit(caller_id, iso_image_id, name, description, disk_controller): """ Updates specified IsoImage's attributes. @cmview_admin_cm @param_post{iso_image_id,string} new IsoImage name @param_post{name,string} new IsoImage name @param_post{description,string} new IsoImage description @param_post{disk_controller} new IsoImage controller optional """ image = IsoImage.admin_get(iso_image_id) if image.state != image_states['ok']: raise CMException('image_edit') image.name = name image.description = description image.disk_controller = disk_controller try: image.save() except: raise CMException('image_edit')
def create(user, name, description, image_id, template_id, public_ip_id, iso_list, disk_list, vnc, groups, ssh_key=None, ssh_username=None, count=1, farm=None, head_template_id=None, node_id=False, lease_id=None, user_data=None): from cm.models.storage_image import StorageImage from cm.utils.threads.vm import VMThread template = Template.get(template_id) image = SystemImage.get(user.id, image_id, groups) if image.state != image_states['ok']: raise CMException('image_unavailable') if farm: head_template = Template.get(head_template_id) wn_template = template user.check_quota([(head_template, 1), (wn_template, count)]) count += 1 else: user.check_quota([(template, count)]) vms = [] reservation_id = None for i in range(count): # create VM instance log.debug(user.id, "Looking for node") node = Node.get_free_node(head_template, image, node_id) if farm and i == 0 else Node.get_free_node(template, image, node_id) log.info(user.id, 'Selected node: %d' % node.id) vm = VM() vm.libvirt_id = -1 if farm: if i == 0: vm.name = '%s-head' % name vm.description = 'Farm head' vm.template = head_template else: vm.name = '%s-wn%d' % (name, i) vm.description = 'Worker Node' vm.template = wn_template else: vm.template = template vm.description = description if count > 1: vm.name = '%s_%d' % (name, i + 1) else: vm.name = name vm.user = user vm.state = vm_states['init'] vm.start_time = datetime.now() vm.system_image = image vm.node = node vm.save_vm = True if farm: vm.farm = farm # Find first free vnc port used_ports = VM.objects.exclude(state__in=[vm_states['closed'], vm_states['erased']]).values_list('vnc_port', flat=True) for new_vnc_port in xrange(VNC_PORTS['START'], VNC_PORTS['END'] + 1): if new_vnc_port not in used_ports and new_vnc_port not in VNC_PORTS['EXCLUDE']: break else: raise CMException('vm_vnc_not_found') log.debug(user.id, "Found vnc port: %d" % new_vnc_port) vm.vnc_port = new_vnc_port # Find first free novnc port used_ports = VM.objects.exclude(state__in=[vm_states['closed'], vm_states['erased']]).values_list('novnc_port', flat=True) for new_novnc_port in xrange(NOVNC_PORTS['START'], NOVNC_PORTS['END'] + 1): if new_novnc_port not in used_ports and new_novnc_port not in NOVNC_PORTS['EXCLUDE']: break else: raise CMException('vm_novnc_not_found') log.debug(user.id, "Found novnc port: %d" % new_novnc_port) vm.novnc_port = new_novnc_port if vnc: vm.attach_vnc() vm.vnc_passwd = password_gen(13, chars=['letters', 'digits'], extra_chars='!@#$%^&*()') vm.ssh_key = ssh_key vm.ssh_username = ssh_username vm.user_data = user_data vm.save() if not reservation_id: reservation_id = vm.id vm.reservation_id = reservation_id vm.save() if farm and i == 0: farm.head = vm vms.append(vm) log.debug(user.id, "Attaching disks") disk_devs = [] if i == 0 and disk_list: for disk_id in disk_list: log.debug(user.id, 'Attaching disks to first VM') disk = StorageImage.get(user.id, disk_id) if disk.vm != None: raise CMException('image_attached') while disk.disk_dev in disk_devs: disk.disk_dev += 1 disk_devs.append(disk.disk_dev) disk.vm = vm disk.save() log.debug(user.id, "Attaching CD") if i == 0 and iso_list: for iso_id in iso_list: log.debug(user.id, 'Attaching iso to first VM') # cd image have not be attached to any other vm iso = IsoImage.get(user.id, iso_id) iso.check_attached() vm.iso_image = iso vm.save() for i, vm in enumerate(vms): if lease_id != None: lease = Lease.objects.get(id=lease_id) if lease.user_network.user != user: raise CMException('lease_permission') if lease.vm != None: raise CMException('lease_attached') lease.vm = vm log.debug(user.id, "Attached ip: %s" % lease.address) else: lease = AvailableNetwork.get_lease(user) lease.vm = vm lease.save() log.debug(user.id, "Attached ip: %s" % lease.address) if i == 0 and public_ip_id > 0: log.debug(user.id, "Attaching PublicIP") try: publicip = PublicIP.objects.filter(user=user).get(id=public_ip_id) publicip.assign(lease) publicip.save() except Exception, e: log.exception(user.id, str(e)) raise CMException("lease_not_found")
def create(user, name, description, image_id, template_id, public_ip_id, iso_list, disk_list, vnc, groups, ssh_key=None, ssh_username=None, count=1, farm=None, head_template_id=None, node_id=False, lease_id=None, user_data=None): from cm.models.storage_image import StorageImage from cm.utils.threads.vm import VMThread template = Template.get(template_id) image = SystemImage.get(user.id, image_id, groups) if image.state != image_states['ok']: raise CMException('image_unavailable') if farm: head_template = Template.get(head_template_id) wn_template = template user.check_quota([(head_template, 1), (wn_template, count)]) count += 1 else: user.check_quota([(template, count)]) vms = [] reservation_id = None for i in range(count): # create VM instance log.debug(user.id, "Looking for node") node = Node.get_free_node( head_template, image, node_id) if farm and i == 0 else Node.get_free_node( template, image, node_id) log.info(user.id, 'Selected node: %d' % node.id) vm = VM() vm.libvirt_id = -1 if farm: if i == 0: vm.name = '%s-head' % name vm.description = 'Farm head' vm.template = head_template else: vm.name = '%s-wn%d' % (name, i) vm.description = 'Worker Node' vm.template = wn_template else: vm.template = template vm.description = description if count > 1: vm.name = '%s_%d' % (name, i + 1) else: vm.name = name vm.user = user vm.state = vm_states['init'] vm.start_time = datetime.now() vm.system_image = image vm.node = node vm.save_vm = True if farm: vm.farm = farm # Find first free vnc port used_ports = VM.objects.exclude( state__in=[vm_states['closed'], vm_states['erased'] ]).values_list('vnc_port', flat=True) for new_vnc_port in xrange(VNC_PORTS['START'], VNC_PORTS['END'] + 1): if new_vnc_port not in used_ports and new_vnc_port not in VNC_PORTS[ 'EXCLUDE']: break else: raise CMException('vm_vnc_not_found') log.debug(user.id, "Found vnc port: %d" % new_vnc_port) vm.vnc_port = new_vnc_port # Find first free novnc port used_ports = VM.objects.exclude( state__in=[vm_states['closed'], vm_states['erased'] ]).values_list('novnc_port', flat=True) for new_novnc_port in xrange(NOVNC_PORTS['START'], NOVNC_PORTS['END'] + 1): if new_novnc_port not in used_ports and new_novnc_port not in NOVNC_PORTS[ 'EXCLUDE']: break else: raise CMException('vm_novnc_not_found') log.debug(user.id, "Found novnc port: %d" % new_novnc_port) vm.novnc_port = new_novnc_port if vnc: vm.attach_vnc() vm.vnc_passwd = password_gen(13, chars=['letters', 'digits'], extra_chars='!@#$%^&*()') vm.ssh_key = ssh_key vm.ssh_username = ssh_username vm.user_data = user_data vm.save() if not reservation_id: reservation_id = vm.id vm.reservation_id = reservation_id vm.save() if farm and i == 0: farm.head = vm vms.append(vm) log.debug(user.id, "Attaching disks") disk_devs = [] if i == 0 and disk_list: for disk_id in disk_list: log.debug(user.id, 'Attaching disks to first VM') disk = StorageImage.get(user.id, disk_id) if disk.vm != None: raise CMException('image_attached') while disk.disk_dev in disk_devs: disk.disk_dev += 1 disk_devs.append(disk.disk_dev) disk.vm = vm disk.save() log.debug(user.id, "Attaching CD") if i == 0 and iso_list: for iso_id in iso_list: log.debug(user.id, 'Attaching iso to first VM') # cd image have not be attached to any other vm iso = IsoImage.get(user.id, iso_id) iso.check_attached() vm.iso_image = iso vm.save() for i, vm in enumerate(vms): if lease_id != None: lease = Lease.objects.get(id=lease_id) if lease.user_network.user != user: raise CMException('lease_permission') if lease.vm != None: raise CMException('lease_attached') lease.vm = vm log.debug(user.id, "Attached ip: %s" % lease.address) else: lease = AvailableNetwork.get_lease(user) lease.vm = vm lease.save() log.debug(user.id, "Attached ip: %s" % lease.address) if i == 0 and public_ip_id > 0: log.debug(user.id, "Attaching PublicIP") try: publicip = PublicIP.objects.filter(user=user).get( id=public_ip_id) publicip.assign(lease) publicip.save() except Exception, e: log.exception(user.id, str(e)) raise CMException("lease_not_found")