start anatomy creation forms

anatomy/views.py

@@ -6,14 +6,22 @@ from django import forms
 from django.contrib.auth.decorators import login_required, user_passes_test
 from django.contrib.auth.models import User
 
+from django.contrib.auth.mixins import LoginRequiredMixin
+
+from django.views.generic.edit import CreateView, UpdateView, DeleteView
+from django.views.generic import ListView
+
 from django.db.models.functions import Lower
 
 from django.core.cache import cache
 
+from django.urls import reverse_lazy, reverse
+
 from django.http import Http404, JsonResponse
 
+from django.http import HttpResponseRedirect, HttpResponse
 
-from .forms import AnatomyAnswerForm, MarkAnatomyQuestionForm
+from .forms import AnatomyAnswerForm, AnswerFormSet, MarkAnatomyQuestionForm, AnatomyQuestionForm
 from .models import (
     AnatomyQuestion,
     CidUserAnswer,
@@ -748,4 +756,107 @@ def cid_scores(request, pk):
             "exams": exams,
             "cid": cid,
         },
-    )
+    )
+
+class AnatomyQuestionCreateBase(LoginRequiredMixin, CreateView):
+    model = AnatomyQuestion
+    form_class = AnatomyQuestionForm
+
+    def get_context_data(self, **kwargs):
+        context = super(AnatomyQuestionCreateBase, self).get_context_data(**kwargs)
+        if self.request.POST:
+            context['answer_formset'] = AnswerFormSet(self.request.POST,
+                                                    self.request.FILES)
+        else:
+            context['answer_formset'] = AnswerFormSet()
+        return context
+
+    def form_valid(self, form):
+
+        self.object = form.save(commit=False)
+        self.object.save()
+
+        form.instance.author.add(self.request.user.id)
+
+        context = self.get_context_data(form=form)
+        formset = context['answer_formset']
+
+        if formset.is_valid():
+            response = super().form_valid(form)
+            formset.instance = self.object
+            formset.save()
+            # If the normal submit button is pressed we save as normal
+            if "submit" in self.request.POST:
+                return response
+            # else we redirect to the clone url
+            else:
+                return redirect('AnatomyQuestions:AnatomyQuestion_clone', pk=self.object.pk)
+
+        else:
+            return super().form_invalid(form)
+
+
+#@login_required
+class AnatomyQuestionCreate(AnatomyQuestionCreateBase):
+
+    #initial = {'laterality': AnatomyQuestion.NONE}
+
+    def get_initial(self):
+        pass
+#        # There has to be a better way...
+#        try:
+#            s = (i.pk for i in self.request.user.AnatomyQuestion_default.site.all())
+#            self.initial.update({'site': s})
+#        except AttributeError:
+#            pass
+#        return self.initial
+#
+    #    fields = '__all__'
+    #    #fields = [ 'condition' ]
+    #    #initial = {'date_of_death': '05/01/2018'}
+    #    exclude = [ 'created_date', 'published_date' ]
+
+    #self.object = form.save(commit=False)
+    #self.object.save()
+
+    #form.instance.author.add(self.request.user.id)
+    #return super().form_valid(form)
+
+
+#class AnatomyQuestionUpdate(LoginRequiredMixin, AuthorOrCheckerRequiredMixin,
+class AnatomyQuestionUpdate(LoginRequiredMixin, UpdateView):
+    model = AnatomyQuestion
+    form_class = AnatomyQuestionForm
+
+    #    fields = '__all__'
+    #    #fields = [ 'condition' ]
+    #    #initial = {'date_of_death': '05/01/2018'}
+    #    exclude = [ 'created_date', 'published_date' ]
+    def get_context_data(self, **kwargs):
+        context = super(AnatomyQuestionUpdate, self).get_context_data(**kwargs)
+        if self.request.POST:
+            context['answer_formset'] = AnswerFormSet(self.request.POST,
+                                                    self.request.FILES,
+                                                    instance=self.object)
+            context['answer_formset'].full_clean()
+        else:
+            context['answer_formset'] = AnswerFormSet(instance=self.object)
+        return context
+
+    def form_valid(self, form):
+
+        self.object = form.save(commit=False)
+        self.object.save()
+
+        form.instance.author.add(self.request.user.id)
+
+        context = self.get_context_data(form=form)
+        formset = context['answer_formset']
+        #logger.debug(formset.is_valid())
+        if formset.is_valid():
+            response = super().form_valid(form)
+            formset.instance = self.object
+            formset.save()
+            return response
+        else:
+            return super().form_invalid(form)