41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
from django.core.urlresolvers import reverse_lazy
|
|
from django.http import HttpResponseRedirect
|
|
from django.views.generic import ListView, DetailView, CreateView, UpdateView
|
|
from .models import (
|
|
Village,
|
|
)
|
|
|
|
|
|
class VillageListView(ListView):
|
|
model = Village
|
|
template_name = 'village_list.html'
|
|
context_object_name = 'villages'
|
|
|
|
|
|
class VillageDetailView(DetailView):
|
|
model = Village
|
|
template_name = 'village_detail.html'
|
|
context_object_name = 'village'
|
|
|
|
|
|
class VillageCreateView(CreateView):
|
|
model = Village
|
|
template_name = 'village_form.html'
|
|
fields = ['name', 'description', 'open']
|
|
success_url = reverse_lazy('villages:list')
|
|
|
|
def form_valid(self, form):
|
|
village = form.save(commit=False)
|
|
village.contact = self.request.user
|
|
village.save()
|
|
return HttpResponseRedirect(village.get_absolute_url())
|
|
|
|
|
|
class VillageUpdateView(UpdateView):
|
|
model = Village
|
|
template_name = 'village_form.html'
|
|
fields = ['name', 'description', 'open']
|
|
|
|
def get_success_url(self):
|
|
return self.get_object().get_absolute_url()
|