matthias/mayan-edms
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add klass argument to get_object_or_404 usage

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

4
mayan/apps/document_parsing/api_views.py
View File

@@ -25,11 +25,11 @@ class APIDocumentPageContentView(generics.RetrieveAPIView):
serializer_class = DocumentPageContentSerializer
def get_document(self):
return get_object_or_404(Document, pk=self.kwargs['document_pk'])
return get_object_or_404(klass=Document, pk=self.kwargs['document_pk'])
def get_document_version(self):
return get_object_or_404(
self.get_document().versions.all(), pk=self.kwargs['version_pk']
klass=self.get_document().versions.all(), pk=self.kwargs['version_pk']
)
def get_queryset(self):

4
mayan/apps/document_parsing/views.py
View File

@@ -84,7 +84,7 @@ class DocumentParsingErrorsListView(SingleObjectListView):
view_permission = permission_content_view
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 {
@@ -144,7 +144,7 @@ class DocumentTypeSettingsEditView(SingleObjectEditView):
post_action_redirect = reverse_lazy('documents:document_type_list')
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 {