Fix remaining mypy errors

Author: underchemist

.pre-commit-config.yaml

@@ -16,4 +16,5 @@ repos:
     rev: v0.910
     hooks:
     - id: mypy
-      language_version: python
+      language_version: python
+      args: [--install-types, --non-interactive]

oaff/app/oaff/app/data/sources/postgresql/stac_hybrid/postgresql_data_source.py

@@ -2,7 +2,7 @@
 from hashlib import sha256
 from logging import getLogger
 from os import path
-from typing import Any, Coroutine, Dict, Final, List, Type
+from typing import Any, Awaitable, Callable, Dict, Final, List, Type
 
 # geoalchemy import required for sa.MetaData reflection, even though unused in module
 import geoalchemy2 as ga  # noqa: F401
@@ -51,7 +51,7 @@ class PostgresqlDataSource(DataSource):
     def __init__(
         self,
         connection_name: str,
-        connection_tester: Coroutine[None, None, Database],
+        connection_tester: Callable[[Database, str], Awaitable[None]],
     ):
         super().__init__(f"{self.DATA_SOURCE_NAME}:{connection_name}")
         self.db = Database(settings.url(connection_name))
@@ -225,7 +225,7 @@ async def _get_derived_layers(self) -> Dict[str, PostgresqlLayer]:
                 bboxes=[table_spatial_extents[qualified_layer_name]],
                 intervals=table_temporal_extents[qualified_layer_name]
                 if qualified_layer_name in table_temporal_extents
-                else [[None, None]],
+                else [[None, None]],  # type: ignore
                 data_source_id=self.id,
                 schema_name=tables[qualified_layer_name]["schema_name"],
                 table_name=tables[qualified_layer_name]["table_name"],

oaff/app/oaff/app/request_handlers/collection_items.py

@@ -130,14 +130,13 @@ async def _spatial_bounds_to_node(
     ) -> BBox:
         # recommended usage for Union of types
         # https://github.com/python/mypy/issues/1178#issuecomment-176185607
-        x_min, y_min, x_max, y_max = (
-            cast(Tuple[float, float, float, float], spatial_bounds)  # type: ignore
-            if len(spatial_bounds) == 4
-            else (
-                cast(Tuple[float, float, float, float, float, float], spatial_bounds)[i]
-                for i in [0, 1, 3, 4]
+        if len(spatial_bounds) == 4:
+            a, b, c, d = cast(Tuple[float, float, float, float], spatial_bounds)
+        else:
+            a, b, _, c, d, _ = cast(
+                Tuple[float, float, float, float, float, float], spatial_bounds
             )
-        )
+        x_min, y_min, x_max, y_max = (a, b, c, d)
         transformer = Transformer.from_crs(
             "EPSG:4326",  # True until Features API spec part 2 is implemented
             f"{layer.geometry_crs_auth_name}:{layer.geometry_crs_auth_code}",
@@ -336,6 +335,6 @@ def _match_query_time_to_end_field(
         )
 
     def _match_query_time_to(
-        self, query_time: datetime, tz_aware: bool, tz: Type[tzinfo]
+        self, query_time: datetime, tz_aware: bool, tz: tzinfo
     ) -> datetime:
         return query_time if tz_aware else query_time.astimezone(tz).replace(tzinfo=None)

oaff/app/oaff/app/request_handlers/collections_list.py

@@ -12,7 +12,7 @@
 LOGGER: Final = getLogger(__file__)
 
 
-class CollectionsList(RequestHandler):
+class CollectionsList(RequestHandler):  # type: ignore
     @classmethod
     def type_name(cls) -> str:
         return CollectionsList.__name__

oaff/app/oaff/app/request_handlers/conformance.py

@@ -7,7 +7,7 @@
 from oaff.app.responses.response_format import ResponseFormat
 
 
-class Conformance(RequestHandler):
+class Conformance(RequestHandler):  # type: ignore
     @classmethod
     def type_name(cls) -> str:
         return Conformance.__name__

oaff/app/oaff/app/request_handlers/landing_page.py

@@ -12,7 +12,7 @@
 from oaff.app.settings import OPENAPI_OGC_TYPE
 
 
-class LandingPage(RequestHandler):
+class LandingPage(RequestHandler):  # type: ignore
     @classmethod
     def type_name(cls) -> str:
         return LandingPage.__name__

oaff/app/oaff/app/responses/response_format.py

@@ -3,7 +3,7 @@
 from oaff.app.responses.response_type import ResponseType
 
 
-class ResponseFormat(dict, Enum):
+class ResponseFormat(dict, Enum):  # type: ignore
     html = {
         ResponseType.DATA: "text/html",
         ResponseType.METADATA: "text/html",

oaff/app/oaff/app/responses/templates/templates.py

@@ -13,7 +13,7 @@ def get_rendered_html(template_name: str, data: object, locale: Locales) -> str:
         loader=PackageLoader("oaff.app", path.join("responses", "templates", "html")),
         autoescape=select_autoescape(["html"]),
     )
-    env.install_gettext_translations(get_translations_for_locale(locale))
+    env.install_gettext_translations(get_translations_for_locale(locale))  # type: ignore
     frontend_config = get_frontend_configuration()
     return env.get_template(f"{template_name}.jinja2").render(
         response=data,

oaff/fastapi/api/openapi/openapi.py

@@ -1,10 +1,12 @@
+from typing import Callable
+
 from fastapi.applications import FastAPI
 from fastapi.requests import Request
 
 from oaff.fastapi.api.openapi.vnd_response import VndResponse
 
 
-def get_openapi_handler(app: FastAPI) -> VndResponse:
+def get_openapi_handler(app: FastAPI) -> Callable[[Request], VndResponse]:
     def handler(_: Request):
         # OpenAPI spec must be modified because FastAPI doesn't support
         # encoding style: https://github.com/tiangolo/fastapi/issues/283

oaff/fastapi/api/routes/collections.py

@@ -1,13 +1,12 @@
 from datetime import datetime
 from logging import getLogger
-from typing import Final, Optional, Tuple, Union
+from typing import Final, Optional, Tuple, Union, cast
 from urllib.parse import quote
 
 import iso8601
 import pytz
-from fastapi import APIRouter, HTTPException
+from fastapi import APIRouter, HTTPException, Query
 from fastapi.param_functions import Depends
-from fastapi.params import Query
 from fastapi.requests import Request
 
 from oaff.app.requests.collection import Collection as CollectionRequestType
@@ -216,7 +215,7 @@ def parse_datetime(datetime_str: str) -> datetime:
             raise HTTPException(
                 status_code=400, detail="datetime start cannot be after end"
             )
-    return tuple(result)
+    return cast(Union[Tuple[datetime], Tuple[datetime, datetime]], tuple(result))
 
 
 def _get_safe_url(path_template: str, request: Request, root: str) -> str:

oaff/fastapi/api/routes/common/common_parameters.py

@@ -1,5 +1,5 @@
 from re import compile, search, sub
-from typing import Final, Optional, Set
+from typing import Final, List, Optional
 
 from fastapi import Header, Query
 from fastapi.requests import Request
@@ -66,7 +66,7 @@ async def populate(
         )
 
     @classmethod
-    def _header_options_by_preference(cls, header_value: str) -> Set[str]:
+    def _header_options_by_preference(cls, header_value: str) -> List[str]:
         options = list(
             filter(
                 lambda option: len(option) > 0,
@@ -84,6 +84,4 @@ def _header_options_by_preference(cls, header_value: str) -> Set[str]:
             option_and_weight[0].strip(): option_and_weight[1].strip()
             for option_and_weight in [option.split(";q=") for option in weighted]
         }
-        return set(
-            unweighted + sorted(options_with_weight, key=options_with_weight.get)[::-1]
-        )
+        return unweighted + sorted(options_with_weight, key=options_with_weight.get)[::-1]

pyproject.toml

@@ -38,6 +38,5 @@ exclude="^(tests|testing)$"
 namespace_packages=true
 explicit_package_bases=true
 ignore_missing_imports=true
-ignore_errors=true
 no_warn_no_return=true
 no_strict_optional=true