diff options
author | Jan Gelety <jgelety@cisco.com> | 2019-11-12 05:27:43 +0100 |
---|---|---|
committer | Jan Gelety <jgelety@cisco.com> | 2019-11-28 18:26:21 +0100 |
commit | d68951ac245150eeefa6e0f4156e4c1b5c9e9325 (patch) | |
tree | 487554a7547218d27f0a61ec02b70502c32cdcb4 /resources/libraries/python/LispSetup.py | |
parent | ed0258a440cfad7023d643f717ab78ac568dc59b (diff) |
Python3: resources and libraries
Change-Id: I1392c06b1d64f62b141d24c0d42a8e36913b15e2
Signed-off-by: Jan Gelety <jgelety@cisco.com>
Diffstat (limited to 'resources/libraries/python/LispSetup.py')
-rw-r--r-- | resources/libraries/python/LispSetup.py | 510 |
1 files changed, 234 insertions, 276 deletions
diff --git a/resources/libraries/python/LispSetup.py b/resources/libraries/python/LispSetup.py index 5fee5db87d..666b6e636d 100644 --- a/resources/libraries/python/LispSetup.py +++ b/resources/libraries/python/LispSetup.py @@ -15,15 +15,13 @@ from ipaddress import ip_address -from resources.libraries.python.topology import NodeType -from resources.libraries.python.PapiExecutor import PapiSocketExecutor from resources.libraries.python.L2Util import L2Util +from resources.libraries.python.PapiExecutor import PapiSocketExecutor +from resources.libraries.python.topology import NodeType -class LispStatus(object): - """Class for lisp API.""" - def __init__(self): - pass +class LispStatus: + """Class for lisp API.""" @staticmethod def vpp_lisp_enable_disable(node, state): @@ -34,25 +32,21 @@ class LispStatus(object): :type node: dict :type state: str """ + args = dict(is_en=0 if state == u"disable" else 1) - args = dict(is_en=0 if state == 'disable' else 1) - - cmd = 'lisp_enable_disable' - err_msg = "Failed to set LISP status on host {host}".format( - host=node['host']) + cmd = u"lisp_enable_disable" + err_msg = f"Failed to set LISP status on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) -class LispRemoteMapping(object): +class LispRemoteMapping: """Class for lisp remote mapping API.""" - def __init__(self): - pass - @staticmethod - def vpp_add_lisp_remote_mapping(node, vni, deid, deid_prefix, seid, - seid_prefix, rloc, is_mac=False): + def vpp_add_lisp_remote_mapping( + node, vni, deid, deid_prefix, seid, seid_prefix, rloc, + is_mac=False): """Add lisp remote mapping on the VPP node in topology. :param node: VPP node. @@ -72,11 +66,10 @@ class LispRemoteMapping(object): :type rloc: str :type is_mac: bool """ - if not is_mac: - eid_type = 0 if ip_address(unicode(deid)).version == 4 else 1 - eid_packed = ip_address(unicode(deid)).packed - seid_packed = ip_address(unicode(seid)).packed + eid_type = 0 if ip_address(deid).version == 4 else 1 + eid_packed = ip_address(deid).packed + seid_packed = ip_address(seid).packed eid_len = deid_prefix seid_len = seid_prefix else: @@ -86,30 +79,35 @@ class LispRemoteMapping(object): eid_len = 0 seid_len = 0 - rlocs = [dict(is_ip4=1 if ip_address(unicode(rloc)).version == 4 else 0, - addr=ip_address(unicode(rloc)).packed)] - - args = dict(is_add=1, - is_src_dst=1, - vni=int(vni), - eid_type=eid_type, - eid=eid_packed, - eid_len=eid_len, - seid=seid_packed, - seid_len=seid_len, - rloc_num=1, - rlocs=rlocs) - - cmd = 'lisp_add_del_remote_mapping' - err_msg = "Failed to add remote mapping on host {host}".format( - host=node['host']) + rlocs = [ + dict( + is_ip4=1 if ip_address(rloc).version == 4 else 0, + addr=ip_address(rloc).packed + ) + ] + + args = dict( + is_add=1, + is_src_dst=1, + vni=int(vni), + eid_type=eid_type, + eid=eid_packed, + eid_len=eid_len, + seid=seid_packed, + seid_len=seid_len, + rloc_num=1, + rlocs=rlocs + ) + + cmd = u"lisp_add_del_remote_mapping" + err_msg = f"Failed to add remote mapping on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) @staticmethod - def vpp_del_lisp_remote_mapping(node, vni, deid, deid_prefix, seid, - seid_prefix, rloc): + def vpp_del_lisp_remote_mapping( + node, vni, deid, deid_prefix, seid, seid_prefix, rloc): """Delete lisp remote mapping on the VPP node in topology. :param node: VPP node. @@ -127,14 +125,13 @@ class LispRemoteMapping(object): :type seid_prefix: int :type rloc: str """ - # used only with IPs is_mac = False if not is_mac: - eid_type = 0 if ip_address(unicode(deid)).version == 4 else 1 - eid_packed = ip_address(unicode(deid)).packed - seid_packed = ip_address(unicode(seid)).packed + eid_type = 0 if ip_address(deid).version == 4 else 1 + eid_packed = ip_address(deid).packed + seid_packed = ip_address(seid).packed eid_len = deid_prefix seid_len = seid_prefix else: @@ -144,36 +141,38 @@ class LispRemoteMapping(object): eid_len = 0 seid_len = 0 - rlocs = [dict(is_ip4=1 if ip_address(unicode(rloc)).version == 4 else 0, - addr=ip_address(unicode(rloc)).packed)] - - args = dict(is_add=0, - is_src_dst=1, - vni=int(vni), - eid_type=eid_type, - eid=eid_packed, - eid_len=eid_len, - seid=seid_packed, - seid_len=seid_len, - rloc_num=1, - rlocs=rlocs) - - cmd = 'lisp_add_del_remote_mapping' - err_msg = "Failed to delete remote mapping on host {host}".format( - host=node['host']) + rlocs = [ + dict( + is_ip4=1 if ip_address(str(rloc)).version == 4 else 0, + addr=ip_address(str(rloc)).packed + ) + ] + + args = dict( + is_add=0, + is_src_dst=1, + vni=int(vni), + eid_type=eid_type, + eid=eid_packed, + eid_len=eid_len, + seid=seid_packed, + seid_len=seid_len, + rloc_num=1, + rlocs=rlocs + ) + + cmd = u"lisp_add_del_remote_mapping" + err_msg = f"Failed to delete remote mapping on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) -class LispAdjacency(object): +class LispAdjacency: """Class for lisp adjacency API.""" - def __init__(self): - pass - @staticmethod - def vpp_add_lisp_adjacency(node, vni, deid, deid_prefix, seid, - seid_prefix, is_mac=False): + def vpp_add_lisp_adjacency( + node, vni, deid, deid_prefix, seid, seid_prefix, is_mac=False): """Add lisp adjacency on the VPP node in topology. :param node: VPP node. @@ -191,11 +190,10 @@ class LispAdjacency(object): :type seid_prefix: int :type is_mac: bool """ - if not is_mac: - eid_type = 0 if ip_address(unicode(deid)).version == 4 else 1 - reid = ip_address(unicode(deid)).packed - leid = ip_address(unicode(seid)).packed + eid_type = 0 if ip_address(deid).version == 4 else 1 + reid = ip_address(deid).packed + leid = ip_address(seid).packed reid_len = deid_prefix leid_len = seid_prefix else: @@ -205,24 +203,25 @@ class LispAdjacency(object): reid_len = 0 leid_len = 0 - args = dict(is_add=1, - vni=int(vni), - eid_type=eid_type, - reid=reid, - reid_len=reid_len, - leid=leid, - leid_len=leid_len) + args = dict( + is_add=1, + vni=int(vni), + eid_type=eid_type, + reid=reid, + reid_len=reid_len, + leid=leid, + leid_len=leid_len + ) - cmd = 'lisp_add_del_adjacency' - err_msg = "Failed to add lisp adjacency on host {host}".format( - host=node['host']) + cmd = u"lisp_add_del_adjacency" + err_msg = f"Failed to add lisp adjacency on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) @staticmethod - def vpp_del_lisp_adjacency(node, vni, deid, deid_prefix, seid, - seid_prefix): + def vpp_del_lisp_adjacency( + node, vni, deid, deid_prefix, seid, seid_prefix): """Delete lisp adjacency on the VPP node in topology. :param node: VPP node. @@ -238,14 +237,13 @@ class LispAdjacency(object): :type seid: str :type seid_prefix: int """ - # used only with IPs is_mac = False if not is_mac: - eid_type = 0 if ip_address(unicode(deid)).version == 4 else 1 - reid = ip_address(unicode(deid)).packed - leid = ip_address(unicode(seid)).packed + eid_type = 0 if ip_address(deid).version == 4 else 1 + reid = ip_address(deid).packed + leid = ip_address(seid).packed reid_len = deid_prefix leid_len = seid_prefix else: @@ -255,27 +253,25 @@ class LispAdjacency(object): reid_len = 0 leid_len = 0 - args = dict(is_add=0, - vni=int(vni), - eid_type=eid_type, - reid=reid, - reid_len=reid_len, - leid=leid, - leid_len=leid_len) + args = dict( + is_add=0, + vni=int(vni), + eid_type=eid_type, + reid=reid, + reid_len=reid_len, + leid=leid, + leid_len=leid_len + ) - cmd = 'lisp_add_del_adjacency' - err_msg = "Failed to delete lisp adjacency on host {host}".format( - host=node['host']) + cmd = u"lisp_add_del_adjacency" + err_msg = f"Failed to delete lisp adjacency on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) -class LispGpeStatus(object): +class LispGpeStatus: """Clas for LISP GPE status manipulation.""" - def __init__(self): - pass - @staticmethod def vpp_lisp_gpe_enable_disable(node, state): """Change the state of LISP GPE - enable or disable. @@ -285,45 +281,35 @@ class LispGpeStatus(object): :type node: dict :type state: str """ + args = dict(is_en=0 if state == u"disable" else 1) - args = dict(is_en=0 if state == 'disable' else 1) - - cmd = 'gpe_enable_disable' - err_msg = "Failed to set LISP GPE status on host {host}".format( - host=node['host']) + cmd = u"gpe_enable_disable" + err_msg = f"Failed to set LISP GPE status on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) -class LispGpeForwardEntry(object): +class LispGpeForwardEntry: """The functionality needed for these methods is not implemented in VPP (VAT). Bug https://jira.fd.io/browse/VPP-334 was open to cover this issue. TODO: Implement when VPP-334 is fixed. """ - def __init__(self): - pass - @staticmethod def add_lisp_gpe_forward_entry(node, *args): """Not implemented""" # TODO: Implement when VPP-334 is fixed. - pass @staticmethod def del_lisp_gpe_forward_entry(node, *args): """Not implemented""" # TODO: Implement when VPP-334 is fixed. - pass -class LispMapResolver(object): +class LispMapResolver: """Class for Lisp map resolver API.""" - def __init__(self): - pass - @staticmethod def vpp_add_map_resolver(node, map_resolver_ip): """Set lisp map resolver on the VPP node in topology. @@ -333,15 +319,14 @@ class LispMapResolver(object): :type node: dict :type map_resolver_ip: str """ + args = dict( + is_add=1, + is_ipv6=0 if ip_address(map_resolver_ip).version == 4 else 1, + ip_address=ip_address(map_resolver_ip).packed + ) - args = dict(is_add=1, - is_ipv6=0 if ip_address(unicode(map_resolver_ip)).version \ - == 4 else 1, - ip_address=ip_address(unicode(map_resolver_ip)).packed) - - cmd = 'lisp_add_del_map_resolver' - err_msg = "Failed to add map resolver on host {host}".format( - host=node['host']) + cmd = u"lisp_add_del_map_resolver" + err_msg = f"Failed to add map resolver on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) @@ -355,28 +340,24 @@ class LispMapResolver(object): :type node: dict :type map_resolver_ip: str """ + args = dict( + is_add=0, + is_ipv6=0 if ip_address(map_resolver_ip).version == 4 else 1, + ip_address=ip_address(map_resolver_ip).packed + ) - args = dict(is_add=0, - is_ipv6=0 if ip_address(unicode(map_resolver_ip)).version \ - == 4 else 1, - ip_address=ip_address(unicode(map_resolver_ip)).packed) - - cmd = 'lisp_add_del_map_resolver' - err_msg = "Failed to delete map resolver on host {host}".format( - host=node['host']) + cmd = u"lisp_add_del_map_resolver" + err_msg = f"Failed to delete map resolver on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) -class LispLocalEid(object): +class LispLocalEid: """Class for Lisp local eid API.""" - def __init__(self): - pass - @staticmethod - def vpp_add_lisp_local_eid(node, locator_set_name, vni, eid, - prefix_len=None): + def vpp_add_lisp_local_eid( + node, locator_set_name, vni, eid, prefix_len=None): """Set lisp eid address on the VPP node in topology. :param node: VPP node. @@ -390,31 +371,31 @@ class LispLocalEid(object): :type eid: str :type prefix_len: int """ - if prefix_len: - eid_type = 0 if ip_address(unicode(eid)).version == 4 else 1 - eid_packed = ip_address(unicode(eid)).packed + eid_type = 0 if ip_address(eid).version == 4 else 1 + eid_packed = ip_address(eid).packed else: eid_type = 2 eid_packed = L2Util.mac_to_bin(eid) - args = dict(is_add=1, - eid_type=eid_type, - eid=eid_packed, - prefix_len=prefix_len, - locator_set_name=locator_set_name, - vni=int(vni)) + args = dict( + is_add=1, + eid_type=eid_type, + eid=eid_packed, + prefix_len=prefix_len, + locator_set_name=locator_set_name, + vni=int(vni) + ) - cmd = 'lisp_add_del_local_eid' - err_msg = "Failed to add local eid on host {host}".format( - host=node['host']) + cmd = u"lisp_add_del_local_eid" + err_msg = f"Failed to add local eid on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) @staticmethod - def vpp_del_lisp_local_eid(node, locator_set_name, vni, eid, - prefix_len=None): + def vpp_del_lisp_local_eid( + node, locator_set_name, vni, eid, prefix_len=None): """Set lisp eid addres on the VPP node in topology. :param node: VPP node. @@ -428,34 +409,31 @@ class LispLocalEid(object): :type eid: str :type prefix_len: int """ - if prefix_len: - eid_type = 0 if ip_address(unicode(eid)).version == 4 else 1 - eid_packed = ip_address(unicode(eid)).packed + eid_type = 0 if ip_address(eid).version == 4 else 1 + eid_packed = ip_address(eid).packed else: eid_type = 2 eid_packed = L2Util.mac_to_bin(eid) - args = dict(is_add=0, - eid_type=eid_type, - eid=eid_packed, - prefix_len=prefix_len, - locator_set_name=locator_set_name, - vni=int(vni)) + args = dict( + is_add=0, + eid_type=eid_type, + eid=eid_packed, + prefix_len=prefix_len, + locator_set_name=locator_set_name, + vni=int(vni) + ) - cmd = 'lisp_add_del_local_eid' - err_msg = "Failed to delete local eid on host {host}".format( - host=node['host']) + cmd = u"lisp_add_del_local_eid" + err_msg = f"Failed to delete local eid on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) -class LispLocator(object): +class LispLocator: """Class for the Lisp Locator API.""" - def __init__(self): - pass - @staticmethod def vpp_add_lisp_locator(node, locator_name, sw_if_index, priority, weight): """Set lisp locator on the VPP node in topology. @@ -472,15 +450,16 @@ class LispLocator(object): :type weight: int """ - args = dict(is_add=1, - locator_set_name=locator_name, - sw_if_index=sw_if_index, - priority=priority, - weight=weight) + args = dict( + is_add=1, + locator_set_name=locator_name, + sw_if_index=sw_if_index, + priority=priority, + weight=weight + ) - cmd = 'lisp_add_del_locator' - err_msg = "Failed to add locator on host {host}".format( - host=node['host']) + cmd = u"lisp_add_del_locator" + err_msg = f"Failed to add locator on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) @@ -500,26 +479,23 @@ class LispLocator(object): :type priority: int :type weight: int """ + args = dict( + is_add=0, + locator_set_name=locator_name, + sw_if_index=sw_if_index, + priority=priority, + weight=weight + ) - args = dict(is_add=0, - locator_set_name=locator_name, - sw_if_index=sw_if_index, - priority=priority, - weight=weight) - - cmd = 'lisp_add_del_locator' - err_msg = "Failed to delete locator on host {host}".format( - host=node['host']) + cmd = u"lisp_add_del_locator" + err_msg = f"Failed to delete locator on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) -class LispLocatorSet(object): +class LispLocatorSet: """Class for Lisp Locator Set API.""" - def __init__(self): - pass - @staticmethod def vpp_add_lisp_locator_set(node, name): """Add lisp locator_set on VPP. @@ -529,15 +505,15 @@ class LispLocatorSet(object): :type node: dict :type name: str """ + args = dict( + is_add=1, + locator_set_name=name, + locator_num=0, + locators=[] + ) - args = dict(is_add=1, - locator_set_name=name, - locator_num=0, - locators=[]) - - cmd = 'lisp_add_del_locator_set' - err_msg = "Failed to add locator set on host {host}".format( - host=node['host']) + cmd = u"lisp_add_del_locator_set" + err_msg = f"Failed to add locator set on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) @@ -551,25 +527,22 @@ class LispLocatorSet(object): :type node: dict :type name: str """ + args = dict( + is_add=0, + locator_set_name=name, + locator_num=0, + locators=[] + ) - args = dict(is_add=0, - locator_set_name=name, - locator_num=0, - locators=[]) - - cmd = 'lisp_add_del_locator_set' - err_msg = "Failed to delete locator set on host {host}".format( - host=node['host']) + cmd = u"lisp_add_del_locator_set" + err_msg = f"Failed to delete locator set on host {node[u'host']}" with PapiSocketExecutor(node) as papi_exec: papi_exec.add(cmd, **args).get_reply(err_msg) -class LispSetup(object): +class LispSetup: """Lisp setup in topology.""" - def __init__(self): - pass - @staticmethod def vpp_set_lisp_locator_set(node, locator_set_list): """Set lisp locator_sets on VPP node in topology. @@ -580,25 +553,22 @@ class LispSetup(object): :type locator_set_list: list """ - if node['type'] != NodeType.DUT: - raise ValueError('Node is not DUT') + if node[u"type"] != NodeType.DUT: + raise ValueError(u"Node is not DUT") lisp_locator = LispLocator() lisp_locator_set = LispLocatorSet() for locator_set in locator_set_list: - locator_set_name = locator_set.get('locator-set') - locator_list = locator_set.get('locator') - lisp_locator_set.vpp_add_lisp_locator_set(node, - locator_set_name) + locator_set_name = locator_set.get(u"locator-set") + locator_list = locator_set.get(u"locator") + lisp_locator_set.vpp_add_lisp_locator_set(node, locator_set_name) for locator in locator_list: - sw_if_index = locator.get('locator-index') - priority = locator.get('priority') - weight = locator.get('weight') - lisp_locator.vpp_add_lisp_locator(node, - locator_set_name, - sw_if_index, - priority, - weight) + sw_if_index = locator.get(u"locator-index") + priority = locator.get(u"priority") + weight = locator.get(u"weight") + lisp_locator.vpp_add_lisp_locator( + node, locator_set_name, sw_if_index, priority, weight + ) @staticmethod def vpp_unset_lisp_locator_set(node, locator_set_list): @@ -609,27 +579,23 @@ class LispSetup(object): :type node: dict :type locator_set_list: list """ - - if node['type'] != NodeType.DUT: - raise ValueError('Lisp locator set, node is not DUT') + if node[u"type"] != NodeType.DUT: + raise ValueError(u"Lisp locator set, node is not DUT") lisp_locator = LispLocator() lisp_locator_set = LispLocatorSet() for locator_set in locator_set_list: - locator_set_name = locator_set.get('locator-set') - locator_list = locator_set.get('locator') + locator_set_name = locator_set.get(u"locator-set") + locator_list = locator_set.get(u"locator") for locator in locator_list: - sw_if_index = locator.get('locator-index') - priority = locator.get('priority') - weight = locator.get('weight') - lisp_locator.vpp_del_lisp_locator(node, - locator_set_name, - sw_if_index, - priority, - weight) - - lisp_locator_set.vpp_del_lisp_locator_set(node, - locator_set_name) + sw_if_index = locator.get(u"locator-index") + priority = locator.get(u"priority") + weight = locator.get(u"weight") + lisp_locator.vpp_del_lisp_locator( + node, locator_set_name, sw_if_index, priority, weight + ) + + lisp_locator_set.vpp_del_lisp_locator_set(node, locator_set_name) @staticmethod def vpp_set_lisp_eid_table(node, eid_table): @@ -640,23 +606,20 @@ class LispSetup(object): :type node: dict :type eid_table: dict """ - - if node['type'] != NodeType.DUT: - raise ValueError('Node is not DUT') + if node[u"type"] != NodeType.DUT: + raise ValueError(u"Node is not DUT") lisp_locator_set = LispLocatorSet() lisp_eid = LispLocalEid() for eid in eid_table: - vni = eid.get('vni') - eid_address = eid.get('eid') - eid_prefix_len = eid.get('eid-prefix-len') - locator_set_name = eid.get('locator-set') + vni = eid.get(u"vni") + eid_address = eid.get(u"eid") + eid_prefix_len = eid.get(u"eid-prefix-len") + locator_set_name = eid.get(u"locator-set") lisp_locator_set.vpp_add_lisp_locator_set(node, locator_set_name) - lisp_eid.vpp_add_lisp_local_eid(node, - locator_set_name, - vni, - eid_address, - eid_prefix_len) + lisp_eid.vpp_add_lisp_local_eid( + node, locator_set_name, vni, eid_address, eid_prefix_len + ) @staticmethod def vpp_unset_lisp_eid_table(node, eid_table): @@ -667,26 +630,23 @@ class LispSetup(object): :type node: dict :type eid_table: dict """ - - if node['type'] != NodeType.DUT: - raise ValueError('Node is not DUT') + if node[u"type"] != NodeType.DUT: + raise ValueError(u"Node is not DUT") locator_set_list = [] lisp_locator_set = LispLocatorSet() lisp_eid = LispLocalEid() for eid in eid_table: - vni = eid.get('vni') - eid_address = eid.get('eid') - eid_prefix_len = eid.get('eid-prefix-len') - locator_set_name = eid.get('locator-set') + vni = eid.get(u"vni") + eid_address = eid.get(u"eid") + eid_prefix_len = eid.get(u"eid-prefix-len") + locator_set_name = eid.get(u"locator-set") if locator_set_name not in locator_set_list: locator_set_list.append(locator_set_name) - lisp_eid.vpp_del_lisp_local_eid(node, - locator_set_name, - vni, - eid_address, - eid_prefix_len) + lisp_eid.vpp_del_lisp_local_eid( + node, locator_set_name, vni, eid_address, eid_prefix_len + ) for locator_set_name in locator_set_list: lisp_locator_set.vpp_del_lisp_locator_set(node, locator_set_name) @@ -700,10 +660,9 @@ class LispSetup(object): :type node: dict :type map_resolver: dict """ - lisp_map_res = LispMapResolver() for map_ip in map_resolver: - lisp_map_res.vpp_add_map_resolver(node, map_ip.get('map resolver')) + lisp_map_res.vpp_add_map_resolver(node, map_ip.get(u"map resolver")) @staticmethod def vpp_unset_lisp_map_resolver(node, map_resolver): @@ -714,12 +673,11 @@ class LispSetup(object): :type node: dict :type map_resolver: dict """ - lisp_map_res = LispMapResolver() for map_ip in map_resolver: - lisp_map_res.vpp_del_map_resolver(node, map_ip.get('map resolver')) + lisp_map_res.vpp_del_map_resolver(node, map_ip.get(u"map resolver")) -class LispEidTableMap(object): +class LispEidTableMap: """ Class for EID table map. """ @@ -738,7 +696,6 @@ class LispEidTableMap(object): :type bd_id: int :type vrf: int """ - # adding default mapping vni=0, vrf=0 needs to be skipped skip = False @@ -752,14 +709,15 @@ class LispEidTableMap(object): if (int(vrf) == 0) and (int(vni) == 0): skip = True - args = dict(is_add=1, - vni=int(vni), - dp_table=int(dp_table), - is_l2=is_l2) + args = dict( + is_add=1, + vni=int(vni), + dp_table=int(dp_table), + is_l2=is_l2 + ) - cmd = 'lisp_eid_table_add_del_map' - err_msg = "Failed to add eid table map on host {host}".format( - host=node['host']) + cmd = u"lisp_eid_table_add_del_map" + err_msg = f"Failed to add eid table map on host {node[u'host']}" if not skip: with PapiSocketExecutor(node) as papi_exec: |