diff --git a/examples/clustercli.py b/examples/clustercli.py index 264f901c6d6acc9daa3314fc088a97342ab767f4..c8d26de8b5c019db073e8c7b7fc9928790c199c2 100644 --- a/examples/clustercli.py +++ b/examples/clustercli.py @@ -53,8 +53,8 @@ def do_plot( self, line ): pos = nx.graphviz_layout( g ) opts = { 'ax': None, 'font_weight': 'bold', 'width': 2, 'edge_color': 'darkblue' } - hcolors = [ color[ h.server ] for h in hosts ] - scolors = [ color[ s.server ] for s in switches ] + hcolors = [ color[ getattr( h, 'server', 'localhost' ) ] for h in hosts ] + scolors = [ color[ getattr( s, 'server', 'localhost' ) ] for s in switches ] nx.draw_networkx( g, pos=pos, nodelist=hosts, node_size=800, label='host', node_color=hcolors, node_shape='s', **opts ) nx.draw_networkx( g, pos=pos, nodelist=switches, node_size=1000,