add filter by country for carusel
This commit is contained in:
parent
aab76d219f
commit
660eb347c5
|
|
@ -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."""
|
||||
|
|
|
|||
|
|
@ -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."""
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user