From 97581679d0ba670b3244408cc9c4062482314bd2 Mon Sep 17 00:00:00 2001 From: Rohan Rangray Date: Mon, 8 Jun 2015 04:01:09 -0700 Subject: [PATCH] Fix the server error message upon login --- flashcards/views.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/flashcards/views.py b/flashcards/views.py index 8ed461c..ea935b5 100644 --- a/flashcards/views.py +++ b/flashcards/views.py @@ -118,7 +118,10 @@ class SectionViewSet(ReadOnlyModelViewSet): """ Gets the contents of a user's deck for a given section. """ - qs = request.user.get_deck(self.get_object()) + try: + qs = request.user.get_deck(self.get_object()) + except: + raise PermissionDenied("User not enrolled in that section") serializer = FlashcardSerializer(qs, context={'user': request.user}, many=True) log_event(request, str(self.get_object())) return Response(serializer.data) -- 1.9.1