From ab7c8429737c6a914207d7912812e5a2b3bbe97d Mon Sep 17 00:00:00 2001 From: Andrey Date: Mon, 4 Dec 2017 12:28:48 +0300 Subject: [PATCH] create map --- courses/views.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/courses/views.py b/courses/views.py index e9638bc..3c22d07 100644 --- a/courses/views.py +++ b/courses/views.py @@ -2,6 +2,7 @@ from rest_framework.views import APIView from rest_framework.renderers import JSONRenderer from rest_framework.response import Response +from access.models import Progress from maps.serializers import CourseMapSerializer from courses.models import Course, Vertex from courses.serializers import CourseDetailSerializer, CourseListSerializer, VertexSerializer @@ -25,8 +26,11 @@ class TreeView(APIView): return Response("Course doesn't exist", status=404) if request.user.is_authenticated(): - route = course.progress_set.get(user=request.user).get_template() - return Response(CourseMapSerializer(route.get_first()).data, self.status_code) + try: + route = course.progress_set.get(user=request.user).get_template() + return Response(CourseMapSerializer(route.get_first()).data, self.status_code) + except Progress.DoesNotExist: + pass return Response(CourseMapSerializer(course.route.get_first()).data, self.status_code)