* fixed 404 errors in the gallery.

This commit is contained in:
2018-06-07 14:18:18 +02:00
parent f2836baae5
commit 1313d96e8f
2 changed files with 19 additions and 13 deletions

View File

@@ -1,6 +1,8 @@
"""Mixins for Events."""
from django.db.models import Q
from django.http import Http404
from django.shortcuts import get_object_or_404
from . import models
@@ -54,6 +56,20 @@ class EventDetailMixin(object):
self.event = get_object_or_404(models.Event, pk=self.kwargs['pk'])
queryset = self.model.objects.all()
else:
self.event = get_object_or_404(models.Event, pk=self.kwargs['event'])
self.event = get_object_or_404(models.Event,
pk=self.kwargs['event'])
queryset = self.model.objects.filter(event=self.event)
return queryset.prefetch_related()
class EventPhotoMixin(EventDetailMixin):
def get_queryset(self):
try:
self.event = models.Event.objects.get(id=self.kwargs['event'])
return models.Photo.objects.filter(
Q(event=self.event) |
Q(event__event_series=self.event)
)
except models.Event.DoesNotExist:
raise Http404(_('Event does not exist'))