#!/usr/bin/python # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. # This script reconfigure UCSM vnics for varios OPNFV deployers # Usage: reconfigUcsNet.py [options] # # Options: # -h, --help show this help message and exit # -i IP, --ip=IP [Mandatory] UCSM IP Address # -u USERNAME, --username=USERNAME # [Mandatory] Account Username for UCSM Login # -p PASSWORD, --password=PASSWORD # [Mandatory] Account Password for UCSM Login # -f FILE, --file=FILE # [Optional] Yaml file with network config you # want to set for POD # If not present only current network config # will be printed # import getpass import optparse import platform import yaml import time import sys from UcsSdk import LsmaintAck, LsPower, LsServer, OrgOrg from UcsSdk import UcsHandle, VnicEther, VnicEtherIf, YesOrNo POD_PREFIX = "POD-2" INSTALLER = "POD-21" def getpassword(prompt): if platform.system() == "Linux": return getpass.unix_getpass(prompt=prompt) elif platform.system() == "Windows" or platform.system() == "Microsoft": return getpass.win_getpass(prompt=prompt) else: return getpass.getpass(prompt=prompt) def get_servers(handle=None): """ Return list of servers """ orgObj = handle.GetManagedObject( None, OrgOrg.ClassId(), {OrgOrg.DN: "org-root"})[0] servers = handle.GetManagedObject(orgObj, LsServer.ClassId()) for server in servers: if server.Type == 'instance' and POD_PREFIX in server.Dn: yield server def set_boot_policy(handle=None, server=None, policy=None): """ Modify Boot policy of server """ obj = handle.GetManagedObject(None, LsServer.ClassId(), { LsServer.DN: server.Dn}) handle.SetManagedObject(obj, LsServer.ClassId(), { LsServer.BOOT_POLICY_NAME: policy}) print(" Configured boot policy: {}".format(policy)) def ack_pending(handle=None, server=None): """ Acknowledge pending state of server """ handle.AddManagedObject(server, LsmaintAck.ClassId(), { LsmaintAck.DN: server.Dn + "/ack", LsmaintAck.DESCR: "", LsmaintAck.ADMIN_STATE: "trigger-immediate", LsmaintAck.SCHEDULER: "", LsmaintAck.POLICY_OWNER: "local"}, True) print(" Pending-reboot -> Acknowledged.") def boot_server(handle=None, server=None): """ Boot server (when is in power-off state) """ obj = handle.GetManagedObject( None, LsServer.ClassId(), {LsServer.DN: server.Dn}) handle.AddManagedObject(obj, LsPower.ClassId(), { LsPower.DN: server.Dn + "/power", LsPower.STATE: "admin-up"}, True) print(" Booting.") def get_vnics(handle=None, server=None): """ Return list of vnics for given server """ vnics = handle.ConfigResolveChildren( VnicEther.ClassId(), server.Dn, None, YesOrNo.TRUE) return vnics.OutConfigs.GetChild() def get_network_config(handle=None): """ Print current network config """ print("\nCURRENT NETWORK CONFIG:") print(" d - default, t - tagged") for server in get_servers(handle): print(' {}'.format(server.Name)) print(' Boot policy: {}'.format(server.OperBootPolicyName)) for vnic in get_vnics(handle, server): print(' {}'.format(vnic.Name)) print(' {}'.format(vnic.Addr)) vnicIfs = handle.ConfigResolveChildren( VnicEtherIf.ClassId(), vnic.Dn, None, YesOrNo.TRUE) for vnicIf in vnicIfs.OutConfigs.GetChild(): if vnicIf.DefaultNet == 'yes': print(' Vlan: {}d'.format(vnicIf.Vnet)) else: print(' Vlan: {}t'.format(vnicIf.Vnet)) def add_interface(handle=None, lsServerDn=None, vnicEther=None, templName=None, order=None, macAddr=None): """ Add interface to server specified by server.DN name """ print(" Adding interface: {}, template: {}, server.Dn: {}".format( vnicEther, templName, lsServerDn)) obj = handle.GetManagedObject( None, LsServer.ClassId(), {LsServer.DN: lsServerDn}) vnicEtherDn = lsServerDn + "/ether-" + vnicEther params = { VnicEther.STATS_POLICY_NAME: "default", VnicEther.NAME: vnicEther, VnicEther.DN: vnicEtherDn, VnicEther.SWITCH_ID: "A-B", VnicEther.ORDER: order, "adminHostPort": "ANY", VnicEther.ADMIN_VCON: "any", VnicEther.ADDR: macAddr, VnicEther.NW_TEMPL_NAME: templName, VnicEther.MTU: "1500"} handle.AddManagedObject(obj, VnicEther.ClassId(), params, True) def remove_interface(handle=None, vnicEtherDn=None): """ Remove interface specified by Distinguished Name (vnicEtherDn) """ print(" Removing interface: {}".format(vnicEtherDn)) obj = handle.GetManagedObject( None, VnicEther.ClassId(), {VnicEther.DN: vnicEtherDn}) handle.RemoveManagedObject(obj) def read_yaml_file(yamlFile): """ Read vnic config from yaml file """ # TODO: add check if vnic templates specified in file exist on UCS with open(yamlFile, 'r') as stream: return yaml.safe_load(stream) def set_network(handle=None, yamlFile=None): """ Configure VLANs on POD according specified network """ # add interfaces and bind them with vNIC templates print("\nRECONFIGURING VNICs...") pod_data = read_yaml_file(yamlFile) network = pod_data['network'] for index, server in enumerate(get_servers(handle)): # Assign template to interface for iface, data in network.iteritems(): add_interface(handle, server.Dn, iface, data['template'], data[ 'order'], data['mac-list'][index]) # Remove other interfaces which have not assigned required vnic # template vnics = get_vnics(handle, server) for vnic in vnics: if not any(data['template'] in vnic.OperNwTemplName for iface, data in network.iteritems()): remove_interface(handle, vnic.Dn) print(" {} removed, template: {}".format( vnic.Name, vnic.OperNwTemplName)) # Set boot policy template if INSTALLER not in server.Dn: set_boot_policy(handle, server, pod_data['boot-policy']) if __name__ == "__main__": print("\n*** SKIPING RECONFIGURATION.***\n") sys.exit(0) # Latest urllib2 validate certs by default # The process wide "revert to the old behaviour" hook is to monkeypatch # the ssl module # https://bugs.python.org/issue22417 import ssl if hasattr(ssl, '_create_unverified_context'): ssl._create_default_https_context = ssl._create_unverified_context try: handle = UcsHandle() parser = optparse.OptionParser() parser.add_option('-i', '--ip', dest="ip", help="[Mandatory] UCSM IP Address") parser.add_option('-u', '--username', dest="userName", help="[Mandatory] Account Username for UCSM Login") parser.add_option('-p', '--password', dest="password", help="[Mandatory] Account Password for UCSM Login") parser.add_option('-f', '--file', dest="yamlFile", help=("[Optional] Yaml file contains network " "config you want to set on UCS POD1")) (options, args) = parser.parse_args() if not options.ip: parser.print_help() parser.error("Provide UCSM IP Address") if not options.userName: parser.print_help() parser.error("Provide UCSM UserName") if not options.password: options.password = getpassword("UCSM Password:") handle.Login(options.ip, options.userName, options.password) # Change vnic template if specified in cli option if (options.yamlFile is not None): set_network(handle, options.yamlFile) time.sleep(5) print("\nWait until Overall Status of all nodes is OK...") timeout = time.time() + 60 * 10 # 10 minutes timeout while True: list_of_states = [] for server in get_servers(handle): if server.OperState == "power-off": boot_server(handle, server) if server.OperState == "pending-reboot": ack_pending(handle, server) list_of_states.append(server.OperState) print(" {}, {} seconds remains.".format( list_of_states, round(timeout - time.time()))) if all(state == "ok" for state in list_of_states): break if time.time() > timeout: raise Exception("Timeout reached while waiting for OK status.") time.sleep(10) # Show current vnic MACs and VLANs get_network_config(handle) handle.Logout() except Exception as err: handle.Logout() print("Exception:", str(err)) import traceback import sys print('-' * 60) traceback.print_exc(file=sys.stdout) print('-' * 60) sys.exit(1)