Add klass argument to get_object_or_404 usage
Signed-off-by: Roberto Rosario <Roberto.Rosario@mayan-edms.com>
This commit is contained in:
@@ -41,7 +41,7 @@ class APIDocumentTypeWorkflowListView(generics.ListAPIView):
|
||||
serializer_class = WorkflowSerializer
|
||||
|
||||
def get_document_type(self):
|
||||
document_type = get_object_or_404(DocumentType, pk=self.kwargs['pk'])
|
||||
document_type = get_object_or_404(klass=DocumentType, pk=self.kwargs['pk'])
|
||||
|
||||
AccessControlList.objects.check_access(
|
||||
permissions=permission_document_type_view, user=self.request.user,
|
||||
@@ -106,7 +106,7 @@ class APIWorkflowDocumentTypeList(generics.ListCreateAPIView):
|
||||
else:
|
||||
permission_required = permission_workflow_edit
|
||||
|
||||
workflow = get_object_or_404(Workflow, pk=self.kwargs['pk'])
|
||||
workflow = get_object_or_404(klass=Workflow, pk=self.kwargs['pk'])
|
||||
|
||||
AccessControlList.objects.check_access(
|
||||
permissions=permission_required, user=self.request.user,
|
||||
@@ -159,7 +159,7 @@ class APIWorkflowDocumentTypeView(generics.RetrieveDestroyAPIView):
|
||||
else:
|
||||
permission_required = permission_workflow_edit
|
||||
|
||||
workflow = get_object_or_404(Workflow, pk=self.kwargs['pk'])
|
||||
workflow = get_object_or_404(klass=Workflow, pk=self.kwargs['pk'])
|
||||
|
||||
AccessControlList.objects.check_access(
|
||||
permissions=permission_required, user=self.request.user,
|
||||
@@ -297,7 +297,7 @@ class APIWorkflowStateListView(generics.ListCreateAPIView):
|
||||
else:
|
||||
permission_required = permission_workflow_edit
|
||||
|
||||
workflow = get_object_or_404(Workflow, pk=self.kwargs['pk'])
|
||||
workflow = get_object_or_404(klass=Workflow, pk=self.kwargs['pk'])
|
||||
|
||||
AccessControlList.objects.check_access(
|
||||
permissions=permission_required, user=self.request.user,
|
||||
@@ -340,7 +340,7 @@ class APIWorkflowStateView(generics.RetrieveUpdateDestroyAPIView):
|
||||
else:
|
||||
permission_required = permission_workflow_edit
|
||||
|
||||
workflow = get_object_or_404(Workflow, pk=self.kwargs['pk'])
|
||||
workflow = get_object_or_404(klass=Workflow, pk=self.kwargs['pk'])
|
||||
|
||||
AccessControlList.objects.check_access(
|
||||
permissions=permission_required, user=self.request.user,
|
||||
@@ -393,7 +393,7 @@ class APIWorkflowTransitionListView(generics.ListCreateAPIView):
|
||||
else:
|
||||
permission_required = permission_workflow_edit
|
||||
|
||||
workflow = get_object_or_404(Workflow, pk=self.kwargs['pk'])
|
||||
workflow = get_object_or_404(klass=Workflow, pk=self.kwargs['pk'])
|
||||
|
||||
AccessControlList.objects.check_access(
|
||||
permissions=permission_required, user=self.request.user,
|
||||
@@ -447,7 +447,7 @@ class APIWorkflowTransitionView(generics.RetrieveUpdateDestroyAPIView):
|
||||
else:
|
||||
permission_required = permission_workflow_edit
|
||||
|
||||
workflow = get_object_or_404(Workflow, pk=self.kwargs['pk'])
|
||||
workflow = get_object_or_404(klass=Workflow, pk=self.kwargs['pk'])
|
||||
|
||||
AccessControlList.objects.check_access(
|
||||
permissions=permission_required, user=self.request.user,
|
||||
@@ -471,7 +471,7 @@ class APIWorkflowInstanceListView(generics.ListAPIView):
|
||||
}
|
||||
|
||||
def get_document(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_workflow_view, user=self.request.user,
|
||||
@@ -496,7 +496,7 @@ class APIWorkflowInstanceView(generics.RetrieveAPIView):
|
||||
serializer_class = WorkflowInstanceSerializer
|
||||
|
||||
def get_document(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_workflow_view, user=self.request.user,
|
||||
@@ -515,7 +515,7 @@ class APIWorkflowInstanceLogEntryListView(generics.ListCreateAPIView):
|
||||
post: Transition a document workflow by creating a new document workflow log entry.
|
||||
"""
|
||||
def get_document(self):
|
||||
document = get_object_or_404(Document, pk=self.kwargs['pk'])
|
||||
document = get_object_or_404(klass=Document, pk=self.kwargs['pk'])
|
||||
|
||||
if self.request.method == 'GET':
|
||||
"""
|
||||
@@ -562,7 +562,7 @@ class APIWorkflowInstanceLogEntryListView(generics.ListCreateAPIView):
|
||||
|
||||
def get_workflow_instance(self):
|
||||
workflow = get_object_or_404(
|
||||
self.get_document().workflows, pk=self.kwargs['workflow_pk']
|
||||
klass=self.get_document().workflows, pk=self.kwargs['workflow_pk']
|
||||
)
|
||||
|
||||
return workflow
|
||||
|
||||
Reference in New Issue
Block a user