Fix #142: catch StopIteration and return InvalidProductError

sertit · Jun 3, 2024 · 83f1cca · 83f1cca
1 parent b79ed4f
commit 83f1cca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/eoreader/products/sar/cosmo_product.py b/eoreader/products/sar/cosmo_product.py
@@ -93,7 +93,7 @@ def __init__(
         try:
             product_path = AnyPath(product_path)
             self._img_path = next(product_path.glob("*.h5"))
-        except IndexError as ex:
+        except (IndexError, StopIteration) as ex:
             raise InvalidProductError(
                 f"Image file (*.h5) not found in {product_path}"
             ) from ex

diff --git a/eoreader/products/sar/rcm_product.py b/eoreader/products/sar/rcm_product.py
@@ -205,7 +205,7 @@ def wgs84_extent(self) -> gpd.GeoDataFrame:
         try:
             extent_file = next(self.path.joinpath("preview").glob("*mapOverlay.kml"))
             product_kml = vectors.read(extent_file)
-        except IndexError as ex:
+        except (IndexError, StopIteration) as ex:
             raise InvalidProductError(
                 f"Extent file (product.kml) not found in {self.path}"
             ) from ex