"""
This file contains the viewsets (based on the elsewhere defined data models and serializers)
"""

from rest_framework import viewsets
from .. import models
from .. import serializers


class TaskIORoleViewSet(viewsets.ModelViewSet):
    queryset = models.TaskIORole.objects.all()
    serializer_class = serializers.TaskIORoleSerializer


class TaskStateChoiceViewSet(viewsets.ModelViewSet):
    queryset = models.TaskStateChoice.objects.all()
    serializer_class = serializers.TaskStateChoiceSerializer


class TaskTypeChoiceViewSet(viewsets.ModelViewSet):
    queryset = models.TaskTypeChoice.objects.all()
    serializer_class = serializers.TaskTypeChoiceSerializer


class StationTypeChoiceViewSet(viewsets.ModelViewSet):
    queryset = models.StationTypeChoice.objects.all()
    serializer_class = serializers.StationTypeChoiceSerializer


class TaskTemplateViewSet(viewsets.ModelViewSet):
    queryset = models.TaskTemplate.objects.all()
    serializer_class = serializers.TaskTemplateSerializer


class DefaultTaskTemplateViewSet(viewsets.ModelViewSet):
    queryset = models.DefaultTaskTemplate.objects.all()
    serializer_class = serializers.DefaultTaskTemplateSerializer


class DataproductSpecificationTemplateViewSet(viewsets.ModelViewSet):
    queryset = models.DataproductSpecificationTemplate.objects.all()
    serializer_class = serializers.DataproductSpecificationTemplateSerializer


class DefaultDataproductSpecificationTemplateViewSet(viewsets.ModelViewSet):
    queryset = models.DefaultDataproductSpecificationTemplate.objects.all()
    serializer_class = serializers.DefaultDataproductSpecificationTemplateSerializer


class TaskViewSet(viewsets.ModelViewSet):
    queryset = models.Task.objects.all()
    serializer_class = serializers.TaskSerializer


class DataproductRelationViewSet(viewsets.ModelViewSet):
    queryset = models.DataproductRelation.objects.all()
    serializer_class = serializers.DataproductRelationSerializer


class DataproductViewSet(viewsets.ModelViewSet):
    queryset = models.Dataproduct.objects.all()
    serializer_class = serializers.DataproductSerializer


class AntennaSetViewSet(viewsets.ModelViewSet):
    queryset = models.AntennaSet.objects.all()
    serializer_class = serializers.AntennaSetSerializer