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

added filtering by country code

Browse Source
This commit is contained in:
Anatoly 2020-02-04 10:28:03 +03:00
parent 87686abbb7
commit 7cbfeb9845
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/collection/models.py
View File

@ -170,7 +170,7 @@ class GuideQuerySet(models.QuerySet):
def with_base_related(self): def with_base_related(self):
"""Return QuerySet with related.""" """Return QuerySet with related."""
return self.select_related('site', ) return self.select_related('site', 'guidefilter',)
def with_extended_related(self): def with_extended_related(self):
"""Return QuerySet with extended related.""" """Return QuerySet with extended related."""

8
apps/collection/views/back.py
View File

@ -30,10 +30,10 @@ class GuideBaseView(generics.GenericAPIView):
def get_queryset(self): def get_queryset(self):
"""An overridden get_queryset method.""" """An overridden get_queryset method."""
return ( queryset = models.Guide.objects.with_extended_related
models.Guide.objects.with_extended_related() if hasattr(self.request, 'country_code'):
# .annotate_counters() return queryset().by_country_code(self.request.country_code)
) return queryset()
class GuideFilterBaseView(generics.GenericAPIView): class GuideFilterBaseView(generics.GenericAPIView):