Add message for the use in legeng and hash generation

camptocamp · Nov 30, 2024 · 3a0eb1e · 3a0eb1e
1 parent 0560276
commit 3a0eb1e
Show file tree

Hide file tree

Showing 2 changed files with 72 additions and 4 deletions.
diff --git a/tilecloud_chain/__init__.py b/tilecloud_chain/__init__.py
@@ -1578,7 +1578,8 @@ def __call__(self, tile: Tile) -> Tile:
             if ref is None:
                 ref = px
             elif px != ref:
-                _LOGGER.error("Error: image is not uniform.")
+                print("Error: image is not uniform.")
+                _LOGGER.debug("Error: image is not uniform.")
                 sys.exit(1)
 
         assert tile.data

diff --git a/tilecloud_chain/controller.py b/tilecloud_chain/controller.py
@@ -398,16 +398,83 @@ def _generate_legend_images(gene: TileGeneration) -> None:
                         resolution,
                         url,
                     )
-                    response = session.get(url)
+                    try:
+                        response = session.get(url)
+                    except Exception as e:  # pylint: disable=broad-exception-caught
+                        print(
+                            "\n".join(
+                                [
+                                    f"Unable to get legend image for layer '{layer_name}'-'{wmslayer}', resolution '{resolution}'",
+                                    url,
+                                    str(e),
+                                ]
+                            )
+                        )
+                        _LOGGER.debug(
+                            "Unable to get legend image for layer '%s'-'%s', resolution '%s'",
+                            layer_name,
+                            wmslayer,
+                            resolution,
+                            exc_info=True,
+                        )
+                        continue
+                    if response.status_code != 200:
+                        print(
+                            "\n".join(
+                                [
+                                    f"Unable to get legend image for layer '{layer_name}'-'{wmslayer}', resolution '{resolution}'",
+                                    url,
+                                    f"status code: {response.status_code}: {response.reason}",
+                                    response.text,
+                                ]
+                            )
+                        )
+                        _LOGGER.debug(
+                            "Unable to get legend image for layer '%s'-'%s', resolution '%s': %s",
+                            layer_name,
+                            wmslayer,
+                            resolution,
+                            response.text,
+                        )
+                        continue
+                    if not response.headers["Content-Type"].startswith(layer["legend_mime"].split("/")[0]):
+                        print(
+                            "\n".join(
+                                [
+                                    f"Unable to get legend image for layer '{layer_name}'-'{wmslayer}', resolution '{resolution}'",
+                                    url,
+                                    f"Content-Type: {response.headers['Content-Type']}",
+                                    response.text,
+                                ]
+                            )
+                        )
+                        _LOGGER.debug(
+                            "Unable to get legend image for layer '%s'-'%s', resolution '%s', content-type: %s: %s",
+                            layer_name,
+                            wmslayer,
+                            resolution,
+                            response.headers["Content-Type"],
+                            response.text,
+                        )
+                        continue
                     try:
                         legends.append(Image.open(BytesIO(response.content)))
                     except Exception:  # pylint: disable=broad-exception-caught
-                        _LOGGER.warning(
+                        print(
+                            "\n".join(
+                                [
+                                    f"Unable to read legend image for layer '{layer_name}'-'{wmslayer}', resolution '{resolution}'",
+                                    url,
+                                    response.text,
+                                ]
+                            )
+                        )
+                        _LOGGER.debug(
                             "Unable to read legend image for layer '%s'-'%s', resolution '%s': %s",
                             layer_name,
                             wmslayer,
                             resolution,
-                            response.content,
+                            response.text,
                             exc_info=True,
                         )
                 width = max(1, max(i.size[0] for i in legends) if legends else 0)