completed ch. 6

levelupapi/views/games.py

@@ -5,33 +5,43 @@ from rest_framework import serializers, status
 from levelupapi.models import Game
 
 class GameView(ViewSet):
-    """Level up game types view"""
+    """Level up game s view"""
 
     def retrieve(self, request, pk):
-        """Handle GET requests for single game type
+        """Handle GET requests for single game 
         
 
         Returns:
-            Response -- JSON serialized game type
+            Response -- JSON serialized game 
         """
-        game = Game.objects.get(pk=pk)
-        serializer = GameSerializer(game)
-        return Response(serializer.data)
+        
+
+# Add in the next 3 lines
+        try:
+            game = Game.objects.get(pk=pk)
+            serializer = GameSerializer(game)
+            return Response(serializer.data)
+        except Game.DoesNotExist as ex:
+            return Response({'message': ex.args[0]}, status=status.HTTP_404_NOT_FOUND)
         
 
     def list(self, request):
-        """Handle GET requests to get all game types
+        """Handle GET requests to get all game s
 
         Returns:
-            Response -- JSON serialized list of game types
+            Response -- JSON serialized list of game s
         """
-        game = Game.objects.all()
-        serializer = GameSerializer(game, many=True)
+        game_ = request.query_params.get('type', None)
+        games = Game.objects.all()
+        if game_ is not None:
+                games = games.filter(game_type_id=game_)
+        serializer = GameSerializer(games, many=True)
         return Response(serializer.data)
     
 class GameSerializer(serializers.ModelSerializer):
-    """JSON serializer for game types
+    """JSON serializer for game s
     """
     class Meta:
         model = Game
-        fields = ('id', 'title', 'maker', 'number_of_players', 'skill_level', 'game_type_id', 'gamer_id')
+        fields = ('id', 'title', 'maker', 'number_of_players', 'skill_level', 'game_type_id', 'gamer_id')
+        depth = 1