added filter to FavoritesEstablishmentListView

This commit is contained in:
Anatoly 2019-10-29 15:23:26 +03:00
parent 251edda2ea
commit 13eaa94261

View File

@ -1,6 +1,7 @@
"""Views for app favorites."""
from rest_framework import generics
from establishment.models import Establishment
from establishment.filters import EstablishmentFilter
from establishment.serializers import EstablishmentBaseSerializer
from .models import Favorites
@ -17,6 +18,7 @@ class FavoritesEstablishmentListView(generics.ListAPIView):
"""List views for favorites"""
serializer_class = EstablishmentBaseSerializer
filter_class = EstablishmentFilter
def get_queryset(self):
"""Override get_queryset method"""