.

2021-03-19 11:42:15 +00:00
parent ba8af9ad1e
commit fbd857431e
2 changed files with 17 additions and 8 deletions
@@ -307,16 +307,24 @@ class RapidCreateBase(LoginRequiredMixin, CreateView):
 # @login_required
 class RapidCreate(RapidCreateBase):

-    initial = {"laterality": Rapid.NONE}
+    #initial = {"laterality": Rapid.NONE}

    def get_initial(self):
-        # There has to be a better way...
-        try:
-            s = (i.pk for i in self.request.user.rapid_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(Exam, pk=self.kwargs["pk"])
+
+            initial['exam'] = exam.id
+
+            return initial
+
+        ## There has to be a better way...
+        #try:
+        #    s = (i.pk for i in self.request.user.rapid_default.site.all())
+        #    self.initial.update({"site": s})
+        #except AttributeError:
+        #    pass
+        #return self.initial

    #    fields = '__all__'
    #    #fields = [ 'condition' ]