Permalink
Switch branches/tags
Nothing to show
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
101 lines (81 sloc) 3 KB
#!/usr/bin/env python2
import pyeapi
from graphviz import Digraph
pyeapi.load_config('.eapi.conf')
#test devices in lab see .eapi.conf file in same directory for connection parameters
switches = ['Arista249', 'Arista250']
def gather_topos(devices):
"""
Function which takes a list of devices as input and runs the lldp neigh command against them
using the Arista pyeapi library
:param devices:
:return: list of lldp neigh output
"""
topo = []
for switch in devices:
node = pyeapi.connect_to(switch)
output = node.enable('show lldp neighbors')
device_neigh = {'name' :switch, 'neighbors': (output[0]['result']['lldpNeighbors'])}
topo.append(device_neigh)
return topo
# runs the gather_topos using the switches VAR from above and captures in the all_devs_neigh... VAR
all_devs_neighbors_output = gather_topos(switches)
def create_topo(dev_name, neigh_list):
"""
Function to take the neighbor list output from a single device and split into the nodes and edges for DOT consumption
:param dev_name:
:param neigh_list:
:return:
"""
nodes = [dev_name]
edges = []
for neighbor in neigh_list:
nodes.append(neighbor['neighborDevice'])
edges.append([dev_name, neighbor['neighborDevice'], neighbor['neighborPort'] + "-" + neighbor['port'] ])
return [nodes, edges]
def join_neighbors(all_devs_neighbors_output):
"""
function to take output of all_devs_neighbors_output VAR and process them through the create topo function above
:param all_devs_neighbors_output:
:return:
"""
links_list = []
for device in all_devs_neighbors_output:
dev_links = create_topo(device['name'], device['neighbors'])
links_list.append(dev_links)
return links_list
#gen full neighbors list and edge links
full = join_neighbors(all_devs_neighbors_output)
def clean_full_list(full_list):
"""Function to eliminate multiple nodes from the output of the join_neighbors"""
cleaned_node = []
cleaned_edge = []
for dev in full_list:
for node in dev[0]:
if node in cleaned_node:
continue
else:
cleaned_node.append(node)
for edge in dev[1]:
cleaned_edge.append(edge)
return [cleaned_node, cleaned_edge]
my_topo = clean_full_list(full)
#Creating the Topo graphic
def make_topology(network_name, mytopo):
dot = Digraph(comment=network_name, format='png')
dot.attr('node', shape='box')
dot.attr('node', image="./images/switch1.png")
dot.attr('edge', weight='10')
dot.attr('edge', arrowhead='none')
dot.body.append(r'label = "\n\nMy Prettier Network Diagram"')
dot.body.append('fontsize=20')
for i in mytopo[0]:
dot.node(i)
for i in mytopo[1]:
dot.edge(i[0], i[1], i[2])
return dot
dot = make_topology("My New Network", my_topo)
#run the following command if you want to see the object in DOT format
#print dot.source
#render to file
dot.render(filename='MultiTopo')