Remove opennebula_functions.py
This commit is contained in:
		
					parent
					
						
							
								cfaf201064
							
						
					
				
			
			
				commit
				
					
						c816d280ee
					
				
			
		
					 1 changed files with 0 additions and 567 deletions
				
			
		| 
						 | 
				
			
			@ -1,567 +0,0 @@
 | 
			
		|||
import logging
 | 
			
		||||
import random
 | 
			
		||||
import socket
 | 
			
		||||
import string
 | 
			
		||||
 | 
			
		||||
import oca
 | 
			
		||||
from django.conf import settings
 | 
			
		||||
from django.conf.urls import url
 | 
			
		||||
from django.contrib import admin
 | 
			
		||||
from django.contrib import messages
 | 
			
		||||
from django.shortcuts import redirect
 | 
			
		||||
from django.template.response import TemplateResponse
 | 
			
		||||
from django.utils.translation import ugettext_lazy as _
 | 
			
		||||
from oca.exceptions import OpenNebulaException
 | 
			
		||||
from oca.pool import WrongNameError
 | 
			
		||||
 | 
			
		||||
from django import forms
 | 
			
		||||
# Get an instance of a logger
 | 
			
		||||
logger = logging.getLogger(__name__)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class OpenNebulaManager:
 | 
			
		||||
 | 
			
		||||
    VM_STATE = {
 | 
			
		||||
        '0': 'INIT',
 | 
			
		||||
        '1': 'PENDING',
 | 
			
		||||
        '2': 'HOLD',
 | 
			
		||||
        '3': 'ACTIVE',
 | 
			
		||||
        '4': 'STOPPED',
 | 
			
		||||
        '5': 'SUSPENDED',
 | 
			
		||||
        '6': 'DONE',
 | 
			
		||||
        '8': 'POWEROFF',
 | 
			
		||||
        '9': 'UNDEPLOYED',
 | 
			
		||||
        '10': 'CLONING',
 | 
			
		||||
        '11': 'CLONING_FAILURE',
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    def __init__(self, email=None, password=None):
 | 
			
		||||
 | 
			
		||||
        # Get oneadmin client
 | 
			
		||||
        self.oneadmin_client = self._get_opennebula_client(
 | 
			
		||||
            settings.OPENNEBULA_USERNAME,
 | 
			
		||||
            settings.OPENNEBULA_PASSWORD
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
        # Get or create oppenebula user using given credentials
 | 
			
		||||
        self.opennebula_user = self._get_or_create_user(
 | 
			
		||||
            email,
 | 
			
		||||
            password
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
        # If opennebula user was created/obtained, get his client
 | 
			
		||||
        if self.opennebula_user:
 | 
			
		||||
            self.client = self._get_opennebula_client(
 | 
			
		||||
                email,
 | 
			
		||||
                password
 | 
			
		||||
            )
 | 
			
		||||
 | 
			
		||||
    def _get_opennebula_client(self, username, password):
 | 
			
		||||
        return oca.Client("{0}:{1}".format(
 | 
			
		||||
            username,
 | 
			
		||||
            password),
 | 
			
		||||
            "{protocol}://{domain}:{port}{endpoint}".format(
 | 
			
		||||
                protocol=settings.OPENNEBULA_PROTOCOL,
 | 
			
		||||
                domain=settings.OPENNEBULA_DOMAIN,
 | 
			
		||||
                port=settings.OPENNEBULA_PORT,
 | 
			
		||||
                endpoint=settings.OPENNEBULA_ENDPOINT
 | 
			
		||||
        ))
 | 
			
		||||
 | 
			
		||||
    def _get_or_create_user(self, email, password):
 | 
			
		||||
        try:
 | 
			
		||||
            user_pool = oca.UserPool(self.oneadmin_client)
 | 
			
		||||
            user_pool.info()
 | 
			
		||||
            opennebula_user = user_pool.get_by_name(email)
 | 
			
		||||
            return opennebula_user
 | 
			
		||||
        except WrongNameError as wrong_name_err:
 | 
			
		||||
            # TODO: Store this password so that we can use it later to 
 | 
			
		||||
            # connect to opennebula
 | 
			
		||||
            opennebula_user = self.oneadmin_client.call(oca.User.METHODS['allocate'], email,
 | 
			
		||||
                                                        password, 'core')
 | 
			
		||||
            logger.debug(
 | 
			
		||||
                "User {0} does not exist. Created the user. User id = {1}",
 | 
			
		||||
                email,
 | 
			
		||||
                opennebula_user
 | 
			
		||||
            )
 | 
			
		||||
            return opennebula_user
 | 
			
		||||
        except OpenNebulaException as err:
 | 
			
		||||
            logger.error("Error : {0}".format(err))
 | 
			
		||||
 | 
			
		||||
    @classmethod
 | 
			
		||||
    def get_vm_state(self, state):
 | 
			
		||||
        return self.VM_STATE.get(str(state))
 | 
			
		||||
 | 
			
		||||
    @classmethod
 | 
			
		||||
    def parse_vm(self, vm):
 | 
			
		||||
        name = vm.name
 | 
			
		||||
        cores = int(vm.template.vcpu)
 | 
			
		||||
        memory = int(vm.template.memory) / 1024
 | 
			
		||||
        # Check if vm has more than one disk
 | 
			
		||||
        if 'DISK' in vm.template.multiple:
 | 
			
		||||
            disk_size = 0
 | 
			
		||||
            for disk in vm.template.disks:
 | 
			
		||||
                disk_size += int(disk.size) / 1024
 | 
			
		||||
        else:
 | 
			
		||||
            disk_size = int(vm.template.disk.size) / 1024
 | 
			
		||||
 | 
			
		||||
        #TODO: Replace with vm plan
 | 
			
		||||
        price = 0.6 * disk_size + 2 * memory + 5 * cores
 | 
			
		||||
        vm_data = {}
 | 
			
		||||
        vm_data['name'] = name
 | 
			
		||||
        vm_data['price'] = price
 | 
			
		||||
        vm_data['disk_size'] = disk_size
 | 
			
		||||
        vm_data['cores'] = cores
 | 
			
		||||
        vm_data['memory'] = memory
 | 
			
		||||
        vm_data['deploy_id'] = vm.deploy_id
 | 
			
		||||
        vm_data['id'] = vm.id
 | 
			
		||||
        vm_data['state'] = self.get_vm_state(vm.state)
 | 
			
		||||
 | 
			
		||||
        return vm_data
 | 
			
		||||
 | 
			
		||||
    def change_user_password(self, new_password):
 | 
			
		||||
        self.oneadmin_client.call(
 | 
			
		||||
            oca.User.METHODS['passwd'],
 | 
			
		||||
            self.opennebula_user.id,
 | 
			
		||||
            new_password
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
    def create_vm(self, specs):
 | 
			
		||||
        vm_id = None
 | 
			
		||||
        try:
 | 
			
		||||
 | 
			
		||||
            # We do have the vm_template param set. Get and parse it
 | 
			
		||||
            # and check it to be in the desired range.
 | 
			
		||||
            # We have 8 possible VM templates for the moment which are 1x, 2x, 4x ...
 | 
			
		||||
            # the basic template of 10GB disk, 1GB ram, 1 vcpu, 0.1 cpu
 | 
			
		||||
            vm_string_formatter = """<VM>
 | 
			
		||||
                                      <MEMORY>{memory}</MEMORY>
 | 
			
		||||
                                      <VCPU>{vcpu}</VCPU>
 | 
			
		||||
                                      <CPU>{cpu}</CPU>
 | 
			
		||||
                                      <DISK>
 | 
			
		||||
                                        <TYPE>{disk_type}</TYPE>
 | 
			
		||||
                                        <SIZE>{size}</SIZE>
 | 
			
		||||
                                      </DISK>
 | 
			
		||||
                                      <CONTEXT>
 | 
			
		||||
                                        <SSH_PUBLIC_KEY>{ssh_key}</SSH_PUBLIC_KEY>
 | 
			
		||||
                                      </CONTEXT>
 | 
			
		||||
                                    </VM>
 | 
			
		||||
                                    """
 | 
			
		||||
            vm_id = oca.VirtualMachine.allocate(
 | 
			
		||||
                self.oneadmin_client,
 | 
			
		||||
                vm_string_formatter.format(
 | 
			
		||||
                    memory=1024 * specs.get('memory'),
 | 
			
		||||
                    vcpu=specs.get('cores'),
 | 
			
		||||
                    cpu=0.1 * specs.get('cores'),
 | 
			
		||||
                    disk_type='fs',
 | 
			
		||||
                    size=10000 * specs.get('disk_size'),
 | 
			
		||||
                    ssh_key=specs.get('ssh_key')
 | 
			
		||||
                )
 | 
			
		||||
            )
 | 
			
		||||
 | 
			
		||||
            self.oneadmin_client.call(
 | 
			
		||||
                oca.VirtualMachine.METHODS['chown'],
 | 
			
		||||
                vm_id,
 | 
			
		||||
                self.opennebula_user.id,
 | 
			
		||||
                self.opennebula_user.group_ids[0]
 | 
			
		||||
            )
 | 
			
		||||
 | 
			
		||||
        except socket.timeout as socket_err:
 | 
			
		||||
            logger.error("Socket timeout error: {0}".format(socket_err))
 | 
			
		||||
        except OpenNebulaException as opennebula_err:
 | 
			
		||||
            logger.error("OpenNebulaException error: {0}".format(opennebula_err))
 | 
			
		||||
        except OSError as os_err:
 | 
			
		||||
            logger.error("OSError : {0}".format(os_err))
 | 
			
		||||
        except ValueError as value_err:
 | 
			
		||||
            logger.error("ValueError : {0}".format(value_err))
 | 
			
		||||
 | 
			
		||||
        return vm_id
 | 
			
		||||
 | 
			
		||||
    def terminate_vm(self, vm_id):
 | 
			
		||||
 | 
			
		||||
        TERMINATE_ACTION = 'terminate'
 | 
			
		||||
        vm_terminated = False
 | 
			
		||||
 | 
			
		||||
        try:
 | 
			
		||||
            self.oneadmin_client.call(
 | 
			
		||||
                oca.VirtualMachine.METHODS['action'],
 | 
			
		||||
                TERMINATE_ACTION,
 | 
			
		||||
                int(vm_id),
 | 
			
		||||
            )
 | 
			
		||||
            vm_terminated = True
 | 
			
		||||
        except socket.timeout as socket_err:
 | 
			
		||||
            logger.error("Socket timeout error: {0}".format(socket_err))
 | 
			
		||||
        except OpenNebulaException as opennebula_err:
 | 
			
		||||
            logger.error("OpenNebulaException error: {0}".format(opennebula_err))
 | 
			
		||||
        except OSError as os_err:
 | 
			
		||||
            logger.error("OSError : {0}".format(os_err))
 | 
			
		||||
        except ValueError as value_err:
 | 
			
		||||
            logger.error("ValueError : {0}".format(value_err))
 | 
			
		||||
 | 
			
		||||
        return vm_terminated
 | 
			
		||||
 | 
			
		||||
    def get_vm_templates(self):
 | 
			
		||||
        template_pool = oca.VmTemplatePool(self.oneadmin_client)
 | 
			
		||||
        template_pool.info()
 | 
			
		||||
        return template_pool
 | 
			
		||||
 | 
			
		||||
    def get_vm(self, email, vm_id):
 | 
			
		||||
        # Get vm's
 | 
			
		||||
        vms = self.get_vms(email)
 | 
			
		||||
 | 
			
		||||
        # Filter vm by given id
 | 
			
		||||
        return vms.get_by_id(int(vm_id))
 | 
			
		||||
 | 
			
		||||
    def get_vms(self, email):
 | 
			
		||||
        client = self.oneadmin_client
 | 
			
		||||
 | 
			
		||||
        # Get open nebula user id for given email
 | 
			
		||||
        user_pool = oca.UserPool(client)
 | 
			
		||||
        user_pool.info()
 | 
			
		||||
 | 
			
		||||
        # TODO: handle potential name error
 | 
			
		||||
        user_id = user_pool.get_by_name(email).id
 | 
			
		||||
 | 
			
		||||
        # Get vm_pool for given user_id
 | 
			
		||||
        vm_pool = oca.VirtualMachinePool(client)
 | 
			
		||||
        vm_pool.info()
 | 
			
		||||
 | 
			
		||||
        # TODO: this is just to test with oneadmin user, remove this
 | 
			
		||||
        # user_id = 0
 | 
			
		||||
        vm_pool.info(filter=user_id)
 | 
			
		||||
 | 
			
		||||
        return vm_pool
 | 
			
		||||
 | 
			
		||||
class HostingManageVMAdmin(admin.ModelAdmin):
 | 
			
		||||
    client = None
 | 
			
		||||
    oneadmin_client = None
 | 
			
		||||
 | 
			
		||||
    def get_urls(self):
 | 
			
		||||
        urls = super().get_urls()
 | 
			
		||||
        my_urls = [
 | 
			
		||||
            url(r'^$', self.admin_site.admin_view(self.show_vms, cacheable=True), name='showvms'),
 | 
			
		||||
            url(r'^create_vm/$', self.admin_site.admin_view(self.create_vm, cacheable=True), name='createvm'),
 | 
			
		||||
            url(r'^delete_vm/(?P<vmid>\d+)/$', self.admin_site.admin_view(self.delete_vm, cacheable=True),
 | 
			
		||||
                name='deletevm'),
 | 
			
		||||
            url(r'^stop_vm/(?P<vmid>\d+)/$', self.admin_site.admin_view(self.stop_vm, cacheable=True), name='stopvm'),
 | 
			
		||||
            url(r'^start_vm/(?P<vmid>\d+)/$', self.admin_site.admin_view(self.start_vm, cacheable=True),
 | 
			
		||||
                name='startvm'),
 | 
			
		||||
        ]
 | 
			
		||||
        return my_urls + urls
 | 
			
		||||
 | 
			
		||||
    # Function to initialize opennebula client based on the logged in
 | 
			
		||||
    # user
 | 
			
		||||
    def init_opennebula_client(self, request):
 | 
			
		||||
        if self.oneadmin_client is None:
 | 
			
		||||
            self.oneadmin_client = oca.Client("{0}:{1}".format(settings.OPENNEBULA_USERNAME,
 | 
			
		||||
                                                               settings.OPENNEBULA_PASSWORD),
 | 
			
		||||
                                              "{protocol}://{domain}:{port}{endpoint}".format(
 | 
			
		||||
                                                  protocol=settings.OPENNEBULA_PROTOCOL,
 | 
			
		||||
                                                  domain=settings.OPENNEBULA_DOMAIN,
 | 
			
		||||
                                                  port=settings.OPENNEBULA_PORT,
 | 
			
		||||
                                                  endpoint=settings.OPENNEBULA_ENDPOINT
 | 
			
		||||
                                              ))
 | 
			
		||||
            logger.debug("{0}:{1}".format(settings.OPENNEBULA_USERNAME,
 | 
			
		||||
                                   settings.OPENNEBULA_PASSWORD))
 | 
			
		||||
            logger.debug("{protocol}://{domain}:{port}{endpoint}".format(
 | 
			
		||||
                protocol=settings.OPENNEBULA_PROTOCOL,
 | 
			
		||||
                domain=settings.OPENNEBULA_DOMAIN,
 | 
			
		||||
                port=settings.OPENNEBULA_PORT,
 | 
			
		||||
                endpoint=settings.OPENNEBULA_ENDPOINT
 | 
			
		||||
            ))
 | 
			
		||||
            self.create_opennebula_user(request)
 | 
			
		||||
        if self.client is None:
 | 
			
		||||
            opennebula_user = request.user.email
 | 
			
		||||
            # TODO: get the password stored in django
 | 
			
		||||
            opennebula_user_password ='19737450'
 | 
			
		||||
            self.client = oca.Client("{0}:{1}".format(opennebula_user, opennebula_user_password),
 | 
			
		||||
                                     "{protocol}://{domain}:{port}{endpoint}".format(
 | 
			
		||||
                                         protocol=settings.OPENNEBULA_PROTOCOL,
 | 
			
		||||
                                         domain=settings.OPENNEBULA_DOMAIN,
 | 
			
		||||
                                         port=settings.OPENNEBULA_PORT,
 | 
			
		||||
                                         endpoint=settings.OPENNEBULA_ENDPOINT
 | 
			
		||||
                                     ))
 | 
			
		||||
 | 
			
		||||
    # Function that shows the VMs of the current user
 | 
			
		||||
    def show_vms(self, request):
 | 
			
		||||
        vm_pool = None
 | 
			
		||||
        try:
 | 
			
		||||
            self.init_opennebula_client(request)
 | 
			
		||||
            vm_pool = oca.VirtualMachinePool(self.client)
 | 
			
		||||
            vm_pool.info()
 | 
			
		||||
        except socket.timeout:
 | 
			
		||||
            messages.add_message(request, messages.ERROR, _("Socket timeout error."))
 | 
			
		||||
        except OpenNebulaException as opennebula_err:
 | 
			
		||||
            messages.add_message(request, messages.ERROR, _("OpenNebulaException occurred. {0}".format(opennebula_err)))
 | 
			
		||||
        except OSError as err:
 | 
			
		||||
            messages.add_message(request, messages.ERROR, "OS error: {0}".format(err))
 | 
			
		||||
        context = dict(
 | 
			
		||||
            # Include common variables for rendering the admin template.
 | 
			
		||||
            self.admin_site.each_context(request),
 | 
			
		||||
            vms=vm_pool,
 | 
			
		||||
            form=HostingManageVMForm
 | 
			
		||||
        )
 | 
			
		||||
        return TemplateResponse(request, "hosting/managevms.html", context)
 | 
			
		||||
 | 
			
		||||
    # Function that shows the VMs of the current user
 | 
			
		||||
    def show_vms_view(self, request):
 | 
			
		||||
        """
 | 
			
		||||
            Implemented by Levi for the API
 | 
			
		||||
        """ 
 | 
			
		||||
        vm_pool = None
 | 
			
		||||
        try:
 | 
			
		||||
            self.init_opennebula_client(request)
 | 
			
		||||
            vm_pool = oca.VirtualMachinePool(self.client)
 | 
			
		||||
            vm_pool.info()
 | 
			
		||||
        except socket.timeout as socket_err:
 | 
			
		||||
            logger.error("Socket timeout error.".format(socket_err))
 | 
			
		||||
        except OpenNebulaException as opennebula_err:
 | 
			
		||||
            logger.error("OpenNebulaException error: {0}".format(opennebula_err))
 | 
			
		||||
        except OSError as os_err:
 | 
			
		||||
            logger.error("OSError : {0}".format(os_err))
 | 
			
		||||
        except ValueError as value_err:
 | 
			
		||||
            logger.error("ValueError : {0}".format(value_err))
 | 
			
		||||
        context = dict(
 | 
			
		||||
            # Include common variables for rendering the admin template.
 | 
			
		||||
            # self.admin_site.each_context(request),
 | 
			
		||||
            vms=vm_pool,
 | 
			
		||||
        )
 | 
			
		||||
        return context
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    def create_vm_view(self, specs):
 | 
			
		||||
        vm_id = None
 | 
			
		||||
        try:
 | 
			
		||||
            # We do have the vm_template param set. Get and parse it
 | 
			
		||||
            # and check it to be in the desired range.
 | 
			
		||||
            # We have 8 possible VM templates for the moment which are 1x, 2x, 4x ...
 | 
			
		||||
            # the basic template of 10GB disk, 1GB ram, 1 vcpu, 0.1 cpu
 | 
			
		||||
            vm_string_formatter = """<VM>
 | 
			
		||||
                                      <MEMORY>{memory}</MEMORY>
 | 
			
		||||
                                      <VCPU>{vcpu}</VCPU>
 | 
			
		||||
                                      <CPU>{cpu}</CPU>
 | 
			
		||||
                                      <DISK>
 | 
			
		||||
                                        <TYPE>{disk_type}</TYPE>
 | 
			
		||||
                                        <SIZE>{size}</SIZE>
 | 
			
		||||
                                      </DISK>
 | 
			
		||||
                                    </VM>
 | 
			
		||||
                                    """
 | 
			
		||||
            vm_id = oca.VirtualMachine.allocate(
 | 
			
		||||
                self.client,
 | 
			
		||||
                vm_string_formatter.format(
 | 
			
		||||
                    memory=1024 * specs.get('memory'),
 | 
			
		||||
                    vcpu=specs.get('cores'),
 | 
			
		||||
                    cpu=0.1 * specs.get('cores'),
 | 
			
		||||
                    disk_type='fs',
 | 
			
		||||
                    size=10000 * specs.get('disk_size')
 | 
			
		||||
                )
 | 
			
		||||
            )
 | 
			
		||||
                # message = _("Created with id = " + str(vm_id))
 | 
			
		||||
                # messages.add_message(request, messages.SUCCESS, message)
 | 
			
		||||
        except socket.timeout as socket_err:
 | 
			
		||||
            logger.error("Socket timeout error: {0}".format(socket_err))
 | 
			
		||||
        except OpenNebulaException as opennebula_err:
 | 
			
		||||
            logger.error("OpenNebulaException error: {0}".format(opennebula_err))
 | 
			
		||||
        except OSError as os_err:
 | 
			
		||||
            logger.error("OSError : {0}".format(os_err))
 | 
			
		||||
        except ValueError as value_err:
 | 
			
		||||
            logger.error("ValueError : {0}".format(value_err))
 | 
			
		||||
 | 
			
		||||
        return vm_id
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    # Creating VM by using method allocate(client, template)
 | 
			
		||||
    def create_vm(self, request):
 | 
			
		||||
        # check if the request contains the template parameter, if it is
 | 
			
		||||
        # not set warn the user of setting this.
 | 
			
		||||
        vm_template = request.POST.get('vm_template')
 | 
			
		||||
        if vm_template == 'select':
 | 
			
		||||
            messages.add_message(request, messages.ERROR, "Please select a vm template")
 | 
			
		||||
        else:
 | 
			
		||||
            try:
 | 
			
		||||
                # We do have the vm_template param set. Get and parse it
 | 
			
		||||
                # and check it to be in the desired range.
 | 
			
		||||
                # We have 8 possible VM templates for the moment which are 1x, 2x, 4x ...
 | 
			
		||||
                # the basic template of 10GB disk, 1GB ram, 1 vcpu, 0.1 cpu
 | 
			
		||||
                vm_template_int = int(vm_template)
 | 
			
		||||
                if 1 <= vm_template_int <= 8:
 | 
			
		||||
                    vm_string_formatter = """<VM>
 | 
			
		||||
                                              <CONTEXT>
 | 
			
		||||
                                                <SSH_PUBLIC_KEY>
 | 
			
		||||
                                                 {ssh_key}
 | 
			
		||||
                                                </SSH_PUBLIC_KEY>
 | 
			
		||||
                                              </CONTEXT>                    
 | 
			
		||||
                                              <MEMORY>{memory}</MEMORY>
 | 
			
		||||
                                              <VCPU>{vcpu}</VCPU>
 | 
			
		||||
                                              <CPU>{cpu}</CPU>
 | 
			
		||||
                                              <DISK>
 | 
			
		||||
                                                <TYPE>{disk_type}</TYPE>
 | 
			
		||||
                                                <SIZE>{size}</SIZE>
 | 
			
		||||
                                                <DEV_PREFIX>{dev_prefix}</DEV_PREFIX>
 | 
			
		||||
                                              </DISK>
 | 
			
		||||
                                            </VM>
 | 
			
		||||
                                            """
 | 
			
		||||
                    vm_id = oca.VirtualMachine.allocate(self.client,
 | 
			
		||||
                          vm_string_formatter.format(
 | 
			
		||||
                              ssh_key='',                    # public key of the user
 | 
			
		||||
                              memory=1024 * vm_template_int, # memory in MB
 | 
			
		||||
                              vcpu=vm_template_int,          # vpcu
 | 
			
		||||
                              cpu=0.1 * vm_template_int,     # cpu
 | 
			
		||||
                              disk_type='fs',      
 | 
			
		||||
                              size=10000 * vm_template_int,
 | 
			
		||||
                              dev_prefix='vd'))              # We need KVM virtual disk
 | 
			
		||||
                    message = _("Created with id = " + str(vm_id))
 | 
			
		||||
                    messages.add_message(request, messages.SUCCESS, message)
 | 
			
		||||
                else:
 | 
			
		||||
                    messages.add_message(request, messages.ERROR,
 | 
			
		||||
                                         _("Please select an appropriate value for vm template."))
 | 
			
		||||
            except socket.timeout as socket_err:
 | 
			
		||||
                messages.add_message(request, messages.ERROR, _("Socket timeout error."))
 | 
			
		||||
                logger.error("Socket timeout error: {0}".format(socket_err))
 | 
			
		||||
            except OpenNebulaException as opennebula_err:
 | 
			
		||||
                messages.add_message(request, messages.ERROR,
 | 
			
		||||
                                     _("OpenNebulaException occurred. {0}".format(opennebula_err)))
 | 
			
		||||
                logger.error("OpenNebulaException error: {0}".format(opennebula_err))
 | 
			
		||||
            except OSError as os_err:
 | 
			
		||||
                messages.add_message(request, messages.ERROR, _("OS error: {0}".format(os_err)))
 | 
			
		||||
                logger.error("OSError : {0}".format(os_err))
 | 
			
		||||
            except ValueError as value_err:
 | 
			
		||||
                messages.add_message(request, messages.ERROR,
 | 
			
		||||
                                     _("Please select an appropriate value for vm template."))
 | 
			
		||||
                logger.error("ValueError : {0}".format(value_err))
 | 
			
		||||
        return redirect('admin:showvms')
 | 
			
		||||
 | 
			
		||||
    # Delete VM from the pool and DB by using method finalize()
 | 
			
		||||
    def delete_vm(self, request, vmid):
 | 
			
		||||
        vm_id = int(vmid)
 | 
			
		||||
        # get the desired vm from the pool
 | 
			
		||||
        logger.debug("Deleting vm with id {0}".format(vm_id))
 | 
			
		||||
        vm = self.get_vm_by_id(vm_id)
 | 
			
		||||
        if vm is None:
 | 
			
		||||
            messages.add_message(request, messages.ERROR, _("Did not find a vm with id = {0}".format(vm_id)))
 | 
			
		||||
        else:
 | 
			
		||||
            logger.debug("Deleting vm_id = " + str(vm_id) + " state = " + vm.str_state)
 | 
			
		||||
            if vm.str_state == 'PENDING' or vm.str_state == 'POWEROFF' or vm.str_state == 'ACTIVE':
 | 
			
		||||
                vm.delete()
 | 
			
		||||
                messages.add_message(request, messages.SUCCESS,
 | 
			
		||||
                                     _("Deleted from {0} state vm with id = {1}".format(vm.str_state, str(vm_id))))
 | 
			
		||||
            else:
 | 
			
		||||
                vm.finalize()
 | 
			
		||||
                messages.add_message(request, messages.SUCCESS,
 | 
			
		||||
                                     _("Deleted (using finalize()) from {0} state vm with id = {1}".format(vm.str_state,
 | 
			
		||||
                                                                                                           str(vm_id))))
 | 
			
		||||
        return redirect('admin:showvms')
 | 
			
		||||
 | 
			
		||||
    def stop_vm(self, request, vmid):
 | 
			
		||||
        vm_id = int(vmid)
 | 
			
		||||
        vm = self.get_vm_by_id(vm_id)
 | 
			
		||||
        if vm is None:
 | 
			
		||||
            messages.add_message(request, messages.ERROR, _("Did not find a vm with id = {0}", vm_id))
 | 
			
		||||
        else:
 | 
			
		||||
            vm.stop()
 | 
			
		||||
            messages.add_message(request, messages.SUCCESS, _("Stopped the vm with id = {0}", vm_id))
 | 
			
		||||
        return redirect('admin:showvms')
 | 
			
		||||
 | 
			
		||||
    def start_vm(self, request, vmid):
 | 
			
		||||
        vm_id = int(vmid)
 | 
			
		||||
        vm = self.get_vm_by_id(vm_id)
 | 
			
		||||
        if vm is None:
 | 
			
		||||
            messages.add_message(request, messages.ERROR, _("Did not find a vm with id = {0}", vm_id))
 | 
			
		||||
        else:
 | 
			
		||||
            vm.resume()
 | 
			
		||||
            messages.add_message(request, messages.SUCCESS, _("Started the vm with id = {0}", vm_id))
 | 
			
		||||
        return redirect('admin:showvms')
 | 
			
		||||
 | 
			
		||||
    # Retrives virtual machine pool information
 | 
			
		||||
    def get_vm_pool(self):
 | 
			
		||||
        vm_pool = oca.VirtualMachinePool(self.client)
 | 
			
		||||
        vm_pool.info()
 | 
			
		||||
        return vm_pool
 | 
			
		||||
 | 
			
		||||
    def get_vm_by_id(self, vmid):
 | 
			
		||||
        vm_pool = self.get_vm_pool()
 | 
			
		||||
        return vm_pool.get_by_id(vmid)
 | 
			
		||||
 | 
			
		||||
    def create_opennebula_user(self, request):
 | 
			
		||||
        # Notes:
 | 
			
		||||
        #     1. python-oca library's oca.User.allocate(client, user, pass)
 | 
			
		||||
        #     method does not work with python-oca version oca-4.15.0a1-py3.5
 | 
			
		||||
        #     This is because the call is missing a fourth parameter
 | 
			
		||||
        #     auth_driver.
 | 
			
		||||
        #     To overcome this issue, we make a direct call to xml-rpc method
 | 
			
		||||
        #     'user.allocate' passing this fourth parameter.
 | 
			
		||||
        #
 | 
			
		||||
        #     2. We have a dummy authentication driver in opennebula and we
 | 
			
		||||
        #     use this so as to avoid opennebula authentication. However, we
 | 
			
		||||
        #     need to supply a dummy password. Without this, we can not
 | 
			
		||||
        #     create an OpenNebula user. We use dummy string 'a' as password
 | 
			
		||||
        #     for all users.
 | 
			
		||||
        #
 | 
			
		||||
        #     3. We user the user's email as the user name.
 | 
			
		||||
        #     4. If the user's email is not registered with OpenNebula,
 | 
			
		||||
        #     WrongNameError is raised. We create an OpenNebula user in 
 | 
			
		||||
        #     such case.
 | 
			
		||||
        try:
 | 
			
		||||
            user_pool = oca.UserPool(self.oneadmin_client)
 | 
			
		||||
            user_pool.info()
 | 
			
		||||
            opennebula_user = user_pool.get_by_name(request.user.email)
 | 
			
		||||
            logger.debug("User {0} exists. User id = {1}".format(request.user.email, opennebula_user.id))
 | 
			
		||||
        except WrongNameError as wrong_name_err:
 | 
			
		||||
            # TODO: Store this password so that we can use it later to 
 | 
			
		||||
            # connect to opennebula
 | 
			
		||||
            password = get_random_password()
 | 
			
		||||
            oca.User.allocate(self.oneadmin_client, request.user.email, password)
 | 
			
		||||
            logger.debug("User {0} does not exist. Created the user. User id = {1}", request.user.email, user_id)
 | 
			
		||||
        except OpenNebulaException as err:
 | 
			
		||||
            messages.add_message(request, messages.ERROR,
 | 
			
		||||
                                 "Error : {0}".format(err))
 | 
			
		||||
            logger.error("Error : {0}".format(err))
 | 
			
		||||
 | 
			
		||||
def set_field_html_name(cls, new_name):
 | 
			
		||||
    """
 | 
			
		||||
    This creates wrapper around the normal widget rendering, 
 | 
			
		||||
    allowing for a custom field name (new_name).
 | 
			
		||||
    """
 | 
			
		||||
    old_render = cls.widget.render
 | 
			
		||||
    def _widget_render_wrapper(name, value, attrs=None):
 | 
			
		||||
        return old_render(new_name, value, attrs)
 | 
			
		||||
    cls.widget.render = _widget_render_wrapper
 | 
			
		||||
            
 | 
			
		||||
class HostingManageVMForm(forms.Form):
 | 
			
		||||
    vm_templates = []
 | 
			
		||||
    VM_CHOICES = (('1', 'disk = 10GB, vcpu=1, ram=1GB'),
 | 
			
		||||
                  ('2', 'disk = 20GB, vcpu=2, ram=2GB'),
 | 
			
		||||
                  ('3', 'disk = 40GB, vcpu=4, ram=4GB'),
 | 
			
		||||
                  ('4', 'disk = 80GB, vcpu=8, ram=8GB'),
 | 
			
		||||
                  ('5', 'disk = 160GB, vcpu=16, ram=16GB'),
 | 
			
		||||
                  ('6', 'disk = 320GB, vcpu=32, ram=32GB'),
 | 
			
		||||
                  ('7', 'disk = 640GB, vcpu=64, ram=64GB'),
 | 
			
		||||
                  ('8', 'disk = 1280GB, vcpu=128, ram=128GB'))    
 | 
			
		||||
    #for i in range(0,8):
 | 
			
		||||
    #    factor = pow(2, i)
 | 
			
		||||
    #    vm_templates.append(VMTemplate(i, VM_CHOICES[i], 10000 * factor, factor , 0.1 * factor, 1024 * factor))
 | 
			
		||||
    field = forms.ChoiceField(label="Choose a VM Template ", choices=VM_CHOICES, widget=forms.Select(attrs={"id": "vm_template"}))
 | 
			
		||||
    set_field_html_name(field, 'vm_template')
 | 
			
		||||
        
 | 
			
		||||
        
 | 
			
		||||
 | 
			
		||||
class VMTemplate:
 | 
			
		||||
    """A simple representation of a VM template.
 | 
			
		||||
 | 
			
		||||
    :param template_id: The id of the template
 | 
			
		||||
    :param label: A string representation describing the template. Used as the label in view
 | 
			
		||||
    :param disk: VM disk space in MB
 | 
			
		||||
    :param vcpu: Virtual cpu for the VM
 | 
			
		||||
    :param cpu: CPU for the VM
 | 
			
		||||
    :param ram: The RAM for the VM
 | 
			
		||||
    """
 | 
			
		||||
    def __init__(self, template_id, label, disk, vcpu, cpu, ram):
 | 
			
		||||
        self.template_id = template_id
 | 
			
		||||
        self.label = label
 | 
			
		||||
        self.disk = disk
 | 
			
		||||
        self.vcpu = vcpu
 | 
			
		||||
        self.cpu = cpu
 | 
			
		||||
 | 
			
		||||
# Returns random password that is needed by OpenNebula
 | 
			
		||||
def get_random_password():
 | 
			
		||||
    return ''.join(random.SystemRandom().choice(string.ascii_uppercase + string.digits) for _ in range(20))
 | 
			
		||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue