"""Views for app favorites."""
from rest_framework import generics

from establishment.models import Establishment
from establishment.serializers import EstablishmentListSerializer
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(generics.ListAPIView):
    """List views for favorites"""
    serializer_class = EstablishmentListSerializer

    def get_queryset(self):
        """Override get_queryset method"""
        return Establishment.objects.filter(favorites__user=self.request.user)\
                                    .order_by('-favorites')