mirrors/recipes
1
0
Fork 0
mirror of https://github.com/TandoorRecipes/recipes.git synced 2026-01-01 12:18:45 -05:00
Code Issues Packages Projects Releases Wiki Activity

added test for cook log

Browse Source
This commit is contained in:
vabene1111
2021-03-10 21:25:18 +01:00
parent 528f329ebb
commit e7f8d58a7d
6 changed files with 166 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
cookbook/views/api.py
View File

@@ -382,7 +382,7 @@ class ViewLogViewSet(viewsets.ModelViewSet):
permission_classes = [CustomIsOwner]
def get_queryset(self):
return CookLog.objects.filter(created_by=self.request.user).filter(space=self.request.user.userpreference.space).all()[:5]
return self.queryset.filter(created_by=self.request.user).filter(space=self.request.user.userpreference.space).all()[:5]
class CookLogViewSet(viewsets.ModelViewSet):
@@ -391,8 +391,11 @@ class CookLogViewSet(viewsets.ModelViewSet):
permission_classes = [CustomIsOwner]
def get_queryset(self):
queryset = CookLog.objects.filter(created_by=self.request.user).filter(space=self.request.user.userpreference.space).all()[:5]
return queryset
self.queryset = self.queryset.filter(created_by=self.request.user).filter(space=self.request.user.userpreference.space).all()
if self.request.method == 'GET':
return self.queryset[:5]
else:
return self.queryset
# -------------- non django rest api views --------------------