work on admin ui

aircox/views/admin.py

@@ -0,0 +1,64 @@
+"""
+Aircox admin tools and views.
+"""
+import datetime
+
+from django.contrib import admin
+from django.contrib.auth.mixins import LoginRequiredMixin, \
+        PermissionRequiredMixin, UserPassesTestMixin
+from django.utils.translation import ugettext_lazy as _
+from django.views.generic import ListView
+
+from ..models import Program
+from .log import BaseLogListView
+
+
+class BaseAdminView(LoginRequiredMixin, UserPassesTestMixin):
+    title = ''
+
+    def test_func(self):
+        return self.request.user.is_staff
+
+    def get_context_data(self, **kwargs):
+        kwargs.update(admin.site.each_context(self.request))
+        kwargs.setdefault('title', self.title)
+        return super().get_context_data(**kwargs)
+
+
+class StatisticsView(BaseAdminView, BaseLogListView, ListView):
+    template_name = 'admin/aircox/statistics.html'
+    title = _('Statistics')
+    date = None
+
+    def get_queryset(self):
+        return super().get_queryset().today(self.date)
+
+    def get_diffusions_queryset(self):
+        return super().get_diffusions_queryset().today(self.date)
+
+    def get(self, request, *args, date=None, **kwargs):
+        self.date = datetime.date.today() if date is None else date
+        return super().get(request, *args, date=date, **kwargs)
+
+
+class AdminSite(admin.AdminSite):
+    def each_context(self, request):
+        context = super().each_context(request)
+        context.update({
+            'programs': Program.objects.all().active().values('pk', 'title'),
+        })
+        return context
+
+    def get_urls(self):
+        from django.urls import path, include
+        urls = super().get_urls()
+        urls += [
+            path('tools/statistics/',
+                 self.admin_view(StatisticsView.as_view()),
+                 name='tools-stats'),
+        ]
+        return urls
+
+
+admin_site = AdminSite()
+