Add klass argument to get_object_or_404 usage

Signed-off-by: Roberto Rosario <Roberto.Rosario@mayan-edms.com>
This commit is contained in:
Roberto Rosario
2018-12-16 01:55:33 -04:00
parent 727d2ecd71
commit 798446f362
39 changed files with 164 additions and 164 deletions

View File

@@ -62,7 +62,7 @@ class APIIndexNodeInstanceDocumentListView(generics.ListAPIView):
def get_queryset(self):
index_node_instance = get_object_or_404(
IndexInstanceNode, pk=self.kwargs['pk']
klass=IndexInstanceNode, pk=self.kwargs['pk']
)
AccessControlList.objects.check_access(
permissions=permission_document_indexing_view,
@@ -109,7 +109,7 @@ class APIDocumentIndexListView(generics.ListAPIView):
serializer_class = IndexInstanceNodeSerializer
def get_queryset(self):
document = get_object_or_404(Document, pk=self.kwargs['pk'])
document = get_object_or_404(klass=Document, pk=self.kwargs['pk'])
AccessControlList.objects.check_access(
permissions=permission_document_view, user=self.request.user,
obj=document

View File

@@ -119,7 +119,7 @@ class SetupIndexDocumentTypesView(AssignRemoveView):
}
def get_object(self):
return get_object_or_404(Index, pk=self.kwargs['pk'])
return get_object_or_404(klass=Index, pk=self.kwargs['pk'])
def left_list(self):
return AssignRemoveView.generate_choices(
@@ -149,7 +149,7 @@ class SetupIndexTreeTemplateListView(SingleObjectListView):
}
def get_index(self):
return get_object_or_404(Index, pk=self.kwargs['pk'])
return get_object_or_404(klass=Index, pk=self.kwargs['pk'])
def get_object_list(self):
return self.get_index().template_root.get_descendants(
@@ -185,7 +185,7 @@ class TemplateNodeCreateView(SingleObjectCreateView):
}
def get_parent_node(self):
return get_object_or_404(IndexTemplateNode, pk=self.kwargs['pk'])
return get_object_or_404(klass=IndexTemplateNode, pk=self.kwargs['pk'])
class TemplateNodeDeleteView(SingleObjectDeleteView):
@@ -260,7 +260,7 @@ class IndexInstanceNodeView(DocumentListView):
def dispatch(self, request, *args, **kwargs):
self.index_instance_node = get_object_or_404(
IndexInstanceNode, pk=self.kwargs['pk']
klass=IndexInstanceNode, pk=self.kwargs['pk']
)
AccessControlList.objects.check_access(
@@ -340,7 +340,7 @@ class DocumentIndexNodeListView(SingleObjectListView):
).dispatch(request, *args, **kwargs)
def get_document(self):
return get_object_or_404(Document, pk=self.kwargs['pk'])
return get_object_or_404(klass=Document, pk=self.kwargs['pk'])
def get_extra_context(self):
return {