From 07d17d5bf1103eb50ef4edb523de85691210abec Mon Sep 17 00:00:00 2001 From: Anatoly Date: Fri, 20 Sep 2019 16:55:04 +0300 Subject: [PATCH] refactored similar establishment endpoint --- apps/establishment/views/web.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) 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):