pre series form update

2022-07-24 11:11:58 +01:00
parent fde658c71c
commit 8f5c878bab
11 changed files with 286 additions and 91 deletions
@@ -645,13 +645,20 @@ class AtlasCreateBase(RevisionMixin, LoginRequiredMixin):
 # @login_required
 class AtlasCreate(AtlasCreateBase, CreateView):
    def get_initial(self):
-        # There has to be a better way...
-        try:
-            s = (i.pk for i in self.request.user.atlas_default.site.all())
-            self.initial.update({"site": s})
-        except AttributeError:
-            pass
-        return self.initial
+        if "pk" in self.kwargs:
+            initial = super().get_initial()
+            exam = get_object_or_404(CaseCollection, pk=self.kwargs["pk"])
+
+            initial["exams"] = [exam.id]
+
+            return initial
+        ## There has to be a better way...
+        #try:
+        #    s = (i.pk for i in self.request.user.atlas_default.site.all())
+        #    self.initial.update({"site": s})
+        #except AttributeError:
+        #    pass
+        #return self.initial


 class AtlasUpdate(