From 85d6cfc1733fa584d30ebbad0f5fadd2bbfa79a9 Mon Sep 17 00:00:00 2001 From: Ivlev Denis Date: Wed, 14 Feb 2018 17:07:11 +0300 Subject: [PATCH] Auto create draft on open course edit view without ID --- apps/course/views.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/apps/course/views.py b/apps/course/views.py index 01bdfd09..75ff396a 100644 --- a/apps/course/views.py +++ b/apps/course/views.py @@ -146,8 +146,11 @@ class CourseEditView(TemplateView): template_name = 'course/course_edit.html' def get(self, request, pk=None): + drafts = Course.objects.filter(author=request.user, status=Course.DRAFT) if pk: self.object = get_object_or_404(Course, pk=pk) + elif drafts.exists(): + self.object = drafts.last() else: self.object = None return super().get(request)