From d148daf4e215f5437ffa105e127f805bdd902337 Mon Sep 17 00:00:00 2001 From: Mehdi Abaakouk Date: Mon, 16 Jul 2012 08:09:17 +0200 Subject: [PATCH] Fix exception when bgpmap can not be generated --- lg.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/lg.py b/lg.py index bba6fdf..7ccca13 100755 --- a/lg.py +++ b/lg.py @@ -388,7 +388,7 @@ def show_bgpmap(): edge.set_style("bold") #colors = [ "#009e23", "#1a6ec1" , "#d05701", "#6f879f", "#939a0e", "#0e9a93", "#9a0e85", "#56d8e1" ] - + previous_as = None hosts = data.keys() for host, asmaps in data.iteritems(): first = True @@ -428,8 +428,9 @@ def show_bgpmap(): previous_as = _as first = False - node = add_node(previous_as) - node.set_shape("box") + if previous_as: + node = add_node(previous_as) + node.set_shape("box") #return Response("
" + graph.create_dot() + "
") return Response(graph.create_png(), mimetype='image/png') @@ -541,7 +542,7 @@ def show_route(request_type, hosts, proto): if bgpmap: detail = json.dumps(detail) - return render_template((bgpmap and 'bgpmap.html' or 'route.html'), detail=detail, command=command, expression=expression, error="
".join(error)) + return render_template((bgpmap and 'bgpmap.html' or 'route.html'), detail=detail, command=command, expression=expression, error="
".join(error)) app.secret_key = app.config["SESSION_KEY"]