Update main.py - basic VRF support. Need to add UI components to dist…

…inguish VRFs

main.py

@@ -152,7 +152,7 @@ def getActiveFlowRoutes(self):
 
                     # data = dev.rpc.get_config(filter_xml='routing-options/flow/route/name')
                     frt = FlowRoutesTable(dev)
-                    for flowtable in ['inetflow.0', 'inet6flow.0']:
+                    for flowtable in ['inetflow.0', 'inet6flow.0', 'trcps.inetflow.0', 'trcps.inet6flow.0']:
                         frt.get(table=flowtable)
 
                         for flow in frt:
@@ -262,7 +262,7 @@ def getActiveFlowRouteFilter(self):
                     with Device(host=value['ip'], user=self.dev_user, password=self.dev_pw) as dev:
 
                         frft = FlowFilterTable(dev)
-                        for table in ['__flowspec_default_inet__', '__flowspec_default_inet6__']:
+                        for table in ['__flowspec_default_inet__', '__flowspec_default_inet6__', '__flowspec_trcps_inet__', '__flowspec_trcps_inet6__']:
                             frft.get(filtername=table)
 
                             for filter in frft:
@@ -299,6 +299,10 @@ def loadFlowRouteConfig(self):
 
                 # Retrieving all routing-options so we can get both v4 and v6.  Might also work for VRFs
                 data = dev.rpc.get_config(options={'format': 'xml'}, filter_xml='routing-options')
+                # Retrieving all routing-instances.  Should figure out how to limit to specified VRFs
+                data_trcps = dev.rpc.get_config(options={'format': 'xml'}, filter_xml='routing-instances')
+
+                data.append(data_trcps)
 
                 # iterate only over <flow> children. Appears to enter into multiple levels implicitly
                 for routes in data.iter('flow'):