Display guests list

apps/activity/views.py

@@ -7,15 +7,18 @@ from django.views.generic import CreateView, DetailView, UpdateView, TemplateVie
 from django.utils.translation import gettext_lazy as _
 from django_tables2.views import SingleTableView
 
+from permission.backends import PermissionBackend
 from .forms import ActivityForm, GuestForm
 from .models import Activity, Guest
-from .tables import ActivityTable
+from .tables import ActivityTable, GuestTable
 
 
 class ActivityCreateView(LoginRequiredMixin, CreateView):
     model = Activity
     form_class = ActivityForm
-    success_url = reverse_lazy('activity:activity_list')
+
+    def get_success_url(self, **kwargs):
+        return reverse_lazy('activity:activity_detail', kwargs={"pk": self.kwargs["pk"]})
 
 
 class ActivityListView(LoginRequiredMixin, SingleTableView):
@@ -34,11 +37,22 @@ class ActivityDetailView(LoginRequiredMixin, DetailView):
     model = Activity
     context_object_name = "activity"
 
+    def get_context_data(self, **kwargs):
+        ctx = super().get_context_data()
+
+        table = GuestTable(data=Guest.objects.filter(activity=self.object)
+                           .filter(PermissionBackend.filter_queryset(self.request.user, Guest, "view")))
+        ctx["guests"] = table
+
+        return ctx
+
 
 class ActivityUpdateView(LoginRequiredMixin, UpdateView):
     model = Activity
     form_class = ActivityForm
-    success_url = reverse_lazy('activity:activity_list')
+
+    def get_success_url(self, **kwargs):
+        return reverse_lazy('activity:activity_detail', kwargs={"pk": self.kwargs["pk"]})
 
 
 class ActivityInviteView(LoginRequiredMixin, CreateView):