start taking more advantage of class based views

2024-03-11 11:08:35 +00:00
parent 78e4ed871e
commit 0cb38f5c31
27 changed files with 244 additions and 104 deletions
@@ -2247,7 +2247,17 @@ class CaseCollectionAuthorUpdate(RevisionMixin, AuthorRequiredMixin, UpdateView)
    form_class = CaseCollectionAuthorForm
    template_name = "author_form.html"

+    def get_context_data(self, **kwargs):
+        context = super(CaseCollectionAuthorUpdate, self).get_context_data(**kwargs)
+        context["collection"] = context["object"]
+        return context
+
 class CaseAuthorUpdate(RevisionMixin, AuthorRequiredMixin, UpdateView):
    model = Case
    form_class = CaseAuthorForm
-    template_name = "author_form.html"
+    template_name = "author_form.html"
+
+    def get_context_data(self, **kwargs):
+        context = super(CaseAuthorUpdate, self).get_context_data(**kwargs)
+        context["collection"] = context["object"]
+        return context