Refactor code into fewer files, removing dynamic folders

2022-07-27 13:12:16 +01:00
parent da235fedd6
commit 3d91c4164e
16 changed files with 53 additions and 63 deletions
--- a/core/views/ui/drilldown.py
+++ b/core/views/ui/drilldown.py
@@ -0,0 +1,100 @@
+from django.conf import settings
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.shortcuts import render
+from django.views import View
+from django.http import HttpResponse, HttpResponseForbidden, JsonResponse
+from rest_framework.parsers import FormParser
+from rest_framework.views import APIView
+import json
+from core.lib.opensearch import query_results
+
+
+from core.lib.threshold import (
+    annotate_num_chans,
+    annotate_num_users,
+    get_chans,
+    get_users,
+)
+
+class Drilldown(LoginRequiredMixin, View):
+    template_name = "ui/drilldown/drilldown.html"
+    plan_name = "drilldown"
+
+    def get(self, request):
+        if not request.user.has_plan(self.plan_name):
+            return render(request, "denied.html")
+        context = {
+            "sizes": settings.OPENSEARCH_MAIN_SIZES,
+        }
+        return render(request, self.template_name, context)
+
+class DrilldownSearch(LoginRequiredMixin, View):
+    # parser_classes = [JSONParser]
+    template_name = "ui/drilldown/results.html"
+    plan_name = "drilldown"
+
+    def post(self, request):
+        if not request.user.has_plan(self.plan_name):
+            return HttpResponseForbidden()
+
+        context = query_results(request)
+        if not context:
+            return HttpResponseForbidden()
+        context["data"] = json.dumps(
+            [
+                {
+                    "text": item.get("msg", None) or item.get("id"),
+                    "nick": item.get("nick", None),
+                    "value": item.get("sentiment", None) or None,
+                    "date": item.get("ts"),
+                }
+                for item in context["results"]
+            ]
+        )
+        if context:
+            return render(request, self.template_name, context)
+        else:
+            return HttpResponse("No results")
+
+
+class ThresholdInfoModal(LoginRequiredMixin, APIView):
+    parser_classes = [FormParser]
+    plan_name = "drilldown"
+    template_name = "modals/drilldown.html"
+
+    def post(self, request):
+        if not request.user.has_plan(self.plan_name):
+            return JsonResponse({"success": False})
+        if "net" not in request.data:
+            return JsonResponse({"success": False})
+        if "nick" not in request.data:
+            return JsonResponse({"success": False})
+        if "channel" not in request.data:
+            return JsonResponse({"success": False})
+        net = request.data["net"]
+        nick = request.data["nick"]
+        channel = request.data["channel"]
+        channels = get_chans(net, [nick])
+        users = get_users(net, [channel])
+        num_users = annotate_num_users(net, channels)
+        num_chans = annotate_num_chans(net, users)
+        if channels:
+            inter_users = get_users(net, channels)
+        else:
+            inter_users = []
+        if users:
+            inter_chans = get_chans(net, users)
+        else:
+            inter_chans = []
+        context = {
+            "net": net,
+            "nick": nick,
+            "channel": channel,
+            "chans": channels,
+            "users": users,
+            "inter_chans": inter_chans,
+            "inter_users": inter_users,
+            "num_users": num_users,
+            "num_chans": num_chans,
+        }
+        return render(request, self.template_name, context)