diff --git a/arkindex/images/api.py b/arkindex/images/api.py index 4e4220d00f5eece82403aec5b42c46d8c31d3d00..98386510bded9feb4ac432765ef941571391bcba 100644 --- a/arkindex/images/api.py +++ b/arkindex/images/api.py @@ -55,7 +55,7 @@ class ImageCreate(CreateAPIView): image_hash = serializer.validated_data.get('hash') server = serializer.validated_data.get('server') path = serializer.validated_data.get('path') - existing_image = server.images.filter( + existing_image = server.images.using('default').filter( Q(hash=image_hash) | Q(path=path) ).first() diff --git a/arkindex/images/serializers.py b/arkindex/images/serializers.py index 397e6126bf3e5d70e9248f04cdee971484f47a1f..6e01e9e839bf310050ab6c1d840db1b320b67f06 100644 --- a/arkindex/images/serializers.py +++ b/arkindex/images/serializers.py @@ -105,7 +105,7 @@ class IIIFImageSerializer(ImageSerializer): data['path'] = image_url[len(data['server'].url) + 1:] # Check image with this server and path does not exist - existing_image = Image.objects.filter(server=data['server'], path=data['path']).first() + existing_image = Image.objects.using('default').filter(server=data['server'], path=data['path']).first() if existing_image: raise DuplicatedImage(existing_image.id) @@ -119,7 +119,7 @@ class IIIFImageSerializer(ImageSerializer): # perform_check can alter the image's path since it updates it from info.json. # This can once more cause duplicated images, so we re-check for an existing image. - existing_image = image.server.images.filter(path=image.path).first() + existing_image = image.server.images.using('default').filter(path=image.path).first() if existing_image: raise DuplicatedImage(existing_image.id) @@ -148,7 +148,7 @@ class ImageInformationSerializer(serializers.Serializer): raise ValidationError({'@id': [str(e)]}) image_path = url[len(server.url) + 1:] - image = server.images.filter(path=image_path).first() + image = server.images.using('default').filter(path=image_path).first() if not image: # This is a new image image = Image(server=server, path=image_path) @@ -163,7 +163,7 @@ class ImageInformationSerializer(serializers.Serializer): image = Image(**validated_data) if 'id' in validated_data: try: - image = Image.objects.get(pk=validated_data['id']) + image = Image.objects.using('default').get(pk=validated_data['id']) except Image.DoesNotExist: pass