Added coverage testing

This commit is contained in:
Tris Forster
2023-03-01 13:39:56 +11:00
parent 4268b66b27
commit 6e1b26440f
4 changed files with 16 additions and 22 deletions
+6 -2
View File
@@ -167,16 +167,20 @@ from rest_framework import generics
class CollectionExportView(AuthorizedResourceMixin, generics.RetrieveAPIView):
serializer_class = CollectionSerializer
lookup_url_kwarg = 'collection'
def get_queryset(self):
if self.request.user.is_superuser:
return Collection.objects.all()
return Collection.objects.filter(administrators=self.request.user)
class WorkExportView(AuthorizedResourceMixin, generics.RetrieveAPIView):
serializer_class = WorkSerializer
def get_queryset(self):
return Work.objects.filter(collection__administrators=self.request.user, collection=self.kwargs['collection'])
works = Work.objects.filter(collection=self.kwargs['collection'])
if self.request.user.is_superuser:
return works
return works.filter(collection__administrators=self.request.user)
class WorkImportView(AuthorizedResourceMixin, generics.CreateAPIView):
serializer_class = WorkSerializer