fix: start using the cv2find.Box type instead of trying to import from the typing module

Author: welli7ngton

botcity/core/bot.py

@@ -6,7 +6,9 @@
 import subprocess
 import time
 import webbrowser
-from typing import Union, Tuple, Optional, List, Dict, Box, Generator, Any
+from typing import Union, Tuple, Optional, List, Dict, Generator, Any
+from collections import namedtuple
+
 from numpy import ndarray
 
 import pyperclip
@@ -131,7 +133,7 @@ def add_image(self, label: str, path: str) -> None:
         """
         self.state.map_images[label] = path
 
-    def get_image_from_map(self, label: str) -> None:
+    def get_image_from_map(self, label: str) -> Image.Image:
         """
         Return an image from teh state image map.
 
@@ -257,8 +259,8 @@ def _find_multiple_helper(
         region: Tuple[int, int, int, int],
         confidence: float,
         grayscale: bool,
-        needle: Union[Image.Image, ndarray, str]
-    ) -> Union[Box, None]:
+        needle: Union[Image.Image, ndarray, str],
+    ) -> Union[cv2find.Box, None]:
         ele = cv2find.locate_all_opencv(
             needle, haystack, region=region, confidence=confidence, grayscale=grayscale
         )
@@ -414,7 +416,7 @@ def find_all(
         matching: float = 0.9,
         waiting_time: int = 10000,
         grayscale: bool = False,
-    ) -> Generator[Box, Any, None]:
+    ) -> Generator[cv2find.Box, Any, None]:
         """
         Find all elements defined by label on screen until a timeout happens.
 
@@ -438,7 +440,9 @@ def find_all(
                 None if not found.
         """
 
-        def deduplicate(elems):
+        def deduplicate(
+            elems: list[Generator[cv2find.Box, Any, None]]
+        ) -> list[Generator[cv2find.Box, Any, None]]:
             def find_same(item, items):
                 x_start = item.left
                 x_end = item.left + item.width
@@ -554,7 +558,9 @@ def find_text(
             grayscale=True,
         )
 
-    def find_process(self, name: str = None, pid: str = None) -> Process:
+    def find_process(
+        self, name: Optional[str] = None, pid: Optional[str] = None
+    ) -> Union[Process, None]:
         """
         Find a process by name or PID
 
@@ -686,13 +692,14 @@ def save_screenshot(self, path: str) -> None:
         self.screenshot(path)
 
     def get_element_coords(
-        self, label: str,
+        self,
+        label: str,
         x: Optional[int] = None,
         y: Optional[int] = None,
         width: Optional[int] = None,
         height: Optional[int] = None,
         matching: float = 0.9,
-        best: bool = True
+        best: bool = True,
     ) -> Tuple[int, int] | Tuple[None, None]:
         """
         Find an element defined by label on screen and returns its coordinates.
@@ -754,7 +761,7 @@ def get_element_coords_centered(
         width: Optional[int] = None,
         height: Optional[int] = None,
         matching: float = 0.9,
-        best: bool = True
+        best: bool = True,
     ):
         """
         Find an element defined by label on screen and returns its centered coordinates.

botcity/core/cv2find.py

@@ -30,15 +30,16 @@
 OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 """
+
 import collections
 import cv2
 import numpy
 from PIL.Image import Image
 from typing import Union, Tuple, Optional, Generator, Any
 
-RUNNING_CV_2 = cv2.__version__[0] < '3'
+RUNNING_CV_2 = cv2.__version__[0] < "3"
 
-Box = collections.namedtuple('Box', 'left top width height')
+Box = collections.namedtuple("Box", "left top width height")
 
 if RUNNING_CV_2:
     LOAD_COLOR = cv2.CV_LOAD_IMAGE_COLOR
@@ -48,7 +49,9 @@
     LOAD_GRAYSCALE = cv2.IMREAD_GRAYSCALE
 
 
-def _load_cv2(img: Union[Image, numpy.ndarray, str], grayscale: bool = False) -> numpy.ndarray:
+def _load_cv2(
+    img: Union[Image, numpy.ndarray, str], grayscale: bool = False
+) -> numpy.ndarray:
     """
     TODO
     """
@@ -68,23 +71,25 @@ def _load_cv2(img: Union[Image, numpy.ndarray, str], grayscale: bool = False) ->
         else:
             img_cv = cv2.imread(img, LOAD_COLOR)
         if img_cv is None:
-            raise IOError("Failed to read %s because file is missing, "
-                          "has improper permissions, or is an "
-                          "unsupported or invalid format" % img)
+            raise IOError(
+                "Failed to read %s because file is missing, "
+                "has improper permissions, or is an "
+                "unsupported or invalid format" % img
+            )
     elif isinstance(img, numpy.ndarray):
         # don't try to convert an already-gray image to gray
         if grayscale and len(img.shape) == 3:  # and img.shape[2] == 3:
             img_cv = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
         else:
             img_cv = img
-    elif hasattr(img, 'convert'):
+    elif hasattr(img, "convert"):
         # assume its a PIL.Image, convert to cv format
-        img_array = numpy.array(img.convert('RGB'))
+        img_array = numpy.array(img.convert("RGB"))
         img_cv = img_array[:, :, ::-1].copy()  # -1 does RGB -> BGR
         if grayscale:
             img_cv = cv2.cvtColor(img_cv, cv2.COLOR_BGR2GRAY)
     else:
-        raise TypeError('expected an image filename, OpenCV numpy array, or PIL image')
+        raise TypeError("expected an image filename, OpenCV numpy array, or PIL image")
     return img_cv
 
 
@@ -95,7 +100,7 @@ def locate_all_opencv(
     limit: int = 10000,
     region: Optional[Tuple[int, int, int, int]] = None,
     step: int = 1,
-    confidence: float = 0.999
+    confidence: float = 0.999,
 ) -> Generator[Box, Any, None]:
     """
     TODO - rewrite this
@@ -116,15 +121,19 @@ def locate_all_opencv(
 
     if region:
         haystack_image = haystack_image[
-                         region[1]:region[1] + region[3],
-                         region[0]:region[0] + region[2]
-                         ]
+            region[1] : region[1] + region[3], region[0] : region[0] + region[2]
+        ]
     else:
-        region = (0, 0)  # full image; these values used in the yield statement
-    if (haystack_image.shape[0] < needle_image.shape[0] or
-            haystack_image.shape[1] < needle_image.shape[1]):
+        region = (0, 0, 0, 0)  # full image; these values used in the yield statement
+
+    if (
+        haystack_image.shape[0] < needle_image.shape[0]
+        or haystack_image.shape[1] < needle_image.shape[1]
+    ):
         # avoid semi-cryptic OpenCV error below if bad size
-        raise ValueError('needle dimension(s) exceed the haystack image or region dimensions')
+        raise ValueError(
+            "needle dimension(s) exceed the haystack image or region dimensions"
+        )
 
     if step == 2:
         confidence *= 0.95
@@ -147,6 +156,8 @@ def locate_all_opencv(
     matchy = matches[0] * step + region[1]
 
     # Order results before sending back
-    ordered = sorted(zip(matchx, matchy), key=lambda p: result[p[1]][p[0]], reverse=True)
+    ordered = sorted(
+        zip(matchx, matchy), key=lambda p: result[p[1]][p[0]], reverse=True
+    )
     for x, y in ordered:
         yield Box(x, y, needle_width, needle_height)