Update main.py. Fix key error

internet2 · Sep 27, 2019 · d4341d0 · d4341d0
1 parent 8890c0b
commit d4341d0
Showing 1 changed file with 4 additions and 3 deletions.
diff --git 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']