update and fix

Author: vincentsarago

pyproject.toml

@@ -96,7 +96,7 @@ known_third_party = [
 ]
 default_section = "THIRDPARTY"
 
-[tool.ruff]
+[tool.ruff.lint]
 select = [
     "D1",  # pydocstyle errors
     "E",  # pycodestyle errors

titiler/stacapi/reader.py

@@ -1,16 +1,13 @@
 """Custom STAC reader."""
 
-from typing import Any, Dict, Optional, Set, Type
+from typing import Any, Dict, Optional, Sequence, Set, Type
 
 import attr
 import pystac
 import rasterio
 from morecantile import TileMatrixSet
-from rasterio.crs import CRS
-from rio_tiler.constants import WEB_MERCATOR_TMS, WGS84_CRS
-from rio_tiler.errors import InvalidAssetName
+from rio_tiler.constants import WEB_MERCATOR_TMS
 from rio_tiler.io import BaseReader, Reader, stac
-from rio_tiler.types import AssetInfo
 
 from titiler.stacapi.settings import STACSettings
 
@@ -25,77 +22,30 @@ class STACReader(stac.STACReader):
 
     """
 
-    tms: TileMatrixSet = attr.ib(default=WEB_MERCATOR_TMS)
-    minzoom: int = attr.ib()
-    maxzoom: int = attr.ib()
+    input: pystac.Item = attr.ib()
 
-    geographic_crs: CRS = attr.ib(default=WGS84_CRS)
+    tms: TileMatrixSet = attr.ib(default=WEB_MERCATOR_TMS)
+    minzoom: int = attr.ib(default=None)
+    maxzoom: int = attr.ib(default=None)
 
     include_assets: Optional[Set[str]] = attr.ib(default=None)
     exclude_assets: Optional[Set[str]] = attr.ib(default=None)
 
     include_asset_types: Set[str] = attr.ib(default=stac.DEFAULT_VALID_TYPE)
     exclude_asset_types: Optional[Set[str]] = attr.ib(default=None)
 
+    assets: Sequence[str] = attr.ib(init=False)
+    default_assets: Optional[Sequence[str]] = attr.ib(default=None)
+
     reader: Type[BaseReader] = attr.ib(default=Reader)
     reader_options: Dict = attr.ib(factory=dict)
 
-    fetch_options: Dict = attr.ib(factory=dict)
-
     ctx: Any = attr.ib(default=rasterio.Env)
 
     item: pystac.Item = attr.ib(init=False)
+    fetch_options: Dict = attr.ib(factory=dict)
 
     def __attrs_post_init__(self):
         """Fetch STAC Item and get list of valid assets."""
         self.item = self.input
         super().__attrs_post_init__()
-
-    @minzoom.default
-    def _minzoom(self):
-        return self.tms.minzoom
-
-    @maxzoom.default
-    def _maxzoom(self):
-        return self.tms.maxzoom
-
-    def _get_asset_info(self, asset: str) -> AssetInfo:
-        """Validate asset names and return asset's url.
-
-        Args:
-            asset (str): STAC asset name.
-
-        Returns:
-            str: STAC asset href.
-
-        """
-        if asset not in self.assets:
-            raise InvalidAssetName(
-                f"'{asset}' is not valid, should be one of {self.assets}"
-            )
-
-        asset_info = self.item.assets[asset]
-        extras = asset_info.extra_fields
-
-        url = asset_info.get_absolute_href() or asset_info.href
-        if alternate := stac_config.alternate_url:
-            url = asset_info.to_dict()["alternate"][alternate]["href"]
-
-        info = AssetInfo(
-            url=url,
-            metadata=extras,
-        )
-
-        if head := extras.get("file:header_size"):
-            info["env"] = {"GDAL_INGESTED_BYTES_AT_OPEN": head}
-
-        if bands := extras.get("raster:bands"):
-            stats = [
-                (b["statistics"]["minimum"], b["statistics"]["maximum"])
-                for b in bands
-                if {"minimum", "maximum"}.issubset(b.get("statistics", {}))
-            ]
-            if len(stats) == len(bands):
-                info["dataset_statistics"] = stats
-
-        return info