diff --git a/core/views/__init__.py b/core/views/__init__.py
index 6f7d219..b54ce10 100644
--- a/core/views/__init__.py
+++ b/core/views/__init__.py
@@ -1,5 +1,4 @@
import logging
-import pprint
import stripe
from django.conf import settings
@@ -15,7 +14,6 @@ from core.lib.products import assemble_plan_map
from core.models import ContentBlock, Plan, Session
logger = logging.getLogger(__name__)
-pp = pprint.PrettyPrinter(indent=4)
# Create your views here
diff --git a/core/views/dynamic/search.py b/core/views/dynamic/search.py
index 2de083c..231458c 100644
--- a/core/views/dynamic/search.py
+++ b/core/views/dynamic/search.py
@@ -1,14 +1,13 @@
-from rest_framework.views import APIView
-import logging
-from django.conf import settings
-from django.shortcuts import render
-from django.http import HttpResponse, JsonResponse
-from django.contrib.auth.mixins import LoginRequiredMixin
import json
-from django.views.decorators.csrf import csrf_exempt
+
+from django.conf import settings
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.http import HttpResponse, JsonResponse
+from django.shortcuts import render
from django.views import View
-from rest_framework.parsers import JSONParser
+
from core.lib.opensearch import initialise_opensearch, run_main_query
+
client = initialise_opensearch()
@@ -23,7 +22,6 @@ def query_results(request, post_params, api=False):
results = run_main_query(client, request.user, query, fields, size)
if not results:
return False
- # pp.pprint(results)
results_parsed = []
if "hits" in results.keys():
if "hits" in results["hits"]:
@@ -42,45 +40,41 @@ def query_results(request, post_params, api=False):
}
return context
+
class Search(LoginRequiredMixin, View):
- #parser_classes = [JSONParser]
+ # parser_classes = [JSONParser]
template_name = "ui/results.html"
plan_name = "drilldown"
def post(self, request):
if not request.user.has_plan(self.plan_name):
return render(request, "denied.html")
-
+
context = query_results(request, request.POST)
- print("context: ", context)
- context['data'] = json.dumps(
+ context["data"] = json.dumps(
[
{
- 'id': item.get('id'),
- 'value': item.get("sentiment", 0),
- 'date': item.get("ts"),
+ "id": item.get("id"),
+ "value": item.get("sentiment", 0),
+ "date": item.get("ts"),
}
for item in context["results"]
]
)
- print("context['data']: ", context['data'])
if context:
- print("OGING TO RENDER")
return render(request, self.template_name, context)
else:
return HttpResponse("No results")
+
class APISearch(LoginRequiredMixin, View):
- #parser_classes = [JSONParser]
+ # parser_classes = [JSONParser]
template_name = "ui/results.html"
plan_name = "drilldown"
def post(self, request):
- print("POST")
if not request.user.has_plan(self.plan_name):
return JsonResponse({"success": False})
- print("PERMS")
-
+
context = query_results(request, request.POST)
- print("CONTEXT", context)
return JsonResponse(context)