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
mayan/apps/sources/api_views.py
+1 -1
View File
@@ -21,7 +21,7 @@ class APIStagingSourceFileView(generics.GenericAPIView):
def get(self, request, staging_folder_pk, encoded_filename):
staging_folder = get_object_or_404(
StagingFolderSource, pk=staging_folder_pk
klass=StagingFolderSource, pk=staging_folder_pk
)
return Response(
StagingFolderFileSerializer(
mayan/apps/sources/views.py
+8 -8
View File
@@ -74,7 +74,7 @@ class SourceLogListView(SingleObjectListView):
def get_source(self):
return get_object_or_404(
Source.objects.select_subclasses(), pk=self.kwargs['pk']
klass=Source.objects.select_subclasses(), pk=self.kwargs['pk']
)
@@ -110,7 +110,7 @@ class UploadBaseView(MultiFormView):
def dispatch(self, request, *args, **kwargs):
if 'source_id' in kwargs:
self.source = get_object_or_404(
Source.objects.filter(enabled=True).select_subclasses(),
klass=Source.objects.filter(enabled=True).select_subclasses(),
pk=kwargs['source_id']
)
else:
@@ -196,7 +196,7 @@ class UploadInteractiveView(UploadBaseView):
self.subtemplates_list = []
self.document_type = get_object_or_404(
DocumentType,
klass=DocumentType,
pk=self.request.GET.get(
'document_type_id', self.request.POST.get('document_type_id')
)
@@ -367,7 +367,7 @@ class UploadInteractiveVersionView(UploadBaseView):
self.subtemplates_list = []
self.document = get_object_or_404(Document, pk=kwargs['document_pk'])
self.document = get_object_or_404(klass=Document, pk=kwargs['document_pk'])
# TODO: Try to remove this new version block check from here
if NewVersionBlock.objects.is_blocked(self.document):
@@ -490,7 +490,7 @@ class StagingFileDeleteView(SingleObjectDeleteView):
def get_source(self):
return get_object_or_404(
StagingFolderSource, pk=self.kwargs['pk']
klass=StagingFolderSource, pk=self.kwargs['pk']
)
@@ -509,7 +509,7 @@ class SetupSourceCheckView(ConfirmView):
}
def get_object(self):
return get_object_or_404(Source.objects.select_subclasses(), pk=self.kwargs['pk'])
return get_object_or_404(klass=Source.objects.select_subclasses(), pk=self.kwargs['pk'])
def view_action(self):
task_check_interval_source.apply_async(
@@ -543,7 +543,7 @@ class SetupSourceDeleteView(SingleObjectDeleteView):
def get_object(self):
return get_object_or_404(
Source.objects.select_subclasses(), pk=self.kwargs['pk']
klass=Source.objects.select_subclasses(), pk=self.kwargs['pk']
)
def get_form_class(self):
@@ -562,7 +562,7 @@ class SetupSourceEditView(SingleObjectEditView):
def get_object(self):
return get_object_or_404(
Source.objects.select_subclasses(), pk=self.kwargs['pk']
klass=Source.objects.select_subclasses(), pk=self.kwargs['pk']
)
def get_form_class(self):