fix news in_fav

This commit is contained in:
alex 2019-11-14 17:07:31 +03:00
parent f4e762bb80
commit 322af9efef

View File

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