Commit 3709ee64556a8c5b57de8ed131d90832f2a75163

Authored by Laura Hawkins
1 parent 366764be3b
Exists in master

working on deck view

Showing 2 changed files with 13 additions and 5 deletions Side-by-side Diff

flashcards/tests/test_api.py View file @ 3709ee6
... ... @@ -253,4 +253,8 @@
253 253 def test_section_search(self):
254 254 response = self.client.get('/api/sections/search/?q=Kramer')
255 255 self.assertEqual(response.status_code, HTTP_200_OK)
  256 +
  257 + def test_section_deck(self):
  258 + response = self.client.get('/api/sections/deck/')
  259 + self.assertEqual(response.status_code, HTTP_200_OK)
flashcards/views.py View file @ 3709ee6
... ... @@ -80,14 +80,18 @@
80 80 qs = Section.search(query.split(' '))[:8]
81 81 serializer = SectionSerializer(qs, many=True)
82 82 return Response(serializer.data)
83   -"""
  83 +
84 84 @list_route(methods=['get'], permission_classes=[IsAuthenticated])
85 85 def deck(self, request):
86   - query = request.GET.get('q', None)
  86 + """
  87 + Gets the contents of a user's deck for a given section.
  88 + """
  89 + query = request.GET.get(Flashcard.objects.all(), None)
87 90 if not query: return Response('[]')
88   - qs = Section.search(query.split(' '))[:8]
89   - serializer = SectionSerializer(qs, many=True)
90   - return Response(serializer.data)"""
  91 + qs = Flashcard.objects.all()
  92 + qs = qs.filter(userflashcard__user=request.user)
  93 + serializer = FlashcardSerializer(qs, many=True)
  94 + return Response(serializer.data)
91 95  
92 96  
93 97 class UserSectionListView(ListAPIView):