Merge pull request #1626 from jcpowermac/vmware_migrate_vmk_fix_20

Resolves issue with vmware_migrate_vmk module for v2.0
This commit is contained in:
Brian Coca 2016-03-03 18:45:45 -05:00
commit f6d2763671

View file

@ -75,8 +75,6 @@ Example from Ansible playbook
migrate_switch_name: dvSwitch migrate_switch_name: dvSwitch
migrate_portgroup_name: Management migrate_portgroup_name: Management
''' '''
try: try:
from pyVmomi import vim, vmodl from pyVmomi import vim, vmodl
HAS_PYVMOMI = True HAS_PYVMOMI = True
@ -84,88 +82,93 @@ except ImportError:
HAS_PYVMOMI = False HAS_PYVMOMI = False
def state_exit_unchanged(module): class VMwareMigrateVmk(object):
module.exit_json(changed=False) def __init__(self, module):
self.module = module
self.host_system = None
self.migrate_switch_name = self.module.params['migrate_switch_name']
self.migrate_portgroup_name = self.module.params['migrate_portgroup_name']
self.device = self.module.params['device']
self.esxi_hostname = self.module.params['esxi_hostname']
self.current_portgroup_name = self.module.params['current_portgroup_name']
self.current_switch_name = self.module.params['current_switch_name']
self.content = connect_to_api(module)
def process_state(self):
try:
vmk_migration_states = {
'migrate_vss_vds': self.state_migrate_vss_vds,
'migrate_vds_vss': self.state_migrate_vds_vss,
'migrated': self.state_exit_unchanged
}
def state_migrate_vds_vss(module): vmk_migration_states[self.check_vmk_current_state()]()
module.exit_json(changed=False, msg="Currently Not Implemented")
except vmodl.RuntimeFault as runtime_fault:
self.module.fail_json(msg=runtime_fault.msg)
except vmodl.MethodFault as method_fault:
self.module.fail_json(msg=method_fault.msg)
except Exception as e:
self.module.fail_json(msg=str(e))
def create_host_vnic_config(dv_switch_uuid, portgroup_key, device): def state_exit_unchanged(self):
self.module.exit_json(changed=False)
host_vnic_config = vim.host.VirtualNic.Config() def state_migrate_vds_vss(self):
host_vnic_config.spec = vim.host.VirtualNic.Specification() self.module.exit_json(changed=False, msg="Currently Not Implemented")
host_vnic_config.changeOperation = "edit"
host_vnic_config.device = device
host_vnic_config.portgroup = ""
host_vnic_config.spec.distributedVirtualPort = vim.dvs.PortConnection()
host_vnic_config.spec.distributedVirtualPort.switchUuid = dv_switch_uuid
host_vnic_config.spec.distributedVirtualPort.portgroupKey = portgroup_key
return host_vnic_config def create_host_vnic_config(self, dv_switch_uuid, portgroup_key):
host_vnic_config = vim.host.VirtualNic.Config()
host_vnic_config.spec = vim.host.VirtualNic.Specification()
host_vnic_config.changeOperation = "edit"
host_vnic_config.device = self.device
host_vnic_config.portgroup = ""
host_vnic_config.spec.distributedVirtualPort = vim.dvs.PortConnection()
host_vnic_config.spec.distributedVirtualPort.switchUuid = dv_switch_uuid
host_vnic_config.spec.distributedVirtualPort.portgroupKey = portgroup_key
def create_port_group_config(switch_name, portgroup_name): return host_vnic_config
port_group_config = vim.host.PortGroup.Config()
port_group_config.spec = vim.host.PortGroup.Specification()
port_group_config.changeOperation = "remove" def create_port_group_config(self):
port_group_config.spec.name = portgroup_name port_group_config = vim.host.PortGroup.Config()
port_group_config.spec.vlanId = -1 port_group_config.spec = vim.host.PortGroup.Specification()
port_group_config.spec.vswitchName = switch_name
port_group_config.spec.policy = vim.host.NetworkPolicy()
return port_group_config port_group_config.changeOperation = "remove"
port_group_config.spec.name = self.current_portgroup_name
port_group_config.spec.vlanId = -1
port_group_config.spec.vswitchName = self.current_switch_name
port_group_config.spec.policy = vim.host.NetworkPolicy()
return port_group_config
def state_migrate_vss_vds(module): def state_migrate_vss_vds(self):
content = module.params['content'] host_network_system = self.host_system.configManager.networkSystem
host_system = module.params['host_system']
migrate_switch_name = module.params['migrate_switch_name']
migrate_portgroup_name = module.params['migrate_portgroup_name']
current_portgroup_name = module.params['current_portgroup_name']
current_switch_name = module.params['current_switch_name']
device = module.params['device']
host_network_system = host_system.configManager.networkSystem dv_switch = find_dvs_by_name(self.content, self.migrate_switch_name)
pg = find_dvspg_by_name(dv_switch, self.migrate_portgroup_name)
dv_switch = find_dvs_by_name(content, migrate_switch_name) config = vim.host.NetworkConfig()
pg = find_dvspg_by_name(dv_switch, migrate_portgroup_name) config.portgroup = [self.create_port_group_config()]
config.vnic = [self.create_host_vnic_config(dv_switch.uuid, pg.key)]
host_network_system.UpdateNetworkConfig(config, "modify")
self.module.exit_json(changed=True)
config = vim.host.NetworkConfig() def check_vmk_current_state(self):
config.portgroup = [create_port_group_config(current_switch_name, current_portgroup_name)] self.host_system = find_hostsystem_by_name(self.content, self.esxi_hostname)
config.vnic = [create_host_vnic_config(dv_switch.uuid, pg.key, device)]
host_network_system.UpdateNetworkConfig(config, "modify")
module.exit_json(changed=True)
for vnic in self.host_system.configManager.networkSystem.networkInfo.vnic:
def check_vmk_current_state(module): if vnic.device == self.device:
#self.vnic = vnic
device = module.params['device'] if vnic.spec.distributedVirtualPort is None:
esxi_hostname = module.params['esxi_hostname'] if vnic.portgroup == self.current_portgroup_name:
current_portgroup_name = module.params['current_portgroup_name'] return "migrate_vss_vds"
current_switch_name = module.params['current_switch_name'] else:
dvs = find_dvs_by_name(self.content, self.current_switch_name)
content = connect_to_api(module) if dvs is None:
return "migrated"
host_system = find_hostsystem_by_name(content, esxi_hostname) if vnic.spec.distributedVirtualPort.switchUuid == dvs.uuid:
return "migrate_vds_vss"
module.params['content'] = content
module.params['host_system'] = host_system
for vnic in host_system.configManager.networkSystem.networkInfo.vnic:
if vnic.device == device:
module.params['vnic'] = vnic
if vnic.spec.distributedVirtualPort is None:
if vnic.portgroup == current_portgroup_name:
return "migrate_vss_vds"
else:
dvs = find_dvs_by_name(content, current_switch_name)
if dvs is None:
return "migrated"
if vnic.spec.distributedVirtualPort.switchUuid == dvs.uuid:
return "migrate_vds_vss"
def main(): def main():
@ -181,23 +184,10 @@ def main():
module = AnsibleModule(argument_spec=argument_spec, supports_check_mode=False) module = AnsibleModule(argument_spec=argument_spec, supports_check_mode=False)
if not HAS_PYVMOMI: if not HAS_PYVMOMI:
module.fail_json(msg='pyvmomi required for this module') self.module.fail_json(msg='pyvmomi required for this module')
try: vmware_migrate_vmk = VMwareMigrateVmk(module)
vmk_migration_states = { vmware_migrate_vmk.process_state()
'migrate_vss_vds': state_migrate_vss_vds,
'migrate_vds_vss': state_migrate_vds_vss,
'migrated': state_exit_unchanged
}
vmk_migration_states[check_vmk_current_state(module)](module)
except vmodl.RuntimeFault as runtime_fault:
module.fail_json(msg=runtime_fault.msg)
except vmodl.MethodFault as method_fault:
module.fail_json(msg=method_fault.msg)
except Exception as e:
module.fail_json(msg=str(e))
from ansible.module_utils.vmware import * from ansible.module_utils.vmware import *
from ansible.module_utils.basic import * from ansible.module_utils.basic import *