class TypeEditView(TemplateView):
template_name = 'store/type_edit.html'
@method_decorator(permission_required('typeofstore.change_typeofstore', raise_exception=True))
def dispatch(self, request, *args, **kwargs):
if kwargs['pk'] != 'new':
self.instance = get_object_or_404(TypeOfStore, pk=kwargs['pk'])
else:
self.instance = None
self.form = TypeEditForm(request.POST or None, instance=self.instance)
return super(TypeEditView, self).dispatch(request, *args, **kwargs)
def get_context_data(self, **kwargs):
context = super(TypeEditView, self).get_context_data(**kwargs)
context['form'] = self.form
return context
def post(self, request, *args, **kwargs):
if self.form.is_valid():
saved_instance = self.form.save()
if not self.instance:
messages.success(request, u'Type created successfully')
else:
messages.success(request, u'Type saved successfully')
return redirect('type', pk=saved_instance.pk)
return self.get(request, *args, **kwargs)
def dispatch(self, request, *args, **kwargs):
if kwargs['pk'] != 'new':
self.instance = get_object_or_404(TypeOfStore, pk=kwargs['pk'])
else:
self.instance = None
self.form = TypeEditForm(request.POST or None, instance=self.instance)
return super(TypeEditView, self).dispatch(request, *args, **kwargs)
