def __eq__(self, others):
if not isinstance(others, ListVRF):
raise NotImplementedError
for vrf in self.vrf_lst:
if vrf not in others.vrf_lst:
if verbose_mode(
user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL2
):
printline()
print(f"{H} The following VRF is not in the list \n {vrf}")
print(f"{H} List: \n {others.vrf_lst}")
return False
for vrf in others.vrf_lst:
if vrf not in self.vrf_lst:
if verbose_mode(
user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL2
):
printline()
print(f"{H} The following VRF is not in the list \n {vrf}")
print(f"{H} List: \n {others.vrf_lst}")
return False
return True
def _juniper_bgp_api_converter(hostname: str,
cmd_output: dict,
options={}) -> BGP:
bgp_sessions_vrf_lst = ListBGPSessionsVRF(list())
for k, v in cmd_output.items():
if not isinstance(v.get('bgp'), dict):
v['bgp'] = format_xml_output(v.get('bgp'))
if not isinstance(v.get('rid'), dict):
v['rid'] = format_xml_output(v.get('rid'))
if verbose_mode(user_value=os.environ.get('NETESTS_VERBOSE', NOT_SET),
needed_value=LEVEL3):
printline()
PP.pprint(v)
bgp_sessions_lst = ListBGPSessions(list())
if ('bgp' in v.keys() and 'bgp-information' in v.get('bgp').keys()
and v.get('bgp').get('bgp-information') is not None
and 'bgp-peer' in v.get('bgp').get('bgp-information').keys()):
for n in v.get('bgp').get('bgp-information').get('bgp-peer'):
bgp_sessions_lst.bgp_sessions.append(
BGPSession(src_hostname=hostname,
peer_ip=n.get('peer-address', NOT_SET),
peer_hostname=NOT_SET,
remote_as=n.get('peer-as', NOT_SET),
state_brief=get_bgp_state_brief(
n.get('peer-state', NOT_SET)),
session_state=n.get('peer-state', NOT_SET),
state_time=NOT_SET,
prefix_received=NOT_SET,
options=options))
as_number = n.get('bgp-option-information').get('local-as')
router_id = NOT_SET
if ('rid' in v.keys()
and 'instance-information' in v.get('rid').keys()
and 'instance-core'
in v.get('rid').get('instance-information').keys()):
router_id = v.get('rid') \
.get('instance-information') \
.get('instance-core') \
.get('router-id')
bgp_sessions_vrf_lst.bgp_sessions_vrf.append(
BGPSessionsVRF(vrf_name=k,
as_number=as_number,
router_id=router_id,
bgp_sessions=bgp_sessions_lst))
bgp = BGP(hostname=hostname, bgp_sessions_vrf_lst=bgp_sessions_vrf_lst)
if verbose_mode(user_value=os.environ.get('NETESTS_VERBOSE', NOT_SET),
needed_value=LEVEL1):
printline()
print(f'>>>>> {hostname}')
PP.pprint(bgp.to_json())
return bgp
def _iosxr_facts_netconf_converter(hostname: str,
cmd_output,
options={}) -> Facts:
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL5):
printline()
print(type(cmd_output))
print(cmd_output)
hostname = NOT_SET
domain = NOT_SET
version = NOT_SET
serial = NOT_SET
model = NOT_SET
interfaces_lst = list()
facts = Facts(hostname=hostname,
domain=domain,
version=version,
build=NOT_SET,
serial=serial,
base_mac=NOT_SET,
memory=NOT_SET,
vendor="Cisco",
model=model,
interfaces_lst=interfaces_lst,
options=options)
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1):
printline()
PP.pprint(facts.to_json())
return facts
def _napalm_facts_converter(hostname: str,
platform: str,
cmd_output: json,
options={}) -> Facts:
if cmd_output is None:
return Facts()
i = len(f"{str(cmd_output.get('facts').get('hostname'))}.")
facts = Facts(
hostname=cmd_output.get('facts').get('hostname')
if cmd_output.get("facts").get("hostname", NOT_SET) != '' else NOT_SET,
domain=cmd_output.get("facts").get("fqdn", NOT_SET)[i:]
if cmd_output.get("facts").get("fqdn", NOT_SET) != '' else NOT_SET,
version=cmd_output.get("facts").get("os_version", NOT_SET) if
cmd_output.get("facts").get("os_version", NOT_SET) != '' else NOT_SET,
serial=cmd_output.get("facts").get("serial_number", NOT_SET)
if cmd_output.get("facts").get("serial_number", NOT_SET) != '' else
NOT_SET,
base_mac=NOT_SET,
memory=NOT_SET,
vendor=cmd_output.get("facts").get("vendor", NOT_SET)
if cmd_output.get("facts").get("vendor", NOT_SET) != '' else NOT_SET,
model=cmd_output.get("facts").get("model", NOT_SET)
if cmd_output.get("facts").get("model", NOT_SET) != '' else NOT_SET,
interfaces_lst=cmd_output.get('facts').get('interface_list') if
cmd_output.get("facts").get("interface_list", NOT_SET) != '' else [],
options=options)
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1):
printline()
PP.pprint(facts.to_json())
return facts
def _juniper_facts_netconf_converter(hostname: str(), cmd_output,
options={}) -> Facts:
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL5):
printline()
PP.pprint(cmd_output.get(FACTS_SYS_DICT_KEY))
print(ElementTree.tostring(cmd_output.get(FACTS_INT_DICT_KEY)))
interfaces_lst = list()
for i in format_xml_output(cmd_output.get(FACTS_INT_DICT_KEY)).get(
'interface-information').get('physical-interface'):
interfaces_lst.append(i.get('name'))
facts = Facts(
hostname=cmd_output.get(FACTS_SYS_DICT_KEY).get('hostname', NOT_SET),
domain=cmd_output.get(FACTS_SYS_DICT_KEY).get('domain', NOT_SET),
build=NOT_SET,
version=cmd_output.get(FACTS_SYS_DICT_KEY).get('version', NOT_SET),
serial=cmd_output.get(FACTS_SYS_DICT_KEY).get('serialnumber', NOT_SET),
base_mac=NOT_SET,
memory=NOT_SET,
vendor="Juniper",
model=cmd_output.get(FACTS_SYS_DICT_KEY).get('model', NOT_SET).upper(),
interfaces_lst=interfaces_lst,
options=options)
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1):
printline()
PP.pprint(facts.to_json())
return facts
def _arista_get_facts_api(task, options={}):
c = pyeapi.connect(
transport=task.host.get('secure_api', 'https'),
host=task.host.hostname,
username=task.host.username,
password=task.host.password,
port=task.host.port
)
output = c.execute([
ARISTA_GET_FACTS,
ARISTA_GET_INT,
ARISTA_GET_DOMAIN
])
if verbose_mode(
user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL2
):
printline()
print(output)
task.host[FACTS_DATA_HOST_KEY] = _arista_facts_api_converter(
hostname=task.host.name,
cmd_output=output,
options=options
)
def run_base(nr: Nornir, protocol: str, parameters: dict,
init_data: bool) -> bool:
if (parameters.get('test', False) is True
or str(parameters.get('test', False)).upper() == "INFO"):
same = RUN.get(protocol).get('function')(
nr=nr,
options=parameters.get('options', {}),
)
if init_data is True:
create_truth_vars(nr=nr, protocol=protocol)
elif (protocol != "ping" and protocol != "socket"
and protocol != "bgp_all_up"):
same = RUN.get(protocol).get('compare')(nr=nr,
options=parameters.get(
'options', {}))
printline()
print(f"{HEADER} ({protocol}) defined in {PATH_TO_VERITY_FILES}"
f"{RUN.get(protocol).get('file')} work = {same} !!")
return (parameters.get('test', False) is True and same is False)
else:
return True
def _cumulus_vrf_api_converter(hostname: str(), cmd_output) -> ListVRF:
template = open(
f"{TEXTFSM_PATH}cumulus_net_show_vrf.textfsm"
)
results_template = textfsm.TextFSM(template)
parsed_results = results_template.ParseText(cmd_output.decode())
list_vrf = ListVRF(list())
if verbose_mode(
user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL3
):
printline()
print(parsed_results)
for line in parsed_results:
vrf = VRF(
vrf_name=line[0],
vrf_id=line[1]
)
list_vrf.vrf_lst.append(vrf)
if verbose_mode(
user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1
):
printline()
print(list_vrf)
return list_vrf
def _nxos_vrf_ssh_converter(
hostname: str(),
cmd_output,
options={}
) -> ListVRF:
vrf_list = ListVRF(list())
for vrf in cmd_output.get('TABLE_vrf', NOT_SET).get('ROW_vrf'):
vrf_list.vrf_lst.append(
VRF(
vrf_name=vrf.get('vrf_name', NOT_SET),
vrf_id=vrf.get('vrf_id', NOT_SET),
vrf_type=NOT_SET,
l3_vni=NOT_SET,
rd=vrf.get('rd') if vrf.get('rd') != '0:0' else NOT_SET,
rt_imp=NOT_SET,
rt_exp=NOT_SET,
imp_targ=NOT_SET,
exp_targ=NOT_SET,
options=options
)
)
if verbose_mode(
user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1
):
printline()
print(f">>>>> {hostname}")
PP.pprint(vrf_list.to_json())
return vrf_list
def _ios_facts_api_converter(hostname: str, cmd_output, options={}) -> Facts:
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL5):
printline()
print(type(cmd_output))
PP.pprint(json.loads(cmd_output))
hostname = NOT_SET
domain = NOT_SET
version = NOT_SET
model = NOT_SET
serial = NOT_SET
interfaces_lst = list()
if not isinstance(cmd_output, dict):
cmd_output = json.loads(cmd_output)
if (isinstance(cmd_output, dict)
and 'Cisco-IOS-XE-native:native' in cmd_output.keys()):
hostname = cmd_output.get('Cisco-IOS-XE-native:native') \
.get('hostname')
domain = cmd_output.get('Cisco-IOS-XE-native:native') \
.get('ip') \
.get('domain') \
.get('name')
version = cmd_output.get('Cisco-IOS-XE-native:native') \
.get('version')
serial = cmd_output.get('Cisco-IOS-XE-native:native') \
.get('license') \
.get('udi') \
.get('sn')
model = cmd_output.get('Cisco-IOS-XE-native:native') \
.get('license') \
.get('udi') \
.get('pid')
for t in cmd_output.get('Cisco-IOS-XE-native:native') \
.get('interface').keys():
for i in cmd_output.get('Cisco-IOS-XE-native:native') \
.get('interface') \
.get(t):
interfaces_lst.append(f"{t}{i.get('name')}")
facts = Facts(hostname=hostname,
domain=domain,
version=version,
build=NOT_SET,
serial=serial,
base_mac=NOT_SET,
memory=NOT_SET,
vendor="Cisco",
model=model,
interfaces_lst=interfaces_lst,
options=options)
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1):
printline()
PP.pprint(facts.to_json())
return facts
def _extreme_vsp_bgp_ssh_converter(hostname: str,
cmd_output,
options={}) -> BGP:
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL3):
printline()
PP.pprint(cmd_output)
bgp_sessions_vrf_lst = ListBGPSessionsVRF(list())
if cmd_output is not None:
for vrf in cmd_output:
cmd_output[vrf] = parse_textfsm(
content=str(cmd_output[vrf]),
template_file="extreme_vsp_show_ip_bgp_summary.textfsm")
bgp_sessions_lst = ListBGPSessions(bgp_sessions=list())
asn = NOT_SET
router_id = NOT_SET
for v in cmd_output.get(vrf):
if v[3] != '':
asn = v[1]
router_id = v[2]
bgp_sessions_lst.bgp_sessions.append(
BGPSession(
src_hostname=hostname,
peer_ip=v[3],
peer_hostname=NOT_SET,
remote_as=v[4],
state_brief=get_bgp_state_brief(v[5]),
session_state=v[5],
state_time=_extreme_vsp_peer_uptime_converter(
day=v[13],
hour=v[14],
min=v[15],
sec=v[16],
),
prefix_received=NOT_SET,
))
bgp_sessions_vrf_lst.bgp_sessions_vrf.append(
BGPSessionsVRF(
vrf_name=vrf,
as_number=asn,
router_id=router_id,
bgp_sessions=bgp_sessions_lst,
))
bgp = BGP(hostname=hostname, bgp_sessions_vrf_lst=bgp_sessions_vrf_lst)
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1):
printline()
print(f">>>>> {hostname}")
PP.pprint(bgp.to_json())
return bgp
def _cumulus_bgp_api_converter(hostname: str, cmd_output, options={}) -> BGP:
bgp_sessions_vrf_lst = ListBGPSessionsVRF(list())
for k, v in cmd_output.items():
if not isinstance(v, dict):
v = json.loads(v)
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL3):
printline()
PP.pprint(v)
peer = False
if ('ipv4 unicast' in v.keys()
and 'ipv4Unicast' in v.get('ipv4 unicast').keys() and 'peers'
in v.get('ipv4 unicast').get('ipv4Unicast').keys()):
sub_dict = v.get('ipv4 unicast').get('ipv4Unicast')
peer = True
elif ('ipv4 unicast' in v.keys()
and 'peers' in v.get('ipv4 unicast').keys()):
sub_dict = v.get('ipv4 unicast')
peer = True
if peer:
bgp_vrf = BGPSessionsVRF(
vrf_name=k,
as_number=sub_dict.get('as', NOT_SET),
router_id=sub_dict.get('routerId', NOT_SET),
bgp_sessions=ListBGPSessions(bgp_sessions=list()))
for i, p in sub_dict.get('peers').items():
bgp_vrf.bgp_sessions.bgp_sessions.append(
BGPSession(src_hostname=hostname,
peer_ip=i,
peer_hostname=p.get('hostname', NOT_SET),
remote_as=p.get('remoteAs', NOT_SET),
state_brief=get_bgp_state_brief(
p.get('state', NOT_SET)),
session_state=p.get('state', NOT_SET),
state_time=get_bgp_peer_uptime(
value=p.get('peerUptimeMsec', NOT_SET),
format=BGP_UPTIME_FORMAT_MS),
prefix_received=p.get('prefixReceivedCount',
NOT_SET),
options=options))
bgp_sessions_vrf_lst.bgp_sessions_vrf.append(bgp_vrf)
bgp = BGP(hostname=hostname, bgp_sessions_vrf_lst=bgp_sessions_vrf_lst)
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1):
printline()
print(f">>>>> {hostname}")
PP.pprint(bgp.to_json())
return bgp
def _iosxr_bgp_ssh_converter(hostname: str,
cmd_output: dict,
options={}) -> BGP:
bgp_sessions_vrf_lst = ListBGPSessionsVRF(list())
for k, v in cmd_output.items():
v['peers'] = parse_textfsm(
content=v.get('peers'),
template_file='cisco_xr_show_bgp_neighbors.textfsm')
v['rid'] = parse_textfsm(content=v.get('rid'),
template_file='cisco_xr_show_bgp.textfsm')
if verbose_mode(user_value=os.environ.get('NETESTS_VERBOSE', NOT_SET),
needed_value=LEVEL3):
printline()
print(v)
bgp_sessions_lst = ListBGPSessions(list())
asn = NOT_SET
if len(v['peers']) > 0:
for i in v['peers']:
asn = i[3]
bgp_sessions_lst.bgp_sessions.append(
BGPSession(
src_hostname=hostname,
peer_ip=i[0],
peer_hostname=NOT_SET,
remote_as=i[2],
state_brief=get_bgp_state_brief(i[4]),
session_state=i[4],
state_time=NOT_SET,
prefix_received=NOT_SET,
))
if len(v['rid']) > 0:
rid = v['rid'][0][0]
else:
rid = NOT_SET
bgp_sessions_vrf_lst.bgp_sessions_vrf.append(
BGPSessionsVRF(
vrf_name=k,
as_number=asn,
router_id=rid,
bgp_sessions=bgp_sessions_lst,
))
bgp = BGP(hostname=hostname, bgp_sessions_vrf_lst=bgp_sessions_vrf_lst)
if verbose_mode(user_value=os.environ.get('NETESTS_VERBOSE', NOT_SET),
needed_value=LEVEL1):
printline()
print(f'>>>>> {hostname}')
PP.pprint(bgp.to_json())
return bgp
def _ios_facts_ssh_converter(
hostname: str,
cmd_output,
options={}
) -> Facts:
interfaces_lst = list()
if FACTS_INT_DICT_KEY in cmd_output.keys():
cmd_output[FACTS_INT_DICT_KEY] = parse_textfsm(
content=cmd_output.get(FACTS_INT_DICT_KEY),
template_file='cisco_ios_show_ip_int_brief.textfsm'
)
for i in cmd_output.get(FACTS_INT_DICT_KEY):
interfaces_lst.append(i[0])
hostname = NOT_SET
version = NOT_SET
model = NOT_SET
serial = NOT_SET
build = NOT_SET
memory = NOT_SET
if FACTS_SYS_DICT_KEY in cmd_output.keys():
cmd_output[FACTS_SYS_DICT_KEY] = parse_textfsm(
content=cmd_output.get(FACTS_SYS_DICT_KEY),
template_file='cisco_ios_show_version.textfsm'
)
for i in cmd_output.get(FACTS_SYS_DICT_KEY):
hostname = i[2] if i[2] != "" else NOT_SET
version = i[0] if i[0] != "" else NOT_SET
model = i[10] if i[10] != "" else NOT_SET
serial = i[7][0] if i[7][0] != "" else NOT_SET
build = i[11] if i[11] != "" else NOT_SET
memory = i[12] if i[12] != "" else NOT_SET
facts = Facts(
hostname=hostname,
domain=NOT_SET,
version=version,
build=build,
serial=serial,
base_mac=NOT_SET,
memory=memory,
vendor="Cisco",
model=model,
interfaces_lst=interfaces_lst,
options=options
)
if verbose_mode(
user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1
):
printline()
PP.pprint(facts.to_json())
return facts
def _juniper_vrf_netconf_converter(hostname: str(), cmd_output,
options={}) -> ListVRF:
if isinstance(cmd_output, lxml.etree._Element):
cmd_output = json.dumps(
xmltodict.parse(ElementTree.tostring(cmd_output)))
elif isinstance(cmd_output, str):
cmd_output = json.dumps(xmltodict.parse(cmd_output))
cmd_output = json.loads(cmd_output)
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL4):
printline()
PP.pprint(cmd_output)
vrf_list = ListVRF(list())
for vrf in cmd_output.get('instance-information') \
.get('instance-core'):
if _juniper_vrf_filter(vrf.get('instance-name')):
rd = NOT_SET,
rt_imp = NOT_SET
rt_exp = NOT_SET
imp_targ = NOT_SET
exp_targ = NOT_SET
if "instance-vrf" in vrf.keys():
rd = vrf.get('instance-vrf') \
.get('route-distinguisher', NOT_SET)
rt_imp = vrf.get('instance-vrf') \
.get('vrf-import', NOT_SET)
rt_exp = vrf.get('instance-vrf') \
.get('vrf-export', NOT_SET)
imp_targ = vrf.get('instance-vrf') \
.get('vrf-import-target', NOT_SET)
exp_targ = vrf.get('instance-vrf') \
.get('vrf-export-target', NOT_SET)
vrf_list.vrf_lst.append(
VRF(vrf_name=_juniper_vrf_default_mapping(
vrf.get('instance-name')),
vrf_id=vrf.get('router-id', NOT_SET),
vrf_type=vrf.get('instance-type', NOT_SET),
l3_vni=NOT_SET,
rd=rd,
rt_imp=rt_imp,
rt_exp=rt_exp,
imp_targ=imp_targ,
exp_targ=exp_targ,
options=options))
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1):
printline()
PP.pprint(vrf_list.to_json())
return vrf_list
def _juniper_facts_api_converter(hostname: str(), cmd_output,
options={}) -> Facts:
interfaces_lst = list()
if FACTS_INT_DICT_KEY in cmd_output.keys():
for i in format_xml_output(cmd_output.get(FACTS_INT_DICT_KEY)).get(
'interface-information').get('physical-interface'):
interfaces_lst.append(i.get('name'))
hostname = NOT_SET
version = NOT_SET
model = NOT_SET
if FACTS_SYS_DICT_KEY in cmd_output.keys():
i = format_xml_output(cmd_output.get(FACTS_SYS_DICT_KEY))
hostname = i.get('software-information') \
.get('host-name', NOT_SET)
version = i.get('software-information') \
.get('junos-version', NOT_SET)
model = i.get('software-information') \
.get('product-name', NOT_SET).upper()
serial = NOT_SET
if FACTS_SERIAL_DICT_KEY in cmd_output.keys():
i = format_xml_output(cmd_output.get(FACTS_SERIAL_DICT_KEY))
serial = i.get('chassis-inventory') \
.get('chassis') \
.get('serial-number', NOT_SET)
memory = NOT_SET
if FACTS_MEMORY_DICT_KEY in cmd_output.keys():
i = format_xml_output(cmd_output.get(FACTS_MEMORY_DICT_KEY))
memory = i.get('system-memory-information') \
.get('system-memory-summary-information') \
.get('system-memory-total', NOT_SET)
facts = Facts(
hostname=hostname,
domain=NOT_SET,
version=version,
build=NOT_SET,
serial=serial,
base_mac=NOT_SET,
memory=memory,
vendor="Juniper",
model=model,
interfaces_lst=interfaces_lst,
options=options,
)
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1):
printline()
PP.pprint(facts.to_json())
return facts
def _extreme_vsp_facts_api_converter(
hostname: str,
cmd_output,
options={}
) -> Facts:
hostname = NOT_SET
domain = NOT_SET
memory = NOT_SET
if FACTS_SYS_DICT_KEY in cmd_output.keys():
hostname = cmd_output.get(FACTS_SYS_DICT_KEY) \
.get('openconfig-system:system') \
.get('config') \
.get('hostname', NOT_SET)
domain = cmd_output.get(FACTS_SYS_DICT_KEY) \
.get('openconfig-system:system') \
.get('config') \
.get('domain-name', NOT_SET)
memory = cmd_output.get(FACTS_SYS_DICT_KEY) \
.get('openconfig-system:system') \
.get('memory') \
.get('state') \
.get('physical', NOT_SET)
interfaces_lst = list()
if FACTS_INT_DICT_KEY in cmd_output.keys():
for i in cmd_output.get(FACTS_INT_DICT_KEY) \
.get('openconfig-interfaces:interfaces') \
.get('interface'):
interfaces_lst.append(i.get('name'))
facts = Facts(
hostname=hostname,
domain=domain,
version=NOT_SET,
build=NOT_SET,
serial=NOT_SET,
base_mac=NOT_SET,
memory=memory,
vendor="Extreme Networks",
model=NOT_SET,
interfaces_lst=interfaces_lst,
options=options
)
if verbose_mode(
user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1
):
printline()
PP.pprint(facts.to_json())
return facts
def _generic_vrf_napalm(task, options={}):
output = task.run(name=f"NAPALM get_bgp_neighbors {task.host.platform}",
task=napalm_get,
getters=["get_network_instances"])
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL2):
printline()
print_result(output)
if output.result != "":
task.host[VRF_DATA_KEY] = _napalm_vrf_converter(
hostname=task.host.name, cmd_output=output.result, options=options)
def _extreme_vsp_facts_ssh_converter(hostname: str(), cmd_output,
options={}) -> Facts:
interfaces_lst = list()
if FACTS_INT_DICT_KEY in cmd_output.keys():
cmd_output[FACTS_INT_DICT_KEY] = parse_textfsm(
content=cmd_output[FACTS_INT_DICT_KEY],
template_file="extrme_vsp_show_int_gi_name.textfsm")
for i in cmd_output.get(FACTS_INT_DICT_KEY):
interfaces_lst.append(i[0])
hostname = NOT_SET
version = NOT_SET
model = NOT_SET
serial = NOT_SET
base_mac = NOT_SET
if FACTS_SYS_DICT_KEY in cmd_output.keys():
cmd_output[FACTS_SYS_DICT_KEY] = parse_textfsm(
content=cmd_output[FACTS_SYS_DICT_KEY],
template_file="extreme_vsp_show_tech.textfsm")
for v in cmd_output.get(FACTS_SYS_DICT_KEY):
hostname = v[1] if v[1] != "" else NOT_SET
version = v[0] if v[0] != "" else NOT_SET
model = v[2] if v[2] != "" else NOT_SET
# vendor = v[3] if v[3] != "" else NOT_SET
serial = v[4] if v[4] != "" else NOT_SET
base_mac = v[5] if v[5] != "" else NOT_SET
domain = NOT_SET
if FACTS_SYS_DICT_KEY in cmd_output.keys():
cmd_output[FACTS_DOMAIN_DICT_KEY] = parse_textfsm(
content=cmd_output[FACTS_DOMAIN_DICT_KEY],
template_file="extreme_vsp_show_sys_dns.textfsm")
for v in cmd_output.get(FACTS_DOMAIN_DICT_KEY):
domain = v[0] if v[0] != "" else NOT_SET
facts = Facts(hostname=hostname,
domain=domain,
version=version,
build=NOT_SET,
serial=serial,
base_mac=base_mac,
memory=NOT_SET,
vendor="Extreme Networks",
model=model,
interfaces_lst=interfaces_lst,
options=options)
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1):
printline()
PP.pprint(facts.to_json())
return facts
def _arista_vrf_ssh_converter(
hostname: str(),
cmd_output,
options={}
) -> ListVRF:
if not isinstance(cmd_output, dict):
cmd_output = json.loads(cmd_output)
if verbose_mode(
user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL3
):
printline()
print(cmd_output)
vrf_list = ListVRF(list())
for vrf_name, facts in cmd_output.get('vrfs').items():
if (
facts.get('routeDistinguisher', NOT_SET) == NOT_SET or
facts.get('routeDistinguisher', NOT_SET) == ''
):
rd = NOT_SET
else:
rd = facts.get('routeDistinguisher', NOT_SET)
vrf_obj = VRF(
vrf_name=vrf_name,
vrf_id=NOT_SET,
vrf_type=NOT_SET,
l3_vni=NOT_SET,
rd=rd,
rt_imp=NOT_SET,
rt_exp=NOT_SET,
imp_targ=NOT_SET,
exp_targ=NOT_SET,
options=options
)
vrf_list.vrf_lst.append(
vrf_obj
)
if verbose_mode(
user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1
):
printline()
print(f">>>>> {hostname}")
PP.pprint(vrf_list.to_json())
return vrf_list
def _ios_vrf_ssh_converter(hostname: str(), cmd_output, options={}) -> ListVRF:
cmd_output = re.sub(pattern=r"communities[\n\r]\s+RT",
repl="communities:RT",
string=cmd_output)
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NSET),
needed_value=LEVEL4):
printline()
print(cmd_output)
template = open(f"{TEXTFSM_PATH}cisco_ios_show_ip_vrf_detail.textfsm")
results_template = textfsm.TextFSM(template)
# Example : [
# ['mgmt', '1', '<not set>'],
# ['tenant-1', '2', '10.255.255.103:103']
# ]
parsed_results = results_template.ParseText(cmd_output)
vrf_list = ListVRF(list())
# Add the default VRF maually
vrf_list.vrf_lst.append(
VRF(vrf_name="default",
vrf_id="0",
vrf_type=NSET,
l3_vni=NSET,
rd=NSET,
rt_imp=NSET,
rt_exp=NSET,
exp_targ=NSET,
imp_targ=NSET,
options=options))
for l in parsed_results:
vrf_list.vrf_lst.append(
VRF(vrf_name=l[0] if l[0] != "<not set>" and l[0] != '' else NSET,
vrf_id=l[1] if l[1] != "<not set>" and l[1] != '' else NSET,
vrf_type=NSET,
l3_vni=NSET,
rd=l[2] if l[2] != "<not set>" and l[2] != '' else NSET,
rt_imp=l[5] if l[5] != "<not set>" and l[5] != '' else NSET,
rt_exp=l[6] if l[6] != "<not set>" and l[6] != '' else NSET,
exp_targ=NSET,
imp_targ=NSET,
options=options))
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NSET),
needed_value=LEVEL1):
printline()
print(f">>>>> {hostname}")
PP.pprint(vrf_list.to_json())
return vrf_list
def select_host_vars(hostname: str, groups: list, protocol: str):
if truth_vars_exists() is False:
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL4):
printline()
print(f"{HEADER} Truth_vars doesn't exists")
return {}
if host_vars_exists(hostname, protocol):
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL4):
printline()
print(f"{HEADER} Select hosts variables")
return open_file(path=f"truth_vars/hosts/{hostname}/{protocol}.yml")
if group_vars_exists(groups[0], protocol):
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL4):
printline()
print(f"{HEADER} Select groups variables")
return open_file(path=f"truth_vars/groups/{groups[0]}/{protocol}.yml")
if all_vars_exists(protocol):
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL4):
printline()
print(f"{HEADER} Select all variables")
return open_file(path=f"truth_vars/all/{protocol}.yml")
def _juniper_get_bgp_api(task, options={}):
output_dict = dict()
output_dict['default'] = dict()
output_dict['default']['bgp'] = exec_http_call_juniper(
hostname=task.host.hostname,
port=task.host.port,
username=task.host.username,
password=task.host.password,
endpoint="get-bgp-neighbor-information?exact-instance=master",
secure_api=task.host['secure_api'])
output_dict['default']['rid'] = exec_http_call_juniper(
hostname=task.host.hostname,
port=task.host.port,
username=task.host.username,
password=task.host.password,
endpoint="get-instance-information?instance-name=master&detail=",
secure_api=task.host['secure_api'])
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL5):
printline()
print(output_dict['default']['rid'])
print(output_dict['default']['rid'])
for v in task.host[VRF_NAME_DATA_KEY].keys():
if v not in VRF_DEFAULT_RT_LST:
output_dict[v] = dict()
output_dict[v]['bgp'] = exec_http_call_juniper(
hostname=task.host.hostname,
port=task.host.port,
username=task.host.username,
password=task.host.password,
endpoint=f"get-bgp-neighbor-information?exact-instance={v}",
secure_api=task.host['secure_api'])
output_dict[v]['rid'] = exec_http_call_juniper(
hostname=task.host.hostname,
port=task.host.port,
username=task.host.username,
password=task.host.password,
endpoint=f"get-instance-information?instance-name={v}&detail=",
secure_api=task.host['secure_api'])
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE",
NOT_SET),
needed_value=LEVEL5):
printline()
print(output_dict[v]['bgp'])
print(output_dict[v]['rid'])
task.host[BGP_SESSIONS_HOST_KEY] = _juniper_bgp_api_converter(
hostname=task.host.name, cmd_output=output_dict, options=options)
def create_truth_vars(nr: Nornir, protocol: str) -> None:
for host in nr.inventory.hosts:
printline()
create_directories("truth_vars/")
create_directories("truth_vars/hosts")
create_directories(f"truth_vars/hosts/{host}")
with open(f"truth_vars/hosts/{host}/{M.get(protocol).get('filename')}",
'w') as outfile:
yaml.dump(
nr.inventory.hosts[host][M.get(protocol).get('key')].to_json(),
outfile,
default_flow_style=False)
print(f"{HEADER} - Data Generate in "
f"(truth_vars/hosts/{host}/{M.get(protocol).get('filename')}) "
f"for host {host}")
def _juniper_get_vrf_ssh(task, options={}):
if VRF_DATA_KEY not in task.host.keys():
data = task.run(
name=f"{JUNOS_GET_VRF_DETAIL}",
task=netmiko_send_command,
command_string=f"{JUNOS_GET_VRF_DETAIL}",
)
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL2):
printline()
print_result(data)
task.host[VRF_DATA_KEY] = _juniper_vrf_ssh_converter(
hostname=task.host.name,
cmd_output=json.loads(data.result),
options=options)
def _juniper_get_bgp_netconf(task, options={}):
output_dict = dict()
output_dict['default'] = dict()
with Device(
host=task.host.hostname,
port=task.host.port,
user=task.host.username,
passwd=task.host.password,
) as m:
output_dict['default']['bgp'] = m.rpc.get_bgp_neighbor_information(
exact_instance="master")
output_dict['default']['rid'] = m.rpc.get_instance_information(
detail=True, instance_name="master")
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL5):
printline()
print(ElementTree.tostring(output_dict['default']['bgp']))
print(ElementTree.tostring(output_dict['default']['rid']))
ElementTree.fromstring(
ElementTree.tostring(output_dict['default']['bgp']))
ElementTree.fromstring(
ElementTree.tostring(output_dict['default']['rid']))
for vrf in task.host[VRF_NAME_DATA_KEY].keys():
if vrf not in VRF_DEFAULT_RT_LST:
output_dict[vrf] = dict()
output_dict[vrf]['bgp'] = m.rpc.get_bgp_neighbor_information(
exact_instance=vrf)
output_dict[vrf]['rid'] = m.rpc.get_instance_information(
detail=True, instance_name=vrf)
if verbose_mode(user_value=os.environ.get(
"NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL5):
printline()
print(ElementTree.tostring(output_dict[vrf]['bgp']))
print(ElementTree.tostring(output_dict[vrf]['rid']))
ElementTree.fromstring(
ElementTree.tostring(output_dict[vrf]['bgp']))
ElementTree.fromstring(
ElementTree.tostring(output_dict[vrf]['rid']))
task.host[BGP_SESSIONS_HOST_KEY] = _juniper_bgp_netconf_converter(
hostname=task.host.name, cmd_output=output_dict, options=options)
def _arista_bgp_api_converter(hostname: str, cmd_output, options={}) -> BGP:
bgp_sessions_vrf_lst = ListBGPSessionsVRF(list())
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL3):
printline()
PP.pprint(cmd_output)
bgp_sessions_vrf_lst = ListBGPSessionsVRF(list())
for vrf in cmd_output.get('result', []):
if 'vrfs' in vrf.keys() and len(vrf.get('vrfs').keys()) > 0:
bgp_sessions_lst = ListBGPSessions(list())
for name, facts in vrf.get("vrfs", NOT_SET).items():
for n, f in facts.get('peers').items():
bgp_sessions_lst.bgp_sessions.append(
BGPSession(
src_hostname=hostname,
peer_ip=n,
peer_hostname=f.get("hostname", NOT_SET),
remote_as=f.get("asn", NOT_SET),
state_brief=get_bgp_state_brief(
state=f.get('peerState')),
session_state=f.get("peerState", NOT_SET),
state_time=f.get("upDownTime", NOT_SET),
prefix_received=f.get("prefixReceived", NOT_SET),
))
bgp_sessions_vrf_lst.bgp_sessions_vrf.append(
BGPSessionsVRF(
vrf_name=name,
as_number=facts.get('asn', NOT_SET),
router_id=facts.get("routerId", NOT_SET),
bgp_sessions=bgp_sessions_lst,
))
bgp = BGP(hostname=hostname, bgp_sessions_vrf_lst=bgp_sessions_vrf_lst)
if verbose_mode(user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1):
printline()
print(f">>>>> {hostname}")
PP.pprint(bgp.to_json())
return bgp
def _napalm_bgp_converter(hostname: str(), cmd_output: json,
options={}) -> BGP:
bgp_sessions_vrf_lst = ListBGPSessionsVRF(list())
local_as = ""
if "get_bgp_neighbors" in cmd_output.keys():
for k, v in cmd_output.get("get_bgp_neighbors").items():
bgp_sessions_lst = ListBGPSessions(list())
for peer_ip, facts in v.get("peers").items():
bgp_obj = BGPSession(
src_hostname=hostname,
peer_ip=peer_ip,
peer_hostname=facts.get("hostname", NOT_SET),
remote_as=facts.get("remote_as", NOT_SET),
state_brief=_napalm_bgp_status_converter(
facts.get("is_up", NOT_SET)),
session_state=facts.get("session_state", NOT_SET),
state_time=facts.get("uptime", NOT_SET),
prefix_received=facts.get("address_family").get(
"ipv4").get("received_prefixes"),
)
local_as = facts.get("local_as", NOT_SET)
bgp_sessions_lst.bgp_sessions.append(bgp_obj)
bgp_session_vrf = BGPSessionsVRF(
vrf_name=_napalm_bgp_vrf_converter(k),
as_number=local_as,
router_id=v.get("router_id"),
bgp_sessions=bgp_sessions_lst,
)
bgp_sessions_vrf_lst.bgp_sessions_vrf.append(bgp_session_vrf)
bgp = BGP(hostname=hostname, bgp_sessions_vrf_lst=bgp_sessions_vrf_lst)
if verbose_mode(user_value=os.environ.get('NETESTS_VERBOSE', NOT_SET),
needed_value=LEVEL1):
printline()
print(f'>>>>> {hostname}')
PP.pprint(bgp.to_json())
return bgp
def _cumulus_vrf_ssh_converter(
hostname: str(),
cmd_output,
options={}
) -> ListVRF:
template = open(
f"{TEXTFSM_PATH}cumulus_net_show_vrf.textfsm"
)
results_template = textfsm.TextFSM(template)
parsed_results = results_template.ParseText(cmd_output)
vrf_list = ListVRF(list())
if verbose_mode(
user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL3
):
printline()
print(parsed_results)
for line in parsed_results:
vrf = VRF(
vrf_name=line[0],
vrf_id=line[1],
vrf_type=NOT_SET,
l3_vni=NOT_SET,
rd=NOT_SET,
rt_imp=NOT_SET,
rt_exp=NOT_SET,
imp_targ=NOT_SET,
exp_targ=NOT_SET,
options=options
)
vrf_list.vrf_lst.append(vrf)
if verbose_mode(
user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL1
):
printline()
print(f">>>>> {hostname}")
PP.pprint(vrf_list.to_json())
return vrf_list
def _cumulus_get_vrf_ssh(task, options={}):
output = task.run(
name=f"{CUMULUS_GET_VRF}",
task=netmiko_send_command,
command_string=f"{CUMULUS_GET_VRF}",
)
if verbose_mode(
user_value=os.environ.get("NETESTS_VERBOSE", NOT_SET),
needed_value=LEVEL2
):
printline()
print_result(output)
task.host[VRF_DATA_KEY] = _cumulus_vrf_ssh_converter(
hostname=task.host.name,
cmd_output=output.result,
options=options
)