26 lines
895 B
Python
26 lines
895 B
Python
"""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"""
|