NewNetworkAdapter

Complexity

Total Complexity

17

Size/Duplication

Total Lines	90
Duplicated Lines	0 %

2 Methods

Rating	Name	Duplication	Size	Complexity
C	get_vm_reconfig_spec()	0	57	10
C	run()	0	30	7

import sys

from pyVmomi import vim

from vmwarelib import inventory
from vmwarelib.actions import BaseAction


class NewNetworkAdapter(BaseAction):

    @staticmethod
    def get_vm_reconfig_spec(distributed_switch_obj, mac_address, network_obj, port_key, stay_connected, network_type, wake_on_lan):
        # creating virtual device
        if network_type == 'e1000':
            virtual_network = vim.vm.device.VirtualE1000()
        elif network_type == 'flexible':
            virtual_network = vim.vm.device.VirtualPCNet32()
            lance_option = vim.vm.device.VirtualPCNet32Option()
            lance_option.supportsMorphing = True;
            #TODO and question for Manas: We need to plug in that lance_option on something. I could not find any class that uses VirtualPCNet32Option or any of its upstream classes.
        elif network_type == 'vmxnet':
            virtual_network = vim.vm.device.VirtualVmxnet()
        elif network_type == 'enhancedvmxnet':
            virtual_network = vim.vm.device.VirtualVmxnet2()
        elif network_type == 'vmxnet3':
            virtual_network = vim.vm.device.VirtualVmxnet3()
        else:
            virtual_network = vim.vm.device.VirtualEthernetCard()

        virtual_network.wakeOnLanEnabled = wake_on_lan
        connect_info = vim.vm.device.VirtualDevice.ConnectInfo()
        connect_info.startConnected = stay_connected
        virtual_network.connectable = connect_info
        if mac_address:
            virtual_network.macAddress = mac_address
            virtual_network.addressType = vim.vm.device.VirtualEthernetCardOption.MacTypes.manual

        # creating backing info
        backing_info = None
        if distributed_switch_obj:
            dvs_port_connection = vim.dvs.PortConnection()
            dvs_port_connection.switchUuid = distributed_switch_obj.uuid
            if port_key:
                dvs_port_connection.portKey = port_key
                port_criteria = vim.dvs.PortCriteria()
                port_criteria.portKey = port_key
                ports = distributed_switch_obj.FetchDVPorts(port_criteria)
                dvs_port_connection.portgroupKey = ports[0].portgroupKey
            backing_info = vim.vm.device.VirtualEthernetCard.DistributedVirtualPortBackingInfo()
            backing_info.port = dvs_port_connection

        if network_obj:
            backing_info = vim.vm.device.VirtualEthernetCard.NetworkBackingInfo()
            backing_info.network = network_obj
            backing_info.deviceName = network_obj.name

        virtual_network.backing = backing_info

        #creating network device spec
        network_spec = vim.vm.device.VirtualDeviceSpec()
        network_spec.device = virtual_network
        network_spec.operation = vim.vm.device.VirtualDeviceSpec.Operation.add

        #creating reconfig spec
        vm_reconfig_spec = vim.vm.ConfigSpec()
        vm_reconfig_spec.deviceChange = [network_spec]
        return vm_reconfig_spec

    def run(self, vms, distributed_switch=None, mac_address=None, network_name=None, port_key=None,
            stay_connected=False, network_type='Flexible', wake_on_lan=False):
        network_type = network_type.lower() if network_type else None
        si = self.si
        si_content = si.RetrieveContent()

        vm_objs = [vim.VirtualMachine(moid, stub=si._stub) for moid in vms]
        vm_names = [vm_obj.name for vm_obj in vm_objs] # by checking the name property, the vms' existance is checked.

        distributed_switch_obj = None
        if distributed_switch:
            distributed_switch_obj = vim.DistributedVirtualSwitch(distributed_switch, stub=si._stub)
            distributed_switch_obj.name# by checking the name property, the distributed switch existence is checked.

        network_obj = None
        if network_name:
            network_obj = inventory.get_network(si_content, name=network_name)
            network_obj.name # by checking the name property, the network existence is checked.

        result=[]
        for vm in vm_objs:
            vm_reconfig_spec = NewNetworkAdapter.get_vm_reconfig_spec(distributed_switch_obj, mac_address, network_obj, port_key, stay_connected, network_type, wake_on_lan)
            add_disk_task = vm.ReconfigVM_Task(spec=vm_reconfig_spec)
            successfully_added_network = self._wait_for_task(add_disk_task)
            result.append({
                "vm_moid":vm._GetMoId(),
                "success":successfully_added_network
            })

        return result