25 lines
791 B
Python
25 lines
791 B
Python
"""Views for app favorites."""
|
|
from rest_framework import generics
|
|
from establishment.models import Establishment
|
|
from establishment.serializers import EstablishmentBaseSerializer
|
|
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 = EstablishmentBaseSerializer
|
|
|
|
def get_queryset(self):
|
|
"""Override get_queryset method"""
|
|
return Establishment.objects.filter(favorites__user=self.request.user)\
|
|
.order_by('-favorites')
|