bornhack-website/src/villages/views.py
Thomas Steen Rasmussen f952370280 typo
2017-08-23 03:09:41 +02:00

82 lines
2.8 KiB
Python

from django.http import Http404
from django.contrib.auth.mixins import LoginRequiredMixin
from django.core.urlresolvers import reverse_lazy
from django.http import HttpResponseRedirect
from django.views.generic import ListView, DetailView, CreateView, UpdateView, DeleteView
from django.views.generic.detail import SingleObjectMixin
from .models import Village
from camps.models import Camp
from camps.mixins import CampViewMixin
from .mixins import EnsureWritableCampMixin
from django.contrib import messages
class VillageListView(CampViewMixin, ListView):
queryset = Village.objects.not_deleted()
template_name = 'village_list.html'
context_object_name = 'villages'
class VillageDetailView(CampViewMixin, DetailView):
queryset = Village.objects.not_deleted()
template_name = 'village_detail.html'
context_object_name = 'village'
class VillageCreateView(CampViewMixin, LoginRequiredMixin, EnsureWritableCampMixin, CreateView):
model = Village
template_name = 'village_form.html'
fields = ['name', 'description', 'private']
def form_valid(self, form):
village = form.save(commit=False)
village.contact = self.request.user
village.camp = Camp.objects.get(slug=self.request.session['campslug'])
if not village.name:
village.name = "noname"
village.save()
return HttpResponseRedirect(village.get_absolute_url())
def get_success_url(self):
return reverse_lazy('village_list', kwargs={"camp_slug": self.object.camp.slug})
class EnsureUserOwnsVillageMixin(SingleObjectMixin):
model = Village
def dispatch(self, request, *args, **kwargs):
# If the user is not contact for this village OR is not staff
if not request.user.is_staff:
if self.get_object().contact != request.user:
raise Http404("Village not found")
return super(EnsureUserOwnsVillageMixin, self).dispatch(
request, *args, **kwargs
)
class VillageUpdateView(CampViewMixin, EnsureUserOwnsVillageMixin, LoginRequiredMixin, EnsureWritableCampMixin, UpdateView):
model = Village
queryset = Village.objects.not_deleted()
template_name = 'village_form.html'
fields = ['name', 'description', 'private']
def form_valid(self, form):
village = form.save(commit=False)
if not village.name:
village.name = "noname"
return HttpResponseRedirect(village.get_absolute_url())
def get_success_url(self):
return self.get_object().get_absolute_url()
class VillageDeleteView(CampViewMixin, EnsureUserOwnsVillageMixin, LoginRequiredMixin, EnsureWritableCampMixin, DeleteView):
model = Village
template_name = 'village_confirm_delete.html'
context_object_name = 'village'
def get_success_url(self):
return reverse_lazy('village_list', kwargs={"camp_slug": self.object.camp.slug})