views.py

views.py

from rest_framework import viewsets
from semantic_version import SimpleSpec, Version

class GameViewSet(viewsets.ReadOnlyModelViewSet):
    """Game viewset."""

    queryset = Game.objects.all()
    serializer_class = GameSerializer

    def get_queryset(self):
        """Filter class queryset."""
        if self.action == "list":
            teeball_not_introduced = SimpleSpec(f"<{Game.TEEBALL_INTRODUCED_VERSION}")
            if teeball_not_introduced.match(Version(self.request.version)):
                self.queryset = self.queryset.exclude(type=Game.GameType.TEEBALL)

        return self.queryset