provision aggregate dashboards

brian_dashboard_manager/grafana/provision.py

@@ -18,11 +18,12 @@ from brian_dashboard_manager.inventory_provider.interfaces import \
 from brian_dashboard_manager.templating.nren_access import generate_nrens
 
 from brian_dashboard_manager.templating.helpers import is_re_customer, \
-    is_cls, is_ias_customer, is_ias_private, is_ias_public, is_ias_upstream, \
-    is_lag_backbone, is_nren, is_phy_upstream, is_re_peer, is_gcs, \
-    is_geantopen, is_l2circuit, is_lhcone_peer, is_lhcone_customer, is_mdvpn,\
+    is_cls_peer, is_cls, is_ias_customer, is_ias_private, is_ias_public, \
+    is_ias_upstream, is_ias_peer, is_lag_backbone, is_nren, is_phy_upstream, \
+    is_re_peer, is_gcs, is_geantopen, is_l2circuit, is_lhcone_peer, \
+    is_lhcone_customer, is_lhcone, is_mdvpn, get_aggregate_dashboard_data, \
     get_interface_data, parse_backbone_name, parse_phy_upstream_name, \
-    get_dashboard_data
+    get_dashboard_data, get_aggregate_interface_data
 
 from brian_dashboard_manager.templating.render import render_dashboard
 
@@ -61,6 +62,21 @@ def provision_folder(token_request, folder_name,
                             rendered, folder['id'])
 
 
+def provision_aggregate(token_request, agg_type, aggregate_folder,
+                        dash, excluded_interfaces, datasource_name):
+    predicate = dash['predicate']
+    tag = dash['tag']
+
+    relevant_interfaces = filter(predicate, excluded_interfaces)
+    data = get_aggregate_interface_data(relevant_interfaces, agg_type)
+
+    dashboard = get_aggregate_dashboard_data(
+        f'Aggregate - {agg_type}', data, datasource_name, tag)
+
+    rendered = render_dashboard(dashboard)
+    create_dashboard(token_request, rendered, aggregate_folder['id'])
+
+
 def provision(config):
 
     request = AdminRequest(**config)
@@ -200,6 +216,39 @@ def provision(config):
                                 folder_name, dash,
                                 excluded_interfaces, datasource_name)
 
+        aggregate_dashboards = {
+            'CLS PEERS': {
+                'predicate': is_cls_peer,
+                'tag': 'cls_peers',
+            },
+            'IAS PEERS': {
+                'predicate': is_ias_peer,
+                'tag': 'ias_peers',
+            },
+            'GWS UPSTREAMS': {
+                'predicate': is_ias_upstream,
+                'tag': 'gws_upstreams',
+            },
+            'LHCONE': {
+                'predicate': is_lhcone,
+                'tag': 'lhcone',
+            },
+            # 'CAE1': {
+            #     'predicate': is_cae1,
+            #     'tag': 'cae',
+            # }
+        }
+
+        with ProcessPoolExecutor(max_workers=4) as executor:
+            aggregate_folder = find_folder(token_request, 'Aggregates')
+            for agg_type, dash in aggregate_dashboards.items():
+                logger.info(
+                    f'Provisioning {org["name"]}' +
+                    f'/Aggregate {agg_type} dashboards')
+                executor.submit(provision_aggregate, token_request, agg_type,
+                                aggregate_folder, dash,
+                                excluded_interfaces, datasource_name)
+
         # NREN Access dashboards
         # uses a different template than the above.
         logger.info('Provisioning NREN Access dashboards')