Centralized qWait typing fix

Author: Avasam

src/AutoSplit.py

@@ -20,7 +20,8 @@
 
 import certifi
 import cv2
-from PyQt6 import QtCore, QtGui, QtTest
+from PyQt6 import QtCore, QtGui
+from PyQt6.QtTest import QTest
 from PyQt6.QtWidgets import QApplication, QFileDialog, QMainWindow, QMessageBox, QWidget
 from win32 import win32gui
 from AutoSplitImage import COMPARISON_RESIZE, AutoSplitImage, ImageType
@@ -343,13 +344,11 @@ def __start_image_function(self):
                     delay_time_left = start_delay - (time() - delay_start_time)
                     self.current_split_image.setText(
                         f"Delayed Before Starting:\n {seconds_remaining_text(delay_time_left)}")
-                    # Email sent to [email protected]
-                    QtTest.QTest.qWait(1)  # type: ignore
+                    QTest.qWait(1)
 
             self.start_image_status_value_label.setText("started")
             send_command(self, "start")
-            # Email sent to [email protected]
-            QtTest.QTest.qWait(int(1 / self.settings_dict["fps_limit"]))  # type: ignore
+            QTest.qWait(int(1 / self.settings_dict["fps_limit"]))
             self.start_auto_splitter()
 
     # update x, y, width, height when spinbox values are changed
@@ -605,8 +604,7 @@ def __auto_splitter(self):
 
                 # limit the number of time the comparison runs to reduce cpu usage
                 frame_interval: float = 1 / self.settings_dict["fps_limit"]
-                # Email sent to [email protected]
-                QtTest.QTest.qWait(int(frame_interval - (time() - start) % frame_interval))  # type: ignore
+                QTest.qWait(int(frame_interval - (time() - start) % frame_interval))
                 QApplication.processEvents()
 
             # comes here when threshold gets met
@@ -636,8 +634,7 @@ def __auto_splitter(self):
                         capture = self.__get_capture_for_comparison()
                         if self.__reset_if_should(capture):
                             continue
-                        # Email sent to [email protected]
-                        QtTest.QTest.qWait(1)  # type: ignore
+                        QTest.qWait(1)
 
                 self.waiting_for_split_delay = False
 
@@ -691,8 +688,7 @@ def __auto_splitter(self):
                         send_command(self, "reset")
                         self.reset()
                         continue
-                    # Email sent to [email protected]
-                    QtTest.QTest.qWait(1)  # type: ignore
+                    QTest.qWait(1)
 
         # loop breaks to here when the last image splits
         self.gui_changes_on_reset()

src/screen_region.py

@@ -9,7 +9,8 @@
 import cv2
 
 import numpy as np
-from PyQt6 import QtCore, QtGui, QtTest, QtWidgets
+from PyQt6 import QtCore, QtGui, QtWidgets
+from PyQt6.QtTest import QTest
 from win32 import win32gui
 from win32con import GA_ROOT, MAXBYTE, SM_XVIRTUALSCREEN, SM_YVIRTUALSCREEN, SM_CXVIRTUALSCREEN, SM_CYVIRTUALSCREEN
 
@@ -35,8 +36,7 @@ def select_region(autosplit: AutoSplit):
         y = selector.y()
         if width > 0 and height > 0:
             break
-        # Email sent to [email protected]
-        QtTest.QTest.qWait(1)  # type: ignore
+        QTest.qWait(1)
     del selector
 
     hwnd, window_text = __get_window_from_point(x, y)
@@ -66,8 +66,7 @@ def select_window(autosplit: AutoSplit):
         y = selector.y()
         if x and y:
             break
-        # Email sent to [email protected]
-        QtTest.QTest.qWait(1)  # type: ignore
+        QTest.qWait(1)
     del selector
 
     hwnd, window_text = __get_window_from_point(x, y)

typings/PyQt6/QtTest.pyi

@@ -0,0 +1,11 @@
+import typing
+import PyQt6.sip
+
+
+class QTest(PyQt6.sip.simplewrapper):
+    # Email sent to [email protected]
+    @typing.overload
+    @staticmethod
+    def qWait(ms: int) -> None: ...
+    @typing.overload
+    def qWait(self, ms: int) -> None: ...