work on logs, timetable, stats

aircox/views/mixins.py

@@ -0,0 +1,67 @@
+from django.shortcuts import get_object_or_404
+
+from ..utils import str_to_date
+
+
+__all__ = ['GetDateMixin', 'ParentMixin']
+
+
+class GetDateMixin:
+    """
+    Mixin offering utils to get date by `request.GET` or
+    `kwargs['date']`
+    """
+    date = None
+
+    def get_date(self):
+        if 'date' in self.request.GET:
+            return str_to_date(self.request.GET['date'], '-')
+        return self.kwargs['date'] if 'date' in self.kwargs else None
+
+    def get(self, *args, **kwargs):
+        self.date = self.get_date()
+        return super().get(*args, **kwargs)
+
+
+class ParentMixin:
+    """
+    Optional parent page for a list view. Parent is fetched and passed to the
+    template context when `parent_model` is provided (queryset is filtered by
+    parent page in such case).
+    """
+    parent_model = None
+    """ Parent model """
+    parent_url_kwarg = 'parent_slug'
+    """ Url lookup argument """
+    parent_field = 'slug'
+    """ Parent field for url lookup """
+    fk_parent = 'page'
+    """ Page foreign key to the parent """
+    parent = None
+    """ Parent page object """
+
+    def get_parent(self, request, *args, **kwargs):
+        if self.parent_model is None or self.parent_url_kwarg not in kwargs:
+            return
+
+        lookup = {self.parent_field: kwargs[self.parent_url_kwarg]}
+        return get_object_or_404(
+            self.parent_model.objects.select_related('cover'), **lookup)
+
+    def get(self, request, *args, **kwargs):
+        self.parent = self.get_parent(request, *args, **kwargs)
+        return super().get(request, *args, **kwargs)
+
+    def get_queryset(self):
+        if self.parent is not None:
+            lookup = {self.fk_parent: self.parent}
+            return super().get_queryset().filter(**lookup)
+        return super().get_queryset()
+
+    def get_context_data(self, **kwargs):
+        parent = kwargs.setdefault('parent', self.parent)
+        if parent is not None:
+            kwargs.setdefault('cover', parent.cover)
+        return super().get_context_data(**kwargs)
+
+