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})