diff --git a/apps/news/models.py b/apps/news/models.py index e3205c01..85b6b183 100644 --- a/apps/news/models.py +++ b/apps/news/models.py @@ -49,6 +49,12 @@ class NewsBanner(ProjectBaseMixin, TranslatedFieldsMixin): blank=True, null=True, default=None) +class NewsTypeQuerySet(models.QuerySet): + """QuerySet for model NewsType""" + def with_base_related(self): + return self.select_related('default_image') + + class NewsType(ProjectBaseMixin, TypeDefaultImageMixin): """NewsType model.""" @@ -61,6 +67,8 @@ class NewsType(ProjectBaseMixin, TypeDefaultImageMixin): blank=True, null=True, default=None, verbose_name='default image') + objects = NewsTypeQuerySet.as_manager() + class Meta: """Meta class.""" diff --git a/apps/news/views.py b/apps/news/views.py index 178c359f..bd88068d 100644 --- a/apps/news/views.py +++ b/apps/news/views.py @@ -87,9 +87,12 @@ class NewsTypeListView(generics.ListAPIView): pagination_class = None permission_classes = (permissions.AllowAny,) - queryset = models.NewsType.objects.all() serializer_class = serializers.NewsTypeSerializer + def get_queryset(self): + """Override get_queryset method.""" + return models.NewsType.objects.with_base_related() + class NewsBackOfficeMixinView: """News back office mixin view."""