phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add filter by country for carusel

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/main/models.py
View File

@ -286,6 +286,10 @@ class CarouselQuerySet(models.QuerySet):
"""Active carousel objects.""" """Active carousel objects."""
return self.filter(active=True) 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): class Carousel(models.Model):
"""Carousel model.""" """Carousel model."""

7
apps/main/views/common.py
View File

@ -67,6 +67,13 @@ class CarouselListView(generics.ListAPIView):
permission_classes = (permissions.AllowAny,) permission_classes = (permissions.AllowAny,)
pagination_class = None 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): class DetermineLocation(generics.GenericAPIView):
"""Determine user's location.""" """Determine user's location."""