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

pytest shopping list from recipe methods

Browse Source
This commit is contained in:
smilerz
2021-12-12 10:39:01 -06:00
parent ab968f225b
commit 262387da3e
6 changed files with 225 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
cookbook/views/api.py
View File

@ -412,7 +412,8 @@ class FoodViewSet(viewsets.ModelViewSet, TreeMixin):
permission_classes = [CustomIsUser]
pagination_class = DefaultPagination
@decorators.action(detail=True, methods=['PUT'], serializer_class=FoodShoppingUpdateSerializer,)
''
@decorators.action(detail=True, methods=['PUT'], serializer_class=FoodShoppingUpdateSerializer,)
def shopping(self, request, pk):
obj = self.get_object()
shared_users = list(self.request.user.get_shopping_share())
@ -594,6 +595,11 @@ class RecipeViewSet(viewsets.ModelViewSet):
schema = QueryParamAutoSchema()
def get_queryset(self):
if self.detail:
self.queryset = self.queryset.filter(space=self.request.space)
return super().get_queryset()
share = self.request.query_params.get('share', None)
if not (share and self.detail):
self.queryset = self.queryset.filter(space=self.request.space)