Separate core from dashboard

shynet/dashboard/views.py

@@ -0,0 +1,122 @@
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.shortcuts import get_object_or_404, reverse
+from django.utils import timezone
+from django.views.generic import (
+    CreateView,
+    DeleteView,
+    DetailView,
+    ListView,
+    TemplateView,
+    UpdateView,
+)
+from rules.contrib.views import PermissionRequiredMixin
+from django.db.models import Q
+
+from analytics.models import Session
+
+from .forms import ServiceForm
+from .mixins import BaseUrlMixin, DateRangeMixin
+from core.models import Service
+
+class DashboardView(LoginRequiredMixin, DateRangeMixin, TemplateView):
+    template_name = "dashboard/pages/dashboard.html"
+
+    def get_context_data(self, **kwargs):
+        data = super().get_context_data(**kwargs)
+        data["services"] = Service.objects.filter(
+            Q(owner=self.request.user) | Q(collaborators__in=[self.request.user])
+        )
+        for service in data["services"]:
+            service.stats = service.get_core_stats(data["start_date"], data["end_date"])
+        return data
+
+
+class ServiceCreateView(LoginRequiredMixin, PermissionRequiredMixin, CreateView):
+    model = Service
+    form_class = ServiceForm
+    template_name = "dashboard/pages/service_create.html"
+    permission_required = "core.create_service"
+
+    def form_valid(self, form):
+        form.instance.owner = self.request.user
+        return super().form_valid(form)
+
+    def get_success_url(self):
+        return reverse("dashboard:service", kwargs={"pk": self.object.uuid})
+
+
+class ServiceView(
+    LoginRequiredMixin, PermissionRequiredMixin, DateRangeMixin, DetailView
+):
+    model = Service
+    template_name = "dashboard/pages/service.html"
+    permission_required = "core.view_service"
+
+    def get_context_data(self, **kwargs):
+        data = super().get_context_data(**kwargs)
+        data["stats"] = self.object.get_core_stats(data["start_date"], data["end_date"])
+        data["object_list"] = Session.objects.filter(
+            service=self.get_object(),
+            start_time__lt=self.get_end_date(),
+            start_time__gt=self.get_start_date(),
+        ).order_by("-start_time")[:10]
+        return data
+
+
+class ServiceUpdateView(
+    LoginRequiredMixin, PermissionRequiredMixin, BaseUrlMixin, UpdateView
+):
+    model = Service
+    form_class = ServiceForm
+    template_name = "dashboard/pages/service_update.html"
+    permission_required = "core.change_service"
+
+    def get_success_url(self):
+        return reverse("dashboard:service", kwargs={"pk": self.object.uuid})
+
+
+class ServiceDeleteView(LoginRequiredMixin, PermissionRequiredMixin, DeleteView):
+    model = Service
+    form_class = ServiceForm
+    template_name = "dashboard/pages/service_delete.html"
+    permission_required = "core.delete_service"
+
+    def get_success_url(self):
+        return reverse("dashboard:dashboard")
+
+
+class ServiceSessionsListView(
+    LoginRequiredMixin, PermissionRequiredMixin, DateRangeMixin, ListView
+):
+    model = Session
+    template_name = "dashboard/pages/service_session_list.html"
+    paginate_by = 20
+    permission_required = "core.view_service"
+
+    def get_object(self):
+        return get_object_or_404(Service, pk=self.kwargs.get("pk"))
+
+    def get_queryset(self):
+        return Session.objects.filter(
+            service=self.get_object(),
+            start_time__lt=self.get_end_date(),
+            start_time__gt=self.get_start_date(),
+        ).order_by("-start_time")
+
+    def get_context_data(self, **kwargs):
+        data = super().get_context_data(**kwargs)
+        data["object"] = self.get_object()
+        return data
+
+
+class ServiceSessionView(LoginRequiredMixin, PermissionRequiredMixin, DetailView):
+    model = Session
+    template_name = "dashboard/pages/service_session.html"
+    pk_url_kwarg = "session_pk"
+    context_object_name = "session"
+    permission_required = "core.view_service"
+
+    def get_context_data(self, **kwargs):
+        data = super().get_context_data(**kwargs)
+        data["object"] = get_object_or_404(Service, pk=self.kwargs.get("pk"))
+        return data