Add klass argument to get_object_or_404 usage

Signed-off-by: Roberto Rosario <Roberto.Rosario@mayan-edms.com>
2018-12-16 01:55:33 -04:00
parent 727d2ecd71
commit 798446f362
39 changed files with 164 additions and 164 deletions
--- a/mayan/apps/sources/api_views.py
+++ b/mayan/apps/sources/api_views.py
@@ -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(