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