Revert change to factory

developmentseed · Jun 5, 2024 · 28574cd · 28574cd
1 parent 55e5775
commit 28574cd
Showing 1 changed file with 28 additions and 24 deletions.
diff --git a/titiler/stacapi/factory.py b/titiler/stacapi/factory.py
@@ -6,6 +6,7 @@
 from urllib.parse import urlencode
 
 import jinja2
+import rasterio
 from cogeo_mosaic.backends import BaseBackend
 from fastapi import Depends, HTTPException, Path, Query
 from fastapi.dependencies.utils import get_dependant, request_params_to_args
@@ -194,32 +195,35 @@ def tile(
             scale = scale or 1
 
             tms = self.supported_tms.get(tileMatrixSetId)
-            with self.reader(
-                url=api_params["api_url"],
-                headers=api_params.get("headers", {}),
-                tms=tms,
-                reader_options={**reader_params},
-                **backend_params,
-            ) as src_dst:
-                if MOSAIC_STRICT_ZOOM and (z < src_dst.minzoom or z > src_dst.maxzoom):
-                    raise HTTPException(
-                        400,
-                        f"Invalid ZOOM level {z}. Should be between {src_dst.minzoom} and {src_dst.maxzoom}",
+            with rasterio.Env(**env):
+                with self.reader(
+                    url=api_params["api_url"],
+                    headers=api_params.get("headers", {}),
+                    tms=tms,
+                    reader_options={**reader_params},
+                    **backend_params,
+                ) as src_dst:
+                    if MOSAIC_STRICT_ZOOM and (
+                        z < src_dst.minzoom or z > src_dst.maxzoom
+                    ):
+                        raise HTTPException(
+                            400,
+                            f"Invalid ZOOM level {z}. Should be between {src_dst.minzoom} and {src_dst.maxzoom}",
+                        )
+
+                    image, assets = src_dst.tile(
+                        x,
+                        y,
+                        z,
+                        search_query=search_query,
+                        tilesize=scale * 256,
+                        pixel_selection=pixel_selection,
+                        threads=MOSAIC_THREADS,
+                        **tile_params,
+                        **layer_params,
+                        **dataset_params,
                     )
 
-                image, assets = src_dst.tile(
-                    x,
-                    y,
-                    z,
-                    search_query=search_query,
-                    tilesize=scale * 256,
-                    pixel_selection=pixel_selection,
-                    threads=MOSAIC_THREADS,
-                    **tile_params,
-                    **layer_params,
-                    **dataset_params,
-                )
-
             if post_process:
                 image = post_process(image)