diff --git a/show/main.py b/show/main.py index fa5ec68913..dc5956eade 100755 --- a/show/main.py +++ b/show/main.py @@ -829,32 +829,32 @@ def expected(interfacename): #Swap Key and Value from interface: name to name: interface device2interface_dict = {} - for port in natsorted(neighbor_dict['DEVICE_NEIGHBOR'].keys()): + for port in natsorted(neighbor_dict.keys()): temp_port = port if get_interface_mode() == "alias": port = iface_alias_converter.name_to_alias(port) - neighbor_dict['DEVICE_NEIGHBOR'][port] = neighbor_dict['DEVICE_NEIGHBOR'].pop(temp_port) - device2interface_dict[neighbor_dict['DEVICE_NEIGHBOR'][port]['name']] = {'localPort': port, 'neighborPort': neighbor_dict['DEVICE_NEIGHBOR'][port]['port']} + neighbor_dict[port] = neighbor_dict.pop(temp_port) + device2interface_dict[neighbor_dict[port]['name']] = {'localPort': port, 'neighborPort': neighbor_dict[port]['port']} header = ['LocalPort', 'Neighbor', 'NeighborPort', 'NeighborLoopback', 'NeighborMgmt', 'NeighborType'] body = [] if interfacename: - for device in natsorted(neighbor_metadata_dict['DEVICE_NEIGHBOR_METADATA'].keys()): + for device in natsorted(neighbor_metadata_dict.keys()): if device2interface_dict[device]['localPort'] == interfacename: body.append([device2interface_dict[device]['localPort'], device, device2interface_dict[device]['neighborPort'], - neighbor_metadata_dict['DEVICE_NEIGHBOR_METADATA'][device]['lo_addr'], - neighbor_metadata_dict['DEVICE_NEIGHBOR_METADATA'][device]['mgmt_addr'], - neighbor_metadata_dict['DEVICE_NEIGHBOR_METADATA'][device]['type']]) + neighbor_metadata_dict[device]['lo_addr'], + neighbor_metadata_dict[device]['mgmt_addr'], + neighbor_metadata_dict[device]['type']]) else: - for device in natsorted(neighbor_metadata_dict['DEVICE_NEIGHBOR_METADATA'].keys()): + for device in natsorted(neighbor_metadata_dict.keys()): body.append([device2interface_dict[device]['localPort'], device, device2interface_dict[device]['neighborPort'], - neighbor_metadata_dict['DEVICE_NEIGHBOR_METADATA'][device]['lo_addr'], - neighbor_metadata_dict['DEVICE_NEIGHBOR_METADATA'][device]['mgmt_addr'], - neighbor_metadata_dict['DEVICE_NEIGHBOR_METADATA'][device]['type']]) + neighbor_metadata_dict[device]['lo_addr'], + neighbor_metadata_dict[device]['mgmt_addr'], + neighbor_metadata_dict[device]['type']]) click.echo(tabulate(body, header))