add filter by country for carusel

This commit is contained in:
Dmitriy Kuzmenko 2019-11-01 18:10:12 +03:00
parent aab76d219f
commit 660eb347c5
2 changed files with 11 additions and 0 deletions

View File

@ -286,6 +286,10 @@ class CarouselQuerySet(models.QuerySet):
"""Active carousel objects."""
return self.filter(active=True)
def by_country_code(self, code):
"""Filter collection by country code."""
return self.filter(country__code=code)
class Carousel(models.Model):
"""Carousel model."""

View File

@ -67,6 +67,13 @@ class CarouselListView(generics.ListAPIView):
permission_classes = (permissions.AllowAny,)
pagination_class = None
def get_queryset(self):
country_code = self.request.country_code
qs = self.queryset
if country_code:
qs = qs.by_country_code(country_code)
return qs
class DetermineLocation(generics.GenericAPIView):
"""Determine user's location."""