init commit

uncloud/opennebula/management/commands/migrate-one-vm-to-regular.py

@@ -0,0 +1,107 @@
+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)