From 5bbb493b94a63d66151a53837aa66f0d986df497 Mon Sep 17 00:00:00 2001
From: Bryan Newbold <bnewbold@robocracy.org>
Date: Fri, 24 Jul 2020 20:02:05 -0700
Subject: first iteration of flexible search coverage

---
 python/fatcat_web/search.py | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

(limited to 'python/fatcat_web/search.py')

diff --git a/python/fatcat_web/search.py b/python/fatcat_web/search.py
index 7ae7e73b..3ba6fdb2 100644
--- a/python/fatcat_web/search.py
+++ b/python/fatcat_web/search.py
@@ -366,6 +366,42 @@ def get_elastic_entity_stats() -> dict:
 
     return stats
 
+def get_elastic_search_coverage(query: ReleaseQuery) -> dict:
+
+    search = Search(using=app.es_client, index=app.config['ELASTICSEARCH_RELEASE_INDEX'])
+    search = search.query(
+        "query_string",
+        query=query.q,
+        default_operator="AND",
+        analyze_wildcard=True,
+        allow_leading_wildcard=False,
+        lenient=True,
+        fields=["biblio"],
+    )
+    search.aggs.bucket(
+        'preservation',
+        'terms',
+        field='preservation',
+        missing='_unknown',
+    )
+
+    search = search[:0]
+
+    search = search.params(request_cache=True)
+    resp = wrap_es_execution(search)
+
+    preservation_bucket = agg_to_dict(resp.aggregations.preservation)
+    preservation_bucket['total'] = resp.hits.total
+    for k in ('bright', 'dark', 'shadows_only', 'none'):
+        if not k in preservation_bucket:
+            preservation_bucket[k] = 0
+    stats = {
+        'total': resp.hits.total,
+        'preservation': preservation_bucket,
+    }
+
+    return stats
+
 def get_elastic_container_stats(ident, issnl=None):
     """
     Returns dict:
-- 
cgit v1.2.3