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.renderers import JSONRenderer
from rest_framework.response import Response from rest_framework.response import Response
from access.models import Progress
from maps.serializers import CourseMapSerializer from maps.serializers import CourseMapSerializer
from courses.models import Course, Vertex from courses.models import Course, Vertex
from courses.serializers import CourseDetailSerializer, CourseListSerializer, VertexSerializer from courses.serializers import CourseDetailSerializer, CourseListSerializer, VertexSerializer
@ -25,8 +26,11 @@ class TreeView(APIView):
return Response("Course doesn't exist", status=404) return Response("Course doesn't exist", status=404)
if request.user.is_authenticated(): if request.user.is_authenticated():
route = course.progress_set.get(user=request.user).get_template() try:
return Response(CourseMapSerializer(route.get_first()).data, self.status_code) 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) return Response(CourseMapSerializer(course.route.get_first()).data, self.status_code)

Loading…
Cancel
Save