bornhack-website/src/program/mixins.py

49 lines
2 KiB
Python
Raw Normal View History

2017-03-12 14:43:41 +00:00
from django.views.generic.detail import SingleObjectMixin
from django.shortcuts import redirect
from django.urls import reverse
from . import models
from django.contrib import messages
2017-03-12 14:43:41 +00:00
2017-03-12 18:34:27 +00:00
class CreateProposalMixin(SingleObjectMixin):
2017-03-12 14:43:41 +00:00
def form_valid(self, form):
# set camp and user before saving
form.instance.camp = self.camp
form.instance.user = self.request.user
speaker = form.save()
return redirect(reverse('proposal_list', kwargs={'camp_slug': self.camp.slug}))
2017-03-12 14:43:41 +00:00
class EnsureUnpprovedProposalMixin(SingleObjectMixin):
2017-03-12 14:43:41 +00:00
def dispatch(self, request, *args, **kwargs):
# do not permit editing if the proposal is already approved
if self.get_object().proposal_status == models.UserSubmittedModel.PROPOSAL_APPROVED:
messages.error(request, "This proposal has already been approved. Please contact the organisers if you need to modify something." % self.camp.title)
return redirect(reverse('proposal_list', kwargs={'camp_slug': self.camp.slug}))
2017-03-12 14:43:41 +00:00
# alright, continue with the request
return super().dispatch(request, *args, **kwargs)
class EnsureWritableCampMixin(SingleObjectMixin):
2017-03-12 14:43:41 +00:00
def dispatch(self, request, *args, **kwargs):
# do not permit view if camp is in readonly mode
if self.camp.read_only:
messages.error(request, "No thanks")
return redirect(reverse('proposal_list', kwargs={'camp_slug': self.camp.slug}))
# alright, continue with the request
return super().dispatch(request, *args, **kwargs)
class EnsureUserOwnsProposalMixin(SingleObjectMixin):
def dispatch(self, request, *args, **kwargs):
# make sure that this proposal belongs to the logged in user
2017-03-12 14:43:41 +00:00
if self.get_object().user.username != request.user.username:
messages.error(request, "No thanks")
return redirect(reverse('proposal_list', kwargs={'camp_slug': self.camp.slug}))
2017-03-12 14:43:41 +00:00
# alright, continue with the request
return super().dispatch(request, *args, **kwargs)