AttributeError: 'NoneType' object has no attribute 'split'

[oszafraniec]

Sometimes there is an error:

Traceback (most recent call last):
  File "/usr/lib/python2.7/dist-packages/flask/app.py", line 1997, in __call__
    return self.wsgi_app(environ, start_response)
  File "/usr/lib/python2.7/dist-packages/flask/app.py", line 1985, in wsgi_app
    response = self.handle_exception(e)
  File "/usr/lib/python2.7/dist-packages/flask/app.py", line 1540, in handle_exception
    reraise(exc_type, exc_value, tb)
  File "/usr/lib/python2.7/dist-packages/flask/app.py", line 1982, in wsgi_app
    response = self.full_dispatch_request()
  File "/usr/lib/python2.7/dist-packages/flask/app.py", line 1614, in full_dispatch_request
    rv = self.handle_user_exception(e)
  File "/usr/lib/python2.7/dist-packages/flask/app.py", line 1517, in handle_user_exception
    reraise(exc_type, exc_value, tb)
  File "/usr/lib/python2.7/dist-packages/flask/app.py", line 1612, in full_dispatch_request
    rv = self.dispatch_request()
  File "/usr/lib/python2.7/dist-packages/flask/app.py", line 1598, in dispatch_request
    return self.view_functions[rule.endpoint](**req.view_args)
  File "/home/staff/oszafraniec/bird-lg-master/lg.py", line 493, in show_bgpmap
    edge = add_edge(nodes[previous_as], nodes[_as], label=hop_label, fontsize="7")
  File "/home/staff/oszafraniec/bird-lg-master/lg.py", line 442, in add_edge
    labels = e.get_label().split("\r")
AttributeError: 'NoneType' object has no attribute 'split'

I think this will fix it...

diff --git a/lg.py b/lg.py
index ab2dcfe..4851c83 100644
--- a/lg.py
+++ b/lg.py
@@ -437,7 +437,10 @@ def show_bgpmap():
             e = edges[edge_tuple]
 
             label_without_star = kwargs["label"].replace("*", "")
-            labels = e.get_label().split("\r") 
+            if e.get_label() is not None:
+               labels = e.get_label().split("\r") 
+           else:
+               return edges[edge_tuple]
             if "%s*" % label_without_star not in labels:
                 labels = [ kwargs["label"] ]  + [ l for l in labels if not l.startswith(label_without_star) ] 
                 labels = sorted(labels, cmp=lambda x,y: x.endswith("*") and -1 or 1)