Add klass argument to get_object_or_404 usage

Signed-off-by: Roberto Rosario <Roberto.Rosario@mayan-edms.com>

mayan/apps/documents/views/document_page_views.py

@@ -49,7 +49,7 @@ class DocumentPageInteractiveTransformation(RedirectView):
         )
 
     def get_object(self):
-        return get_object_or_404(DocumentPage, pk=self.kwargs['pk'])
+        return get_object_or_404(klass=DocumentPage, pk=self.kwargs['pk'])
 
     def get_redirect_url(self, *args, **kwargs):
         url = reverse(
@@ -79,7 +79,7 @@ class DocumentPageListView(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 {
@@ -107,7 +107,7 @@ class DocumentPageNavigationBase(RedirectView):
         )
 
     def get_object(self):
-        return get_object_or_404(DocumentPage, pk=self.kwargs['pk'])
+        return get_object_or_404(klass=DocumentPage, pk=self.kwargs['pk'])
 
     def get_redirect_url(self, *args, **kwargs):
         parse_result = urlparse(
@@ -251,7 +251,7 @@ class DocumentPageView(SimpleView):
         }
 
     def get_object(self):
-        return get_object_or_404(DocumentPage, pk=self.kwargs['pk'])
+        return get_object_or_404(klass=DocumentPage, pk=self.kwargs['pk'])
 
 
 class DocumentPageViewResetView(RedirectView):

mayan/apps/documents/views/document_type_views.py

@@ -71,7 +71,7 @@ class DocumentTypeDeleteView(SingleObjectDeleteView):
 
 class DocumentTypeDocumentListView(DocumentListView):
     def get_document_type(self):
-        return get_object_or_404(DocumentType, pk=self.kwargs['pk'])
+        return get_object_or_404(klass=DocumentType, pk=self.kwargs['pk'])
 
     def get_document_queryset(self):
         return self.get_document_type().documents.all()
@@ -145,7 +145,7 @@ class DocumentTypeFilenameCreateView(SingleObjectCreateView):
         )
 
     def get_document_type(self):
-        return get_object_or_404(DocumentType, pk=self.kwargs['pk'])
+        return get_object_or_404(klass=DocumentType, pk=self.kwargs['pk'])
 
     def get_extra_context(self):
         return {
@@ -218,7 +218,7 @@ class DocumentTypeFilenameListView(SingleObjectListView):
     object_permission = permission_document_type_view
 
     def get_document_type(self):
-        return get_object_or_404(DocumentType, pk=self.kwargs['pk'])
+        return get_object_or_404(klass=DocumentType, pk=self.kwargs['pk'])
 
     def get_extra_context(self):
         return {

mayan/apps/documents/views/document_version_views.py

@@ -38,7 +38,7 @@ class DocumentVersionListView(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 {
@@ -66,7 +66,7 @@ class DocumentVersionRevertView(ConfirmView):
         }
 
     def get_object(self):
-        return get_object_or_404(DocumentVersion, pk=self.kwargs['pk'])
+        return get_object_or_404(klass=DocumentVersion, pk=self.kwargs['pk'])
 
     def view_action(self):
         try:

mayan/apps/documents/views/document_views.py

@@ -238,7 +238,7 @@ class DocumentDuplicatesListView(DocumentListView):
         ).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):
         context = super(DocumentDuplicatesListView, self).get_extra_context()
@@ -327,7 +327,7 @@ class DocumentRestoreView(ConfirmView):
 
     def object_action(self, instance):
         source_document = get_object_or_404(
-            Document.passthrough, pk=instance.pk
+            klass=Document.passthrough, pk=instance.pk
         )
 
         AccessControlList.objects.check_access(
@@ -338,7 +338,7 @@ class DocumentRestoreView(ConfirmView):
         instance.restore()
 
     def view_action(self):
-        instance = get_object_or_404(DeletedDocument, pk=self.kwargs['pk'])
+        instance = get_object_or_404(klass=DeletedDocument, pk=self.kwargs['pk'])
 
         self.object_action(instance=instance)
 
@@ -366,7 +366,7 @@ class DocumentTrashView(ConfirmView):
         }
 
     def get_object(self):
-        return get_object_or_404(Document, pk=self.kwargs['pk'])
+        return get_object_or_404(klass=Document, pk=self.kwargs['pk'])
 
     def get_post_action_redirect(self):
         return reverse('documents:document_list_recent_access')
@@ -752,7 +752,7 @@ class DocumentTransformationsCloneView(FormView):
         return context
 
     def get_object(self):
-        instance = get_object_or_404(Document, pk=self.kwargs['pk'])
+        instance = get_object_or_404(klass=Document, pk=self.kwargs['pk'])
 
         AccessControlList.objects.check_access(
             permissions=permission_transformation_edit,
@@ -825,7 +825,7 @@ class DocumentPrint(FormView):
         return context
 
     def get_object(self):
-        return get_object_or_404(Document, pk=self.kwargs['pk'])
+        return get_object_or_404(klass=Document, pk=self.kwargs['pk'])
 
     def get_template_names(self):
         if self.page_group or self.page_range: