Improve context passing and implement superuser override for redactions

This commit is contained in:
2022-07-21 13:49:32 +01:00
parent 2362048cc7
commit 4df8afef1c
3 changed files with 41 additions and 27 deletions

View File

@@ -18,7 +18,11 @@ class Drilldown(LoginRequiredMixin, View):
def get(self, request):
if not request.user.has_plan(self.plan_name):
return render(request, "denied.html")
context = {"fields": settings.OPENSEARCH_MAIN_SEARCH_FIELDS}
context = {
"fields": settings.OPENSEARCH_MAIN_SEARCH_FIELDS,
"sizes": settings.OPENSEARCH_MAIN_SIZES,
"timescales": settings.OPENSEARCH_MAIN_TIMESCALES,
}
return render(request, self.template_name, context)
def post(self, request):
@@ -27,13 +31,15 @@ class Drilldown(LoginRequiredMixin, View):
fields = None
if "fields" in request.POST:
fields = request.POST.getlist("fields")
if "results" in request.POST:
results = request.POST["results"]
if "size" in request.POST:
size = request.POST["size"]
if "query" in request.POST:
query = request.POST["query"]
# field = results.POST["field"]
# print("FIELD ", field)
results = run_main_query(client, query, fields, results)
results = run_main_query(client, request.user, query, fields, size)
if not results:
return render(request, "denied.html")
# pp.pprint(results)
results_parsed = []
if "hits" in results.keys():
@@ -46,7 +52,10 @@ class Drilldown(LoginRequiredMixin, View):
"card": results["hits"]["total"]["value"],
"took": results["took"],
"redacted": results["redacted"],
"exemption": results["exemption"],
"fields": settings.OPENSEARCH_MAIN_SEARCH_FIELDS,
"sizes": settings.OPENSEARCH_MAIN_SIZES,
"timescales": settings.OPENSEARCH_MAIN_TIMESCALES,
}
return render(request, self.template_name, context)
return render(request, self.template_name)