cgiacofei/TandoorRecipes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Include random in filtermixin

Browse Source
This commit is contained in:
Sebastian Markgraf
2020-11-14 20:11:40 +01:00
parent 4c92a4b39c
commit c4fbad614e
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
cookbook/views/api.py
View File

@ -104,8 +104,12 @@ class StandardFilterMixin(ViewSetMixin):
queryset = queryset.filter(name__icontains=query) queryset = queryset.filter(name__icontains=query)
limit = self.request.query_params.get('limit', None) limit = self.request.query_params.get('limit', None)
random = self.request.query_params.get('random', False)
if limit is not None: if limit is not None:
queryset = queryset[:int(limit)] if random:
queryset = queryset.random(int(limit))
else:
queryset = queryset[:int(limit)]
return queryset return queryset
@ -209,9 +213,6 @@ class RecipeViewSet(viewsets.ModelViewSet, StandardFilterMixin):
internal = self.request.query_params.get('internal', None) internal = self.request.query_params.get('internal', None)
if internal: if internal:
self.queryset = self.queryset.filter(internal=True) self.queryset = self.queryset.filter(internal=True)
random = self.request.query_params.get('random', False)
if random:
self.queryset = self.queryset.random(5)
return super().get_queryset() return super().get_queryset()