Skip to content
Snippets Groups Projects
Commit 0380a5e8 authored by Yoann Schneider's avatar Yoann Schneider :tennis: Committed by Bastien Abadie
Browse files

Allow resizing sub elements

parent 9fbb705a
No related branches found
No related tags found
1 merge request!396Allow resizing sub elements
Pipeline #132102 passed
...@@ -10,8 +10,6 @@ from typing import Generator, List, Optional ...@@ -10,8 +10,6 @@ from typing import Generator, List, Optional
from PIL import Image from PIL import Image
from requests import HTTPError from requests import HTTPError
from arkindex_worker import logger
class MagicDict(dict): class MagicDict(dict):
""" """
...@@ -172,7 +170,6 @@ class Element(MagicDict): ...@@ -172,7 +170,6 @@ class Element(MagicDict):
from arkindex_worker.image import ( from arkindex_worker.image import (
download_tiles, download_tiles,
open_image, open_image,
polygon_bounding_box,
) )
if not self.get("zone"): if not self.get("zone"):
...@@ -187,20 +184,9 @@ class Element(MagicDict): ...@@ -187,20 +184,9 @@ class Element(MagicDict):
if max_width is None and max_height is None: if max_width is None and max_height is None:
resize = "full" resize = "full"
else: else:
bounding_box = polygon_bounding_box(self.zone.polygon)
original_size = {"w": self.zone.image.width, "h": self.zone.image.height} original_size = {"w": self.zone.image.width, "h": self.zone.image.height}
# No resizing if the element is smaller than the image.
if (
bounding_box.width != original_size["w"]
or bounding_box.height != original_size["h"]
):
resize = "full"
logger.warning(
"Only full image size elements covered, "
+ "downloading full size image."
)
# No resizing if the image is smaller than the wanted size. # No resizing if the image is smaller than the wanted size.
elif (max_width is None or original_size["w"] <= max_width) and ( if (max_width is None or original_size["w"] <= max_width) and (
max_height is None or original_size["h"] <= max_height max_height is None or original_size["h"] <= max_height
): ):
resize = "full" resize = "full"
......
...@@ -141,7 +141,7 @@ def test_open_image_resize_partial_element(mocker): ...@@ -141,7 +141,7 @@ def test_open_image_resize_partial_element(mocker):
assert elt.open_image(max_height=400, use_full_image=True) == "an image!" assert elt.open_image(max_height=400, use_full_image=True) == "an image!"
assert open_mock.call_count == 1 assert open_mock.call_count == 1
assert open_mock.call_args == mocker.call( assert open_mock.call_args == mocker.call(
"http://something/full/full/0/default.jpg", "http://something/full/,400/0/default.jpg",
rotation_angle=0, rotation_angle=0,
mirrored=False, mirrored=False,
) )
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment