From d4341d0704ef09f111f3994a7040ec371eb99e33 Mon Sep 17 00:00:00 2001 From: Karl Newell Date: Fri, 27 Sep 2019 18:46:39 -0400 Subject: [PATCH] Update main.py. Fix key error --- main.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index 303b1a9..d307dbf 100644 --- a/main.py +++ b/main.py @@ -449,9 +449,10 @@ def loadFlowRouteConfig(self): for table in ['inet', 'inet6']: if table == 'inet6': - if 'flow' in _vrf_data['rib'][0]: - _data = _vrf_data['rib'][0]['flow'] - family = 'v6' + if 'rib' in _vrf_data: + if 'flow' in _vrf_data['rib'][0]: + _data = _vrf_data['rib'][0]['flow'] + family = 'v6' else: if 'flow' in _vrf_data: _data = _vrf_data['flow']