formating

Author: vincentsarago

pctiler/pctiler/endpoints/item.py

@@ -132,20 +132,23 @@ def map(
     )
 
 
+prefix = pc_tile_factory.operation_prefix
+
+
 # crop/feature endpoint compat with titiler<0.15 (`/crop` was renamed `/feature`)
 @pc_tile_factory.router.post(
     r"/crop",
-    operation_id=f"{self.operation_prefix}postDataForGeoJSONCrop",
+    operation_id=f"{prefix}postDataForGeoJSONCrop",
     **img_endpoint_params,
 )
 @pc_tile_factory.router.post(
     r"/crop.{format}",
-    operation_id=f"{self.operation_prefix}postDataForGeoJSONWithFormatCrop",
+    operation_id=f"{prefix}postDataForGeoJSONWithFormatCrop",
     **img_endpoint_params,
 )
 @pc_tile_factory.router.post(
     r"/crop/{width}x{height}.{format}",
-    operation_id=f"{self.operation_prefix}postDataForGeoJSONWithSizesAndFormatCrop",
+    operation_id=f"{prefix}postDataForGeoJSONWithSizesAndFormatCrop",
     **img_endpoint_params,
 )
 def geojson_crop(  # type: ignore
@@ -156,7 +159,7 @@ def geojson_crop(  # type: ignore
     format: Annotated[
         ImageType,
         Field(
-            description="Default will be automatically defined if the output image needs a mask (png) or not (jpeg)."
+            description="Default will be automatically defined if the output image needs a mask (png) or not (jpeg)."  # noqa: F722,E501
         ),
     ] = None,  # type: ignore[assignment]
     src_path=Depends(pc_tile_factory.path_dependency),
@@ -196,52 +199,56 @@ def geojson_crop(  # type: ignore
 # /tiles endpoint compat with titiler<0.15, Optional `tileMatrixSetId`
 @pc_tile_factory.router.get(
     "/tiles/{z}/{x}/{y}",
-    operation_id=f"{pc_tile_factory.operation_prefix}getWebMercatorQuadTile",
+    operation_id=f"{prefix}getWebMercatorQuadTile",
     **img_endpoint_params,
 )
 @pc_tile_factory.router.get(
     "/tiles/{z}/{x}/{y}.{format}",
-    operation_id=f"{pc_tile_factory.operation_prefix}getWebMercatorQuadTileWithFormat",
+    operation_id=f"{prefix}getWebMercatorQuadTileWithFormat",
     **img_endpoint_params,
 )
 @pc_tile_factory.router.get(
     "/tiles/{z}/{x}/{y}@{scale}x",
-    operation_id=f"{pc_tile_factory.operation_prefix}getWebMercatorQuadTileWithScale",
+    operation_id=f"{prefix}getWebMercatorQuadTileWithScale",
     **img_endpoint_params,
 )
 @pc_tile_factory.router.get(
     "/tiles/{z}/{x}/{y}@{scale}x.{format}",
-    operation_id=f"{pc_tile_factory.operation_prefix}getWebMercatorQuadTileWithFormatAndScale",
+    operation_id=f"{prefix}getWebMercatorQuadTileWithFormatAndScale",
     **img_endpoint_params,
 )
-def tile_compat(
+def tile_compat(  # type: ignore
     request: fastapi.Request,
     z: Annotated[
         int,
         Path(
-            description="Identifier (Z) selecting one of the scales defined in the TileMatrixSet and representing the scaleDenominator the tile.",
+            description="Identifier (Z) selecting one of the scales defined in the TileMatrixSet and representing the scaleDenominator the tile.",  # noqa: F722,E501
         ),
     ],
     x: Annotated[
         int,
         Path(
-            description="Column (X) index of the tile on the selected TileMatrix. It cannot exceed the MatrixHeight-1 for the selected TileMatrix.",
+            description="Column (X) index of the tile on the selected TileMatrix. It cannot exceed the MatrixHeight-1 for the selected TileMatrix.",  # noqa: F722,E501
         ),
     ],
     y: Annotated[
         int,
         Path(
-            description="Row (Y) index of the tile on the selected TileMatrix. It cannot exceed the MatrixWidth-1 for the selected TileMatrix.",
+            description="Row (Y) index of the tile on the selected TileMatrix. It cannot exceed the MatrixWidth-1 for the selected TileMatrix.",  # noqa: F722,E501
         ),
     ],
     scale: Annotated[
         int,
-        Field(gt=0, le=4, description="Tile size scale. 1=256x256, 2=512x512..."),
+        Field(
+            gt=0,
+            le=4,
+            description="Tile size scale. 1=256x256, 2=512x512...",  # noqa: F722,E501
+        ),
     ] = 1,
     format: Annotated[
-        ImageType,
+        Optional[ImageType],
         Field(
-            description="Default will be automatically defined if the output image needs a mask (png) or not (jpeg)."
+            description="Default will be automatically defined if the output image needs a mask (png) or not (jpeg)."  # noqa: F722,E501
         ),
     ] = None,
     src_path=Depends(pc_tile_factory.path_dependency),
@@ -288,25 +295,29 @@ def tile_compat(
     response_model_exclude_none=True,
     operation_id=f"{pc_tile_factory.operation_prefix}getWebMercatorQuadTileJSON",
 )
-def tilejson_compat(
+def tilejson_compat(  # type: ignore
     request: fastapi.Request,
     tile_format: Annotated[
         Optional[ImageType],
         Query(
-            description="Default will be automatically defined if the output image needs a mask (png) or not (jpeg).",
+            description="Default will be automatically defined if the output image needs a mask (png) or not (jpeg)."  # noqa: F722,E501
         ),
     ] = None,
     tile_scale: Annotated[
         int,
-        Query(gt=0, lt=4, description="Tile size scale. 1=256x256, 2=512x512..."),
+        Query(
+            gt=0,
+            lt=4,
+            description="Tile size scale. 1=256x256, 2=512x512...",  # noqa: E501,F722
+        ),
     ] = 1,
     minzoom: Annotated[
         Optional[int],
-        Query(description="Overwrite default minzoom."),
+        Query(description="Overwrite default minzoom."),  # noqa: F722
     ] = None,
     maxzoom: Annotated[
         Optional[int],
-        Query(description="Overwrite default maxzoom."),
+        Query(description="Overwrite default maxzoom."),  # noqa: F722
     ] = None,
     src_path=Depends(pc_tile_factory.path_dependency),
     reader_params=Depends(pc_tile_factory.reader_dependency),

pctiler/pctiler/endpoints/legend.py

@@ -70,7 +70,7 @@ async def get_classmap_legend(
 
     keys = list(classmap.keys())  # type: ignore
     trimmed_keys = keys[trim_start : len(keys) - trim_end]
-    trimmed_map = {k: classmap[k] for k in trimmed_keys}
+    trimmed_map = {k: classmap[k] for k in trimmed_keys}  # type: ignore
 
     return JSONResponse(content=trimmed_map)
 
@@ -155,7 +155,7 @@ def make_colormap(name: str, trim_start: int, length: int) -> ListedColormap:
     if len(cm) > 256 or max(cm) >= 256:
         raise Exception("Cannot make a colormap for discrete colormap")
 
-    colors = make_lut(cm)
+    colors = make_lut(cm)  # type: ignore
 
     colors = colors[trim_start : length + 1]
     # rescale to 0-1

pctiler/pctiler/endpoints/pg_mosaic.py

@@ -6,7 +6,6 @@
 from psycopg_pool import ConnectionPool
 from pydantic import Field
 from titiler.core import dependencies
-from titiler.core.dependencies import ColorFormulaParams
 from titiler.core.factory import img_endpoint_params
 from titiler.core.resources.enums import ImageType
 from titiler.pgstac.dependencies import SearchIdParams
@@ -87,7 +86,8 @@ def mosaic_info(
 legacy_mosaic_router = APIRouter()
 
 
-# Compat with titiler-pgstac<0.3.0, (`/tiles/{search_id}/...` was renamed `/{search_id}/tiles/...`)
+# Compat with titiler-pgstac<0.3.0,
+# (`/tiles/{search_id}/...` was renamed `/{search_id}/tiles/...`)
 @legacy_mosaic_router.get("/tiles/{search_id}/{z}/{x}/{y}", **img_endpoint_params)
 @legacy_mosaic_router.get(
     "/tiles/{search_id}/{z}/{x}/{y}.{format}", **img_endpoint_params
@@ -118,25 +118,25 @@ def tile_routes(  # type: ignore
     z: Annotated[
         int,
         Path(
-            description="Identifier (Z) selecting one of the scales defined in the TileMatrixSet and representing the scaleDenominator the tile.",
+            description="Identifier (Z) selecting one of the scales defined in the TileMatrixSet and representing the scaleDenominator the tile.",  # noqa: F722,E501
         ),
     ],
     x: Annotated[
         int,
         Path(
-            description="Column (X) index of the tile on the selected TileMatrix. It cannot exceed the MatrixHeight-1 for the selected TileMatrix.",
+            description="Column (X) index of the tile on the selected TileMatrix. It cannot exceed the MatrixHeight-1 for the selected TileMatrix.",  # noqa: F722,E501
         ),
     ],
     y: Annotated[
         int,
         Path(
-            description="Row (Y) index of the tile on the selected TileMatrix. It cannot exceed the MatrixWidth-1 for the selected TileMatrix.",
+            description="Row (Y) index of the tile on the selected TileMatrix. It cannot exceed the MatrixWidth-1 for the selected TileMatrix.",  # noqa: F722,E501
         ),
     ],
     search_id=Depends(pgstac_mosaic_factory.path_dependency),
     tileMatrixSetId: Annotated[  # type: ignore
         Literal[tuple(pgstac_mosaic_factory.supported_tms.list())],
-        f"Identifier selecting one of the TileMatrixSetId supported (default: 'WebMercatorQuad')",  # noqa: E501,F722
+        "Identifier selecting one of the TileMatrixSetId supported (default: 'WebMercatorQuad')",  # noqa: E501,F722
     ] = "WebMercatorQuad",
     scale: Annotated[  # type: ignore
         Optional[Annotated[int, Field(gt=0, le=4)]],
@@ -145,7 +145,7 @@ def tile_routes(  # type: ignore
     format: Annotated[
         Optional[ImageType],
         Field(
-            description="Default will be automatically defined if the output image needs a mask (png) or not (jpeg)."
+            description="Default will be automatically defined if the output image needs a mask (png) or not (jpeg)."  # noqa: F722,E501
         ),
     ] = None,
     backend_params=Depends(pgstac_mosaic_factory.backend_dependency),

pctiler/pctiler/reader.py

@@ -160,7 +160,9 @@ def _get_asset_info(self, asset: str) -> AssetInfo:
             info["media_type"] = media_type
 
         if header_size := asset_info.get("file:header_size"):
-            info["env"]["GDAL_INGESTED_BYTES_AT_OPEN"] = header_size
+            info["env"].update(  # type: ignore
+                {"GDAL_INGESTED_BYTES_AT_OPEN": header_size}
+            )
 
         if bands := asset_info.get("raster:bands"):
             stats = [