fix and add support in viz

Author: vincentsarago

CHANGES.md

@@ -1,3 +1,7 @@
+## 0.15.0 (2025-02-26)
+
+* add support for `VSIFile` backend (https://github.com/developmentseed/tilebench/pull/27)
+
 ## 0.14.0 (2025-01-06)
 
 * remove `python 3.8` support

tilebench/scripts/cli.py

@@ -323,7 +323,14 @@ def random(input, zoom, reader, tms, reader_params):
     callback=options_to_dict,
     help="Reader Options.",
 )
-def viz(src_path, port, host, server_only, reader, config, reader_params):
+@click.option(
+    "--io",
+    "io_backend",
+    type=click.Choice(["vsifile", "rasterio"], case_sensitive=True),
+    help="IO Backend Options.",
+    default="rasterio",
+)
+def viz(src_path, port, host, server_only, reader, config, reader_params, io_backend):
     """WEB UI to visualize VSI statistics for a web mercator tile requests."""
     if reader:
         module, classname = reader.rsplit(".", 1)
@@ -342,6 +349,7 @@ def viz(src_path, port, host, server_only, reader, config, reader_params):
         port=port,
         host=host,
         config=config,
+        io_backend=io_backend,
     )
     if not server_only:
         click.echo(f"Viewer started at {application.template_url}", err=True)

tilebench/viz.py

@@ -131,6 +131,7 @@ class TileDebug:
     port: int = attr.ib(default=8080)
     host: str = attr.ib(default="127.0.0.1")
     config: Dict = attr.ib(default=dict)
+    io_backend: str = attr.ib(default="rasterio")
 
     router: Optional[APIRouter] = attr.ib(init=False)
 
@@ -215,6 +216,7 @@ def tile(
                 add_to_return=True,
                 raw=False,
                 config=self.config,
+                io=self.io_backend,
             )
             def _read_tile(src_path: str, x: int, y: int, z: int):
                 with self.reader(src_path, **self.reader_params) as src_dst:
@@ -295,10 +297,10 @@ def info():
                     }
                 ]
 
-            try:
-                ovr = src_dst.dataset.overviews(1)
-            except Exception:
-                ovr = []
+                try:
+                    ovr = src_dst.dataset.overviews(1)
+                except Exception:
+                    ovr = []
 
             info["overviews"] = len(ovr)