From 1e2993a8e092bf4b725ebc23d6e79fe60a8b8978 Mon Sep 17 00:00:00 2001 From: Matthew Lemon Date: Tue, 29 Sep 2020 16:21:13 +0100 Subject: Delete button now works for Engagement Event --- ctrack/register/views.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) (limited to 'ctrack/register/views.py') diff --git a/ctrack/register/views.py b/ctrack/register/views.py index 3daabd7..32faf4d 100644 --- a/ctrack/register/views.py +++ b/ctrack/register/views.py @@ -2,23 +2,32 @@ from django.contrib.auth import get_user_model from django.contrib.auth.mixins import LoginRequiredMixin from django.http import HttpResponseRedirect from django.urls import reverse, reverse_lazy -from django.views.generic import FormView +from django.views.generic import FormView, DeleteView from ctrack.organisations.models import Organisation from ctrack.register.forms import EngagementEventCreateForm +from ctrack.register.models import EngagementEvent + + +class EngagementEventDelete(DeleteView): + model = EngagementEvent + + def get_success_url(self): + return reverse_lazy("organisations:detail", args=[self.kwargs["slug"]]) + class EngagementEventCreate(LoginRequiredMixin, FormView): fields = "__all__" form_class = EngagementEventCreateForm template_name = "register/engagementevent_form.html" - success_url = reverse_lazy("organisations:list") + # success_url = reverse_lazy("organisations:list") def form_valid(self, form): ee = form.save(commit=False) ee.user = self.request.user form.save() - return HttpResponseRedirect(reverse("organisations:list")) + return HttpResponseRedirect(self.get_success_url()) def get_form_kwargs(self): kwargs = super().get_form_kwargs() @@ -30,3 +39,6 @@ class EngagementEventCreate(LoginRequiredMixin, FormView): context = super().get_context_data() context["org"] = Organisation.objects.get(slug=self.kwargs["slug"]) return context + + def get_success_url(self): + return reverse_lazy("organisations:detail", args=[self.kwargs["slug"]]) -- cgit v1.2.3