Merge pull request #2325 from camptocamp/fix

Avoid error when there are no legend images
camptocamp · May 24, 2024 · 1cc8307 · 1cc8307
2 parents 8a7fe2d + 8e30aec
commit 1cc8307
Showing 1 changed file with 21 additions and 20 deletions.
diff --git a/tilecloud_chain/controller.py b/tilecloud_chain/controller.py
@@ -364,26 +364,27 @@ def _generate_legend_images(gene: TileGeneration) -> None:
                                 response.content,
                                 exc_info=True,
                             )
-                    width = max(i.size[0] for i in legends)
-                    height = sum(i.size[1] for i in legends)
-                    image = Image.new("RGBA", (width, height))
-                    y = 0
-                    for i in legends:
-                        image.paste(i, (0, y))
-                        y += i.size[1]
-                    string_io = BytesIO()
-                    image.save(string_io, FORMAT_BY_CONTENT_TYPE[layer["legend_mime"]])
-                    result = string_io.getvalue()
-                    new_hash = sha1(result).hexdigest()  # nosec
-                    if new_hash != previous_hash:
-                        previous_hash = new_hash
-                        _send(
-                            result,
-                            f"1.0.0/{layer_name}/{layer['wmts_style']}/"
-                            f"legend{zoom}.{layer['legend_extension']}",
-                            layer["legend_mime"],
-                            cache,
-                        )
+                    if legends:
+                        width = max(i.size[0] for i in legends)
+                        height = sum(i.size[1] for i in legends)
+                        image = Image.new("RGBA", (width, height))
+                        y = 0
+                        for i in legends:
+                            image.paste(i, (0, y))
+                            y += i.size[1]
+                        string_io = BytesIO()
+                        image.save(string_io, FORMAT_BY_CONTENT_TYPE[layer["legend_mime"]])
+                        result = string_io.getvalue()
+                        new_hash = sha1(result).hexdigest()  # nosec
+                        if new_hash != previous_hash:
+                            previous_hash = new_hash
+                            _send(
+                                result,
+                                f"1.0.0/{layer_name}/{layer['wmts_style']}/"
+                                f"legend{zoom}.{layer['legend_extension']}",
+                                layer["legend_mime"],
+                                cache,
+                            )
 
 
 def _get_resource(resource: str) -> bytes: