Coverage for timelaps/views.py : 88%

# -*- coding: utf-8 -*- 

from __future__ import unicode_literals 

from django.shortcuts import render 

from rest_framework import generics 

from .models import * 

from .serializers import * 

# Create your views here. 

class ListDriverView(generics.ListAPIView): 

""" 

Provides a get method handler. 

""" 

queryset = Driver.objects.all() 

serializer_class = DriverSerializer 

class ListRaceView(generics.ListAPIView): 

"""  

Provides a get method handler. 

""" 

queryset = Race.objects.all() 

serializer_class = RaceSerializer 

class ListRelayView(generics.ListAPIView): 

"""  

Provides a get method handler. 

""" 

queryset = Relay.objects.all() 

serializer_class = RelaySerializer 

class ListRulesView(generics.ListAPIView): 

"""  

Provides a get method handler. 

""" 

queryset = Rules.objects.all() 

serializer_class = RulesSerializer 

class ListTeamView(generics.ListAPIView): 

"""  

Provides a get method handler. 

""" 

queryset = Team.objects.all() 

serializer_class = TeamSerializer 

class ListTeamPilotView(generics.ListAPIView): 

"""  

Provides a get method handler. 

""" 

queryset = TeamPilot.objects.all() 

serializer_class = TeamPilotSerializer 

class ListRelaysByRaceView(generics.ListAPIView): 

''' 

Getting the params from the url with the "self.kwargs.get"  

and filter with thoses in the relays relation 

''' 

serializer_class = RelayPilotSerializer 

lookup_url_raceid = "raceid" 

lookup_url_teamid = "teamid" 

def get_queryset(self): 

raceid = self.kwargs.get(self.lookup_url_raceid) 

teamid = self.kwargs.get(self.lookup_url_teamid) 

relays = Relay.objects.filter(team_pilot__team=teamid).filter(team_pilot__race=raceid) 

return relays