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 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 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 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 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 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")
