from __future__ import absolute_import

from assessment.api.views import AssessmentViewset

from . import models, serializers
from utils.api import CleanupFieldsBaseViewSet


class IVChemical(AssessmentViewset):
    assessment_filter_args = "assessment"
    model = models.IVChemical
    serializer_class = serializers.IVChemicalSerializer


class IVCellType(AssessmentViewset):
    assessment_filter_args = "assessment"
    model = models.IVCellType
    serializer_class = serializers.IVCellTypeSerializer


class IVExperiment(AssessmentViewset):
    assessment_filter_args = "assessment"
    model = models.IVExperiment
    serializer_class = serializers.IVExperimentSerializerFull


class IVEndpoint(AssessmentViewset):
    assessment_filter_args = "assessment"
    model = models.IVEndpoint
    serializer_class = serializers.IVEndpointSerializer


class Cleanup(CleanupFieldsBaseViewSet):
    serializer_class = serializers.CleanupFieldsSerializer
    model = models.IVEndpoint
