uncloud/opennebula/management/commands/opennebula-to-uncloud.py

import sys
from datetime import datetime

from django.core.management.base import BaseCommand
from django.utils import timezone
from django.contrib.auth import get_user_model

from opennebula.models import VM as VMModel
from uncloud_vm.models import VMHost, VMProduct, VMNetworkCard, VMDiskImageProduct, VMDiskProduct

from uncloud_pay.models import Order

import logging

log = logging.getLogger(__name__)

def convert_mac_to_int(mac_address: str):
    # Remove octet connecting characters
    mac_address = mac_address.replace(':', '')
    mac_address = mac_address.replace('.', '')
    mac_address = mac_address.replace('-', '')
    mac_address = mac_address.replace(' ', '')

    # Parse the resulting number as hexadecimal
    mac_address = int(mac_address, base=16)

    return mac_address


def get_vm_price(core, ram, ssd_size, hdd_size, n_of_ipv4, n_of_ipv6):
    total = 3 * core + 4 * ram + (3.5 * ssd_size/10.) + (1.5 * hdd_size/100.) + 8 * n_of_ipv4 + 0 * n_of_ipv6

    # TODO: Find some reason about the following magical subtraction.
    total -= 8

    return total


def create_nics(one_vm, vm_product):
    for nic in one_vm.nics:
        mac_address = convert_mac_to_int(nic.get('MAC'))
        ip_address = nic.get('IP', None) or nic.get('IP6_GLOBAL', None)

        VMNetworkCard.objects.update_or_create(
            mac_address=mac_address, vm=vm_product, defaults={'ip_address': ip_address}
        )


def sync_disk_and_image(one_vm, vm_product, disk_owner):
    """
    a) Check all opennebula disk if they are in the uncloud VM, if not add
    b) Check all uncloud disks and remove them if they are not in the opennebula VM
    """

    vmdisknum = 0

    one_disks_extra_data = []

    for disk in one_vm.disks:
        vmowner = one_vm.owner
        name = disk.get('image')
        vmdisknum += 1

        log.info("Checking disk {} for VM {}".format(name, one_vm))

        is_os_image, is_public, status = True, False, 'active'

        image_size_in_gb = disk.get('image_size_in_gb')
        disk_size_in_gb = disk.get('size_in_gb')
        storage_class = disk.get('storage_class')
        image_source = disk.get('source')
        image_source_type = disk.get('source_type')

        image, _ = VMDiskImageProduct.objects.update_or_create(
            name=name,
            defaults={
                'owner': disk_owner,
                'is_os_image': is_os_image,
                'is_public': is_public,
                'size_in_gb': image_size_in_gb,
                'storage_class': storage_class,
                'image_source': image_source,
                'image_source_type': image_source_type,
                'status': status
            }
        )

        # identify vmdisk from opennebula - primary mapping key
        extra_data = {
            'opennebula_vm': one_vm.vmid,
            'opennebula_size_in_gb': disk_size_in_gb,
            'opennebula_source': disk.get('opennebula_source'),
            'opennebula_disk_num': vmdisknum
        }
        # Save for comparing later
        one_disks_extra_data.append(extra_data)

        try:
            vm_disk = VMDiskProduct.objects.get(extra_data=extra_data)
        except VMDiskProduct.DoesNotExist:
            vm_disk = VMDiskProduct.objects.create(
                owner=vmowner,
                vm=vm_product,
                image=image,
                size_in_gb=disk_size_in_gb,
                extra_data=extra_data
            )

    # Now remove all disks that are not in above extra_data list
    for disk in VMDiskProduct.objects.filter(vm=vm_product):
        extra_data = disk.extra_data
        if not extra_data in one_disks_extra_data:
            log.info("Removing disk {} from VM {}".format(disk, vm_product))
            disk.delete()

    disks = [ disk.extra_data for disk in VMDiskProduct.objects.filter(vm=vm_product) ]
    log.info("VM {} has disks: {}".format(vm_product, disks))

class Command(BaseCommand):
    help = 'Migrate Opennebula VM to regular (uncloud) vm'

    def add_arguments(self, parser):
        parser.add_argument('--disk-owner', required=True, help="The user who owns the the opennebula disks")

    def handle(self, *args, **options):
        log.debug("{} {}".format(args, options))

        disk_owner = get_user_model().objects.get(username=options['disk_owner'])

        for one_vm in VMModel.objects.all():

            if not one_vm.last_host:
                log.warning("No VMHost for VM {} - VM might be on hold - skipping".format(one_vm.vmid))
                continue

            try:
                vmhost = VMHost.objects.get(hostname=one_vm.last_host)
            except VMHost.DoesNotExist:
                log.error("VMHost {} does not exist, aborting".format(one_vm.last_host))
                raise

            cores = one_vm.cores
            ram_in_gb = one_vm.ram_in_gb
            owner = one_vm.owner
            status = 'active'

            ssd_size = sum([ disk['size_in_gb'] for disk in one_vm.disks if disk['pool_name'] in ['ssd', 'one'] ])
            hdd_size = sum([ disk['size_in_gb'] for disk in one_vm.disks if disk['pool_name'] in ['hdd'] ])

            # List of IPv4 addresses and Global IPv6 addresses
            ipv4, ipv6 = one_vm.ips

            # TODO: Insert actual/real creation_date, starting_date, ending_date
            #       instead of pseudo one we are putting currently
            creation_date = starting_date = datetime.now(tz=timezone.utc)

            # Price calculation based on datacenterlight.ch
            one_time_price = 0
            recurring_period = 'per_month'
            recurring_price = get_vm_price(cores, ram_in_gb,
                                           ssd_size, hdd_size,
                                           len(ipv4), len(ipv6))

            try:
                vm_product = VMProduct.objects.get(extra_data__opennebula_id=one_vm.vmid)
            except VMProduct.DoesNotExist:
                order = Order.objects.create(
                    owner=owner,
                    creation_date=creation_date,
                    starting_date=starting_date
                )
                vm_product = VMProduct(
                    extra_data={ 'opennebula_id': one_vm.vmid },
                    name=one_vm.uncloud_name,
                    order=order
                )

            # we don't use update_or_create, as filtering by json AND setting json
            # at the same time does not work

            vm_product.vmhost = vmhost
            vm_product.owner = owner
            vm_product.cores = cores
            vm_product.ram_in_gb = ram_in_gb
            vm_product.status = status

            vm_product.save()

            # Create VMNetworkCards
            create_nics(one_vm, vm_product)

            # Create VMDiskImageProduct and VMDiskProduct
            sync_disk_and_image(one_vm, vm_product, disk_owner=disk_owner)