23 lines
729 B
Python
23 lines
729 B
Python
"""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)
|