Fix or disable pyright issues

Author: adamtheturtle

pyproject.toml

@@ -144,6 +144,7 @@ module = [
     "cv2",
     "docker",
     "docker.errors",
+    "docker.models.containers",
     "docker.models.networks",
     "multipart",
     "sybil",

spelling_private_dict.txt

@@ -80,6 +80,7 @@ recognitions
 refactoring
 regex
 reimplementation
+reportGeneralTypeIssues
 repr
 reqheader
 reqjson

src/mock_vws/_query_tools.py

@@ -84,7 +84,9 @@ def get_query_match_response_text(
     else:
         include_target_data = parsed_include_target_data.value.lower()
 
-    image_value = parsed.get("image").raw
+    image_part = parsed.get("image")
+    assert image_part is not None
+    image_value = image_part.raw
     gmt = ZoneInfo("GMT")
     now = datetime.datetime.now(tz=gmt)
 

src/mock_vws/_query_validators/image_validators.py

@@ -67,7 +67,9 @@ def validate_image_file_size(
     boundary = email_message.get_boundary()
     assert isinstance(boundary, str)
     parsed = multipart.MultipartParser(stream=body_file, boundary=boundary)
-    image = parsed.get("image").raw
+    image_part = parsed.get("image")
+    assert image_part is not None
+    image_value = image_part.raw
 
     # This is the documented maximum size of a PNG as per.
     # https://library.vuforia.com/web-api/vuforia-query-web-api.
@@ -77,7 +79,7 @@ def validate_image_file_size(
     # Ignore coverage on this as there is a bug in urllib3 which means that we
     # do not trigger this exception.
     # See https://github.com/urllib3/urllib3/issues/2733.
-    if len(image) > max_bytes:  # pragma: no cover
+    if len(image_value) > max_bytes:  # pragma: no cover
         _LOGGER.warning(msg="The image file size is too large.")
         raise RequestEntityTooLarge
 
@@ -104,8 +106,10 @@ def validate_image_dimensions(
     boundary = email_message.get_boundary()
     assert isinstance(boundary, str)
     parsed = multipart.MultipartParser(stream=body_file, boundary=boundary)
-    image = parsed.get("image").raw
-    image_file = io.BytesIO(image)
+    image_part = parsed.get("image")
+    assert image_part is not None
+    image_value = image_part.raw
+    image_file = io.BytesIO(image_value)
     pil_image = Image.open(image_file)
     max_width = 30000
     max_height = 30000
@@ -137,9 +141,11 @@ def validate_image_format(
     boundary = email_message.get_boundary()
     assert isinstance(boundary, str)
     parsed = multipart.MultipartParser(stream=body_file, boundary=boundary)
-    image = parsed.get("image").raw
+    image_part = parsed.get("image")
+    assert image_part is not None
+    image_value = image_part.raw
 
-    image_file = io.BytesIO(image)
+    image_file = io.BytesIO(image_value)
     pil_image = Image.open(image_file)
 
     if pil_image.format in {"PNG", "JPEG"}:
@@ -170,9 +176,11 @@ def validate_image_is_image(
     boundary = email_message.get_boundary()
     assert isinstance(boundary, str)
     parsed = multipart.MultipartParser(stream=body_file, boundary=boundary)
-    image = parsed.get("image").raw
+    image_part = parsed.get("image")
+    assert image_part is not None
+    image_value = image_part.raw
 
-    image_file = io.BytesIO(image)
+    image_file = io.BytesIO(image_value)
 
     try:
         Image.open(image_file)

src/mock_vws/target_raters.py

@@ -32,7 +32,12 @@ def _get_brisque_target_tracking_rating(image_content: bytes) -> int:
     with np.errstate(divide="ignore", invalid="ignore"):
         try:
             score = brisque_obj.score(img=image_array)
-        except (cv2.error, ValueError):  # pylint: disable=no-member
+        # cv2 errors cannot be inferred without type stubs.
+        # See https://github.com/opencv/opencv/issues/14590.
+        except (
+            cv2.error,  # pyright: ignore[reportGeneralTypeIssues], pylint: disable=no-member
+            ValueError,
+        ):
             return 0
     if math.isnan(score):
         return 0

tests/mock_vws/test_docker.py

@@ -13,7 +13,8 @@
 import docker
 import pytest
 import requests
-from docker.errors import BuildError
+from docker.errors import BuildError, NotFound
+from docker.models.containers import Container
 from mock_vws.database import VuforiaDatabase
 from vws import VWS, CloudRecoService
 
@@ -72,13 +73,15 @@ def fixture_custom_bridge_network() -> Iterator[Network]:
             name="test-vws-bridge-" + uuid.uuid4().hex,
             driver="bridge",
         )
-    except docker.errors.NotFound:
+    except NotFound:
         # On Windows the "bridge" network driver is not available and we use
         # the "nat" driver instead.
         network = client.networks.create(
             name="test-vws-bridge-" + uuid.uuid4().hex,
             driver="nat",
         )
+
+    assert isinstance(network, Network)
     try:
         yield network
     finally:
@@ -115,7 +118,7 @@ def test_build_and_run(
     vwq_tag = f"vws-mock-vwq:latest-{random}"
 
     try:
-        target_manager_image, _ = client.images.build(
+        target_manager_build_result = client.images.build(
             path=str(repository_root),
             dockerfile=str(target_manager_dockerfile),
             tag=target_manager_tag,
@@ -133,16 +136,23 @@ def test_build_and_run(
         reason = "We do not currently support using Windows containers."
         pytest.skip(reason)
 
-    vws_image, _ = client.images.build(
+    assert isinstance(target_manager_build_result, tuple)
+    target_manager_image, _ = target_manager_build_result
+
+    vws_build_result = client.images.build(
         path=str(repository_root),
         dockerfile=str(vws_dockerfile),
         tag=vws_tag,
     )
-    vwq_image, _ = client.images.build(
+    assert isinstance(vws_build_result, tuple)
+    vws_image, _ = vws_build_result
+    vwq_build_result = client.images.build(
         path=str(repository_root),
         dockerfile=str(vwq_dockerfile),
         tag=vwq_tag,
     )
+    assert isinstance(vwq_build_result, tuple)
+    vwq_image, _ = vwq_build_result
 
     database = VuforiaDatabase()
     target_manager_container_name = "vws-mock-target-manager-" + random
@@ -178,9 +188,16 @@ def test_build_and_run(
         },
     )
 
+    assert isinstance(target_manager_container, Container)
+    assert isinstance(vws_container, Container)
+    assert isinstance(vwq_container, Container)
     for container in (target_manager_container, vws_container, vwq_container):
         container.reload()
 
+    assert isinstance(target_manager_container.attrs, dict)
+    target_manager_port_attrs = target_manager_container.attrs[
+        "NetworkSettings"
+    ]["Ports"]
     target_manager_port_attrs = target_manager_container.attrs[
         "NetworkSettings"
     ]["Ports"]
@@ -189,10 +206,12 @@ def test_build_and_run(
         "HostPort"
     ]
 
+    assert isinstance(vws_container.attrs, dict)
     vws_port_attrs = vws_container.attrs["NetworkSettings"]["Ports"]
     vws_host_ip = vws_port_attrs["5000/tcp"][0]["HostIp"]
     vws_host_port = vws_port_attrs["5000/tcp"][0]["HostPort"]
 
+    assert isinstance(vwq_container.attrs, dict)
     vwq_port_attrs = vwq_container.attrs["NetworkSettings"]["Ports"]
     vwq_host_ip = vwq_port_attrs["5000/tcp"][0]["HostIp"]
     vwq_host_port = vwq_port_attrs["5000/tcp"][0]["HostPort"]