Merge pull request mikenowak#6 from mikenowak/down-protocol

adding code to handle protocol in a down state
WolffyTheReal · Mar 26, 2018 · 2df2849 · 2df2849
2 parents aee6110 + 12ef4f8
commit 2df2849
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/birdagent.py b/birdagent.py
@@ -255,8 +255,18 @@ def getBGPState(self):
                         match = peerprop_re.search(line)
                         if match:
                             if peerprop_name == 'bgpPeerState':
-                                state["bgp-peers"][bgp_proto][peerprop_name] = \
-                                    self.bgp_states[match.group(1).lower()]
+                                if not match.group(1).lower() == 'down':
+                                    state["bgp-peers"][bgp_proto][peerprop_name] = \
+                                        self.bgp_states[match.group(1).lower()]
+                                else:
+                                    # handle disabled (down) protocols
+                                    state["bgp-peers"][bgp_proto][peerprop_name] = int(
+                                        1)
+                                    state["bgp-peers"][bgp_proto]["bgpPeerAdminStatus"] = int(
+                                        1)
+                                    state["bgp-peers"][bgp_proto]["bgpPeerFsmEstablishedTime"] = int(
+                                        0)
+
                             elif peerprop_name in [
                                 'bgpPeerIdentifier',
                                     'bgpPeerLocalAddr',