fix get empty rib in issue

Signed-off-by: Peng Xiao <xiaoquwl@gmail.com>
smartbgp · Jul 8, 2016 · 3bf7f24 · 3bf7f24
1 parent cc23231
commit 3bf7f24
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/yabgp/api/utils.py b/yabgp/api/utils.py
@@ -161,7 +161,7 @@ def send_update(peer_ip, attr, nlri, withdraw):
 
 
 def get_adj_rib_in(peer_ip, afi_safi, prefix=None):
-    rib_table = cfg.CONF.bgp.running_config[peer_ip]['factory'].fsm.protocol.get_rib_in().get(afi_safi)
+    rib_table = cfg.CONF.bgp.running_config[peer_ip]['factory'].fsm.protocol.get_rib_in().get(afi_safi) or {}
     if prefix:
         attr = rib_table.get(prefix)
         if not attr:
@@ -172,7 +172,7 @@ def get_adj_rib_in(peer_ip, afi_safi, prefix=None):
 
 
 def get_adj_rib_out(peer_ip, afi_safi, prefix=None):
-    rib_table = cfg.CONF.bgp.running_config[peer_ip]['factory'].fsm.protocol.get_rib_out().get(afi_safi)
+    rib_table = cfg.CONF.bgp.running_config[peer_ip]['factory'].fsm.protocol.get_rib_out().get(afi_safi) or {}
     if prefix:
         attr = rib_table.get(prefix)
         if not attr:
@@ -196,6 +196,6 @@ def restart_bgp_connection(peer_ip):
 
 
 def get_adj_rib_all(peer_ip, afi_safi):
-    rib_table = cfg.CONF.bgp.running_config[peer_ip]['factory'].fsm.protocol.get_rib_in().get(afi_safi)
+    rib_table = cfg.CONF.bgp.running_config[peer_ip]['factory'].fsm.protocol.get_rib_in().get(afi_safi) or {}
 
     return rib_table