Delete draft when submitting case study (#52)

2018-04-16 10:57:05 +10:00
parent 054668e1e8
commit a41df71793
2 changed files with 6 additions and 4 deletions
--- a/apps/map/views.py
+++ b/apps/map/views.py
@ -54,6 +54,12 @@ class BaseForm(LoginRequiredMixin, CreateView):
    def form_valid(self, form):
        self.object = form.save()
        self.send_email()
+
+        # Delete the corresponding draft
+        draft = CaseStudyDraft.objects.get(author=request.user)
+        if draft:
+            draft.delete()
+
        return super().form_valid(form)