108 lines
3.9 KiB
Python
108 lines
3.9 KiB
Python
|
from datetime import datetime
|
||
|
|
|
||
|
|
from django.core.management.base import BaseCommand
|
||
|
|
from django.utils import timezone
|
||
|
|
|
||
|
|
from opennebula.models import VM as VMModel
|
||
|
|
from uncloud_vm.models import VMHost, VMProduct, VMNetworkCard, VMDiskImageProduct, VMDiskProduct
|
||
|
|
from uncloud_pay.models import Order
|
||
|
|
|
||
|
|
|
||
|
|
def get_vm_price(core, ram, storage, n_of_ipv4, n_of_ipv6):
|
||
|
|
storage = storage / 10
|
||
|
|
total = 3 * core + 4 * ram + 3.5 * storage + 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 = nic.get('MAC')
|
||
|
|
ip_address = nic.get('IP', None) or nic.get('IP6_GLOBAL', None)
|
||
|
|
|
||
|
|
mac_address = mac_address.replace(':', '')
|
||
|
|
mac_address = mac_address.replace('.', '')
|
||
|
|
mac_address = mac_address.replace('-', '')
|
||
|
|
mac_address = mac_address.replace(' ', '')
|
||
|
|
mac_address = int(mac_address, base=16)
|
||
|
|
|
||
|
|
VMNetworkCard.objects.create(
|
||
|
|
mac_address=mac_address, vm=vm_product, ip_address=ip_address
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def create_disk_and_image(one_vm, vm_product):
|
||
|
|
for disk in one_vm.disks:
|
||
|
|
owner = one_vm.owner
|
||
|
|
name = disk.get('image')
|
||
|
|
|
||
|
|
# TODO: Fix the following hard coded values
|
||
|
|
is_os_image = True
|
||
|
|
is_public = True
|
||
|
|
status = 'active'
|
||
|
|
|
||
|
|
image_size_in_gb = disk.get('image_size_in_gb')
|
||
|
|
disk_size_in_gb = disk.get('size_in_gb')
|
||
|
|
storage_class = disk.get('pool_name')
|
||
|
|
image_source = disk.get('source')
|
||
|
|
image_source_type = disk.get('source_type')
|
||
|
|
|
||
|
|
image = VMDiskImageProduct.objects.create(
|
||
|
|
owner=owner, name=name, 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
|
||
|
|
)
|
||
|
|
vm_disk = VMDiskProduct.objects.create(
|
||
|
|
owner=owner, vm=vm_product, image=image, size_in_gb=disk_size_in_gb
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
help = 'Migrate Opennebula VM to regular (uncloud) vm'
|
||
|
|
|
||
|
|
def add_arguments(self, parser):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def handle(self, *args, **options):
|
||
|
|
for one_vm in VMModel.objects.all():
|
||
|
|
# Host on which the VM is currently residing
|
||
|
|
host = VMHost.objects.filter(vms__icontains=one_vm.vmid).first()
|
||
|
|
|
||
|
|
# VCPU, RAM, Owner, Status
|
||
|
|
# TODO: Set actual status instead of hard coded 'active'
|
||
|
|
cores, ram_in_gb, owner, status = one_vm.cores, one_vm.ram_in_gb, one_vm.owner, 'active'
|
||
|
|
|
||
|
|
# Total Amount of SSD Storage
|
||
|
|
# TODO: What would happen if the attached storage is not SSD but HDD?
|
||
|
|
total_storage_in_gb = sum([disk['size_in_gb'] for disk in one_vm.disks])
|
||
|
|
|
||
|
|
# 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
|
||
|
|
order = Order.objects.create(
|
||
|
|
owner=one_vm.owner,
|
||
|
|
creation_date=datetime.now(tz=timezone.utc),
|
||
|
|
starting_date=datetime.now(tz=timezone.utc),
|
||
|
|
ending_date=datetime.now(tz=timezone.utc),
|
||
|
|
one_time_price=0,
|
||
|
|
recurring_price=get_vm_price(cores, ram_in_gb, total_storage_in_gb, len(ipv4), len(ipv6)),
|
||
|
|
recurring_period='per_month'
|
||
|
|
)
|
||
|
|
|
||
|
|
vm_product = VMProduct.objects.create(
|
||
|
|
cores=cores, ram_in_gb=ram_in_gb,
|
||
|
|
owner=one_vm.owner, vmhost=host,
|
||
|
|
order=order, status=status
|
||
|
|
)
|
||
|
|
|
||
|
|
# Create VMNetworkCards
|
||
|
|
create_nics(one_vm, vm_product)
|
||
|
|
|
||
|
|
# Create VMDiskImageProduct and VMDiskProduct
|
||
|
|
create_disk_and_image(one_vm, vm_product)
|