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

fix news in_fav

Browse Source
This commit is contained in:
alex 2019-11-14 17:07:31 +03:00
parent f4e762bb80
commit 322af9efef
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
apps/news/views.py
View File

@ -18,12 +18,13 @@ class NewsMixinView:
permission_classes = (permissions.AllowAny,) permission_classes = (permissions.AllowAny,)
serializer_class = serializers.NewsBaseSerializer serializer_class = serializers.NewsBaseSerializer
def get_queryset(self, *args, **kwargs): def get_queryset(self):
"""Override get_queryset method.""" """Override get_queryset method."""
qs = models.News.objects.published() \ qs = models.News.objects.published() \
.with_base_related() \ .with_base_related() \
.order_by('-is_highlighted', '-created') \ .annotate_in_favorites(self.request.user) \
.annotate_in_favorites(self.request.user) .order_by('-is_highlighted', '-created')
country_code = self.request.country_code country_code = self.request.country_code
if country_code: if country_code:
qs = qs.by_country_code(country_code) qs = qs.by_country_code(country_code)
@ -62,8 +63,13 @@ class NewsBackOfficeMixinView:
"""News back office mixin view.""" """News back office mixin view."""
permission_classes = (permissions.IsAuthenticated,) permission_classes = (permissions.IsAuthenticated,)
queryset = models.News.objects.with_base_related() \
def get_queryset(self):
"""Override get_queryset method."""
qs = models.News.objects.with_base_related() \
.annotate_in_favorites(self.request.user) \
.order_by('-is_highlighted', '-created') .order_by('-is_highlighted', '-created')
return qs
class NewsBackOfficeLCView(NewsBackOfficeMixinView, class NewsBackOfficeLCView(NewsBackOfficeMixinView,