feature/fix_generate_pass
Andrey 8 years ago
parent afb70f10d9
commit ab7c842973
  1. 8
      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)

Loading…
Cancel
Save