from rest_framework import mixins, viewsets

from .serializers import *
from ..models import *


class ClipperEndpointViewSet(
    mixins.ListModelMixin,
    mixins.RetrieveModelMixin,
    viewsets.GenericViewSet
):
    serializer_class = ClipperEndpointSerializer
    queryset = ClipperEndpoint.objects.available()
    lookup_field = 'slug'


class WebClipViewSet(
    mixins.ListModelMixin,
    mixins.RetrieveModelMixin,
    mixins.CreateModelMixin,
    viewsets.GenericViewSet
):
    serializer_class = WebClipSerializer
    queryset = WebClip.objects.all()
    lookup_field = 'uuid'


__all__ = (
    'ClipperEndpointViewSet',
    'WebClipViewSet',
)