"""Views for app favorites."""
from rest_framework import generics
from .serializers import FavoritesEstablishmentListSerializer
from .models import Favorites


class FavoritesBaseView(generics.GenericAPIView):
    """Base view for Favorites."""
    def get_queryset(self):
        """Override get_queryset method."""
        return Favorites.objects.by_user(self.request.user)


class FavoritesEstablishmentListView(FavoritesBaseView, generics.ListAPIView):
    """List views for favorites"""
    serializer_class = FavoritesEstablishmentListSerializer

    def get_queryset(self):
        """Override get_queryset method"""
        return super().get_queryset().by_content_type(app_label='establishment',
                                                      model='establishment')


class FavoritesDestroyView(FavoritesBaseView, generics.DestroyAPIView):
    """Destroy view for favorites"""