import json from django.conf import settings from django.http import HttpResponse, JsonResponse from django.shortcuts import render from django.views import View from django_tables2 import SingleTableMixin from rest_framework.parsers import FormParser from rest_framework.views import APIView from core.lib.opensearch import query_results from core.lib.threshold import ( annotate_num_chans, annotate_num_users, get_chans, get_users, ) from core.views.ui.tables import DrilldownTable class DrilldownTableView(View, SingleTableMixin): table_class = DrilldownTable template_name = "ui/drilldown/table_results.html" paginate_by = 5 def post(self, request): context = query_results(request) table = DrilldownTable(context["results"]) context["table"] = table del context["results"] if "message" in context: return render(request, self.template_name, context) if self.request.htmx: print("IS HTMX") template_name = "ui/drilldown/table_results.html" else: print("IS NOT HTMX") template_name = "ui/drilldown/table_results_partial.html" if context: return render(request, template_name, context) else: return HttpResponse("No results") class Drilldown(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") if request.user.is_anonymous: sizes = settings.OPENSEARCH_MAIN_SIZES_ANON else: sizes = settings.OPENSEARCH_MAIN_SIZES context = { "sizes": sizes, } return render(request, self.template_name, context) class DrilldownSearch(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 "message" in context: return render(request, self.template_name, context) 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(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)