Add '*' to request all environments

puppetboard/app.py

@@ -79,7 +79,7 @@ def environments():
     return x
 
 def check_env(env):
-    if env not in envs:
+    if env != '*' and env not in envs:
         abort(404)
 
 app.jinja_env.globals['url_for_pagination'] = url_for_pagination
@@ -151,10 +151,15 @@ def index(env):
         'avg_resources_node': "{0:10.0f}".format(avg_resources_node['Value']),
         }
 
-    nodes = get_or_abort(puppetdb.nodes,
+    if env == '*':
+        query = None
+    else:
         query = '["and", {0}]'.format(
                 ", ".join('["=", "{0}", "{1}"]'.format(field, env)
-                for field in ['catalog_environment', 'facts_environment'])),
+                    for field in ['catalog_environment', 'facts_environment']))
+
+    nodes = get_or_abort(puppetdb.nodes,
+        query=query,
         unreported=app.config['UNRESPONSIVE_HOURS'],
         with_status=True)
 
@@ -209,11 +214,16 @@ def nodes(env):
     """
     check_env(env)
 
-    status_arg = request.args.get('status', '')
+    if env == '*':
-    nodelist = puppetdb.nodes(
+        query = None
+    else:
         query = '["and", {0}]'.format(
             ", ".join('["=", "{0}", "{1}"]'.format(field, env)
                 for field in ['catalog_environment', 'facts_environment'])),
+
+    status_arg = request.args.get('status', '')
+    nodelist = puppetdb.nodes(
+        query=query,
         unreported=app.config['UNRESPONSIVE_HOURS'],
         with_status=True)
     nodes = []
@@ -272,6 +282,11 @@ def inventory(env):
         fact_desc.append(description)
         fact_names.append(name)
 
+    if env == '*':
+        query = '["or", {0}]]'.format(
+            ', '.join('["=", "name", "{0}"]'.format(name)
+                for name in fact_names))
+    else:
         query = '["and", ["=", "environment", "{0}"], ["or", {1}]]'.format(
             env,
             ', '.join('["=", "name", "{0}"]'.format(name)
@@ -314,11 +329,16 @@ def node(env, node_name):
     """
     check_env(env)
 
+    if env == '*':
+        query = '["=", "certname", "{0}"]]'.format(node_name)
+    else:
+        query='["and", ["=", "environment", "{0}"],' \
+            '["=", "certname", "{1}"]]'.format(env, node_name),
+
     node = get_or_abort(puppetdb.node, node_name)
     facts = node.facts()
     reports = get_or_abort(puppetdb.reports,
-        query='["and", ["=", "environment", "{0}"],' \
+        query=query,
-            '["=", "certname", "{1}"]]'.format(env, node_name),
         limit=app.config['REPORTS_COUNT'],
         order_by='[{"field": "start_time", "order": "desc"}]')
     reports, reports_events = tee(reports)
@@ -362,16 +382,22 @@ def reports(env, page):
     """
     check_env(env)
 
+    if env == '*':
+        reports_query = None
+        total_query = '["extract", [["function", "count"]], ["~", "certname", ""]]'
+    else:
+        reports_query = '["=", "environment", "{0}"]'.format(env)
+        total_query = '["extract", [["function", "count"]],'\
+            '["and", ["=", "environment", "{0}"]]]'.format(env)
+
     reports = get_or_abort(puppetdb.reports,
-        query='["=", "environment", "{0}"]'.format(env),
+        query=reports_query,
         limit=app.config['REPORTS_COUNT'],
         offset=(page-1) * app.config['REPORTS_COUNT'],
         order_by='[{"field": "start_time", "order": "desc"}]')
     total = get_or_abort(puppetdb._query,
         'reports',
-        query='["extract", [["function", "count"]],'\
+        query=total_query)
-            '["and", ["=", "environment", "{0}"]]]'.format(
-                env))
     total = total[0]['count']
     reports, reports_events = tee(reports)
     report_event_counts = {}
@@ -420,10 +446,15 @@ def reports_node(env, node_name, page):
     """
     check_env(env)
 
-    reports = get_or_abort(puppetdb.reports,
+    if env == '*':
+        query = '["=", "certname", "{0}"]]'.format(node_name)
+    else:
         query='["and",' \
             '["=", "environment", "{0}"],' \
             '["=", "certname", "{1}"]]'.format(env, node_name),
+
+    reports = get_or_abort(puppetdb.reports,
+        query=query,
         limit=app.config['REPORTS_COUNT'],
         offset=(page-1) * app.config['REPORTS_COUNT'],
         order_by='[{"field": "start_time", "order": "desc"}]')
@@ -473,13 +504,19 @@ def report_latest(env, node_name):
     """
     check_env(env)
 
-    reports = get_or_abort(puppetdb.reports,
+    if env == '*':
+        query='["and",' \
+            '["=", "certname", "{1}"],' \
+            '["=", "latest_report?", true]]'.format(node_name)
+    else:
         query='["and",' \
             '["=", "environment", "{0}"],' \
             '["=", "certname", "{1}"],' \
             '["=", "latest_report?", true]]'.format(
                 env,
-                                   node_name))
+                node_name)
+
+    reports = get_or_abort(puppetdb.reports, query=query)
     try:
         report = next(reports)
     except StopIteration:
@@ -509,6 +546,11 @@ def report(env, node_name, report_id):
     """
     check_env(env)
 
+    if env == '*':
+        query = '["and", ["=", "certname", "{0}"],' \
+            '["or", ["=", "hash", "{1}"], ["=", "configuration_version", "{1}"]]]'.format(
+                node_name, report_id)
+    else:
         query = '["and", ["=", "environment", "{0}"], ["=", "certname", "{1}"],' \
             '["or", ["=", "hash", "{2}"], ["=", "configuration_version", "{2}"]]]'.format(
                 env, node_name, report_id)
@@ -574,9 +616,12 @@ def fact(env, fact):
     render_graph = False
     if fact in graph_facts:
         render_graph = True
+    if env == '*':
+        query = None
+    else:
+        query = '["=", "environment", "{0}"]'.format(env)
     localfacts = [f for f in yield_or_stop(puppetdb.facts(
-        name=fact,
+        name=fact, query=query))]
-        query='["=", "environment", "{0}"]'.format(env)))]
     return Response(stream_with_context(stream_template(
         'fact.html',
         name=fact,
@@ -600,10 +645,14 @@ def fact_value(env, fact, value):
     """
     check_env(env)
 
+    if env == '*':
+        query = None
+    else:
+        query = '["=", "environment", "{0}"]'.format(env)
     facts = get_or_abort(puppetdb.facts,
         name=fact,
         value=value,
-        query='["=", "environment", "{0}"]'.format(env))
+        query=query)
     localfacts = [f for f in yield_or_stop(facts)]
     return render_template(
         'fact.html',
@@ -707,10 +756,14 @@ def catalogs(env):
     if app.config['ENABLE_CATALOG']:
         nodenames = []
         catalog_list = []
-        nodes = get_or_abort(puppetdb.nodes,
+        if env == '*':
+            query = '["null?", "catalog_timestamp", false]]'
+        else:
             query = '["and",' \
                 '["=", "catalog_environment", "{0}"],' \
                 '["null?", "catalog_timestamp", false]]'.format(env),
+        nodes = get_or_abort(puppetdb.nodes,
+            query=query,
             with_status=False,
             order_by='[{"field": "certname", "order": "asc"}]')
         nodes, temp = tee(nodes)