Use a variable for image type in src/

Author: adamtheturtle

src/vws/query.py

@@ -28,8 +28,10 @@
 from vws.include_target_data import CloudRecoIncludeTargetData
 from vws.reports import QueryResult, TargetData
 
+_ImageType = io.BytesIO | io.BufferedRandom
 
-def _get_image_data(image: io.BytesIO | io.BufferedRandom) -> bytes:
+
+def _get_image_data(image: _ImageType) -> bytes:
     """Get the data of an image file."""
     original_tell = image.tell()
     image.seek(0)
@@ -61,7 +63,7 @@ def __init__(
 
     def query(
         self,
-        image: io.BytesIO | io.BufferedRandom,
+        image: _ImageType,
         max_num_results: int = 1,
         include_target_data: CloudRecoIncludeTargetData = (
             CloudRecoIncludeTargetData.TOP

src/vws/vws.py

@@ -8,7 +8,6 @@
 import time
 from datetime import date
 from http import HTTPMethod, HTTPStatus
-from io import BufferedRandom
 from urllib.parse import urljoin
 
 import requests
@@ -48,8 +47,10 @@
 
 from .exceptions.response import Response
 
+_ImageType = io.BytesIO | io.BufferedRandom
 
-def _get_image_data(image: io.BytesIO | BufferedRandom) -> bytes:
+
+def _get_image_data(image: _ImageType) -> bytes:
     """Get the data of an image file."""
     original_tell = image.tell()
     image.seek(0)
@@ -234,7 +235,7 @@ def add_target(
         self,
         name: str,
         width: float,
-        image: io.BytesIO | io.BufferedRandom,
+        image: _ImageType,
         application_metadata: str | None,
         *,
         active_flag: bool,
@@ -617,7 +618,7 @@ def update_target(
         target_id: str,
         name: str | None = None,
         width: float | None = None,
-        image: io.BytesIO | io.BufferedRandom | None = None,
+        image: _ImageType | None = None,
         active_flag: bool | None = None,
         application_metadata: str | None = None,
     ) -> None: