From 5f6e92ae30413ba420d54a17c8f5ba1b87ef49b1 Mon Sep 17 00:00:00 2001 From: Ross Date: Thu, 29 Apr 2021 23:39:05 +0100 Subject: [PATCH] . --- anatomy/views.py | 8 ++++++++ rapids/views.py | 8 ++++++++ 2 files changed, 16 insertions(+) diff --git a/anatomy/views.py b/anatomy/views.py index a1b1c7a0..9000e3e2 100644 --- a/anatomy/views.py +++ b/anatomy/views.py @@ -993,10 +993,18 @@ class ExamCreate(LoginRequiredMixin, CreateView): model = Exam form_class = ExamForm + def form_valid(self, form): + form.instance.author.add(self.request.user.id) + return super().form_valid(form) + class ExamUpdate(LoginRequiredMixin, AuthorOrCheckerRequiredMixin, UpdateView): model = Exam form_class = ExamForm + def form_valid(self, form): + form.instance.author.add(self.request.user.id) + return super().form_valid(form) + class ExamViewSet(viewsets.ModelViewSet): #queryset = Exam.objects.all().order_by('name') serializer_class = ExamSerializer diff --git a/rapids/views.py b/rapids/views.py index db27a9ae..37f7d899 100755 --- a/rapids/views.py +++ b/rapids/views.py @@ -932,10 +932,18 @@ class ExamCreate(LoginRequiredMixin, CreateView): model = Exam form_class = ExamForm + def form_valid(self, form): + form.instance.author.add(self.request.user.id) + return super().form_valid(form) + class ExamUpdate(LoginRequiredMixin, AuthorOrCheckerRequiredMixin, UpdateView): model = Exam form_class = ExamForm + def form_valid(self, form): + form.instance.author.add(self.request.user.id) + return super().form_valid(form) + class RapidUserAnswerView(LoginRequiredMixin, SingleTableMixin, FilterView): model = CidUserAnswer table_class = RapidUserAnswerTable