diff --git a/apps/establishment/views/web.py b/apps/establishment/views/web.py index 53382be9..32decde0 100644 --- a/apps/establishment/views/web.py +++ b/apps/establishment/views/web.py @@ -29,12 +29,8 @@ class EstablishmentSimilarListView(EstablishmentListView): def get_queryset(self): """Override get_queryset method""" - qs = super().get_queryset() - if not qs.filter(pk=self.kwargs.get('pk')).exists(): - return qs.none() - return qs.similar(establishment_pk=self.kwargs.get('pk'))\ - .order_by('-total_mark')[:13] - + return super().get_queryset().similar(establishment_pk=self.kwargs.get('pk'))\ + .order_by('-total_mark')[:13] class EstablishmentRetrieveView(EstablishmentMixin, generics.RetrieveAPIView):