views.py

"""
Defines the views for the API
"""

# from django.shortcuts import render
from rest_framework import viewsets
from .serializers import SportListSerializer, CriterionListSerializer
from .models import Sport, Criterion

# Create your views here.


class SportListView(viewsets.ModelViewSet):  # pylint: disable=too-many-ancestors
    """
    A View returning every Sport Object
    """

    serializer_class = SportListSerializer
    queryset = Sport.objects.all()


class CriterionListView(viewsets.ModelViewSet):  # pylint: disable=too-many-ancestors
    """
    A View returning every Criterion Object
    """

    serializer_class = CriterionListSerializer
    queryset = Criterion.objects.all()