ok-script 0.0.95__tar.gz → 0.0.97__tar.gz

{ok-script-0.0.95 → ok-script-0.0.97}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ok-script
-Version: 0.0.95
+Version: 0.0.97
 Summary: Automation with Computer Vision for Python
 Home-page: https://github.com/ok-oldking/ok-script
 Author: ok-oldking

{ok-script-0.0.95 → ok-script-0.0.97}/ok/OK.py

@@ -114,7 +114,7 @@ class OK:
                                           trigger_tasks=self.config.get('trigger_tasks', []),
                                           scenes=self.config['scenes'],
                                           feature_set=self.feature_set,
-                                          config_folder=self.config.get("config_folder"))
+                                          config_folder=self.config.get("config_folder"), debug=self.debug)
 
         ok.gui.executor = self.task_executor
 

{ok-script-0.0.95 → ok-script-0.0.97}/ok/capture/HwndWindow.py

@@ -29,7 +29,6 @@ class HwndWindow:
     left_cut = 0
     ext_left_bounds = 0  # for BitBlt
     ext_top_bounds = 0  # for BitBlt
-    window_change_listeners = []
     frame_aspect_ratio = 0
     hwnd = None
     frame_width = 0
@@ -244,13 +243,17 @@ def enum_child_windows(biggest, frame_aspect_ratio):
 
 def get_exe_by_hwnd(hwnd):
     # Get the process ID associated with the window
-    _, pid = win32process.GetWindowThreadProcessId(hwnd)
+    try:
+        _, pid = win32process.GetWindowThreadProcessId(hwnd)
 
-    # Get the process name and executable path
-    if pid > 0:
-        process = psutil.Process(pid)
-        return process.name(), process.exe(), process.cmdline()
-    else:
+        # Get the process name and executable path
+        if pid > 0:
+            process = psutil.Process(pid)
+            return process.name(), process.exe(), process.cmdline()
+        else:
+            return None, None, None
+    except Exception as e:
+        logger.error('get_exe_by_hwnd error', e)
         return None, None, None
 
 

{ok-script-0.0.95 → ok-script-0.0.97}/ok/capture/adb/DeviceManager.py

@@ -137,11 +137,12 @@ class DeviceManager:
         from ok.alas.emulator_windows import EmulatorManager
         manager = EmulatorManager()
         installed_emulators = manager.all_emulator_instances
+        logger.info(f'installed emulators {installed_emulators}')
         for emulator in installed_emulators:
             preferred = self.get_preferred_device()
             if current and preferred is not None and preferred['imei'] != emulator.name:
                 logger.debug(f"refresh current only skip others {preferred['imei']} != {emulator.name}")
-                break
+                continue
             adb_device = self.adb_connect(emulator.serial)
             width, height = self.get_resolution(adb_device) if adb_device is not None else 0, 0
             name, hwnd, full_path, x, y, width, height = find_hwnd(None,

{ok-script-0.0.95 → ok-script-0.0.97}/ok/capture/windows/WindowsGraphicsCaptureMethod.py

@@ -51,11 +51,12 @@ class WindowsGraphicsCaptureMethod(BaseWindowsCaptureMethod):
 
     def convert_dx_frame(self):
         frame = self.last_dx_frame
-        self.last_dx_frame = None
         if not frame:
-            logger.warning('convert_dx_frame self.last_dx_frame is none')
+            # logger.warning('convert_dx_frame self.last_dx_frame is none')
             return None
-
+        start = time.time()
+        dx_time = self.last_frame_time
+        self.last_dx_frame = None
         need_reset_framepool = False
         if frame.ContentSize.Width != self.last_size.Width or frame.ContentSize.Height != self.last_size.Height:
             need_reset_framepool = True
@@ -94,6 +95,7 @@ class WindowsGraphicsCaptureMethod(BaseWindowsCaptureMethod):
                                         (desc.Height, mapinfo.RowPitch // 4, 4))[
                   :, :desc.Width].copy()
             self.immediatedc.Unmap(cputex, 0)
+            # logger.debug(f'frame latency {(time.time() - start):.3f} {(time.time() - dx_time):.3f}')
             return img
         except OSError as e:
             if e.winerror == d3d11.DXGI_ERROR_DEVICE_REMOVED or e.winerror == d3d11.DXGI_ERROR_DEVICE_RESET:
@@ -150,7 +152,7 @@ class WindowsGraphicsCaptureMethod(BaseWindowsCaptureMethod):
                 self.evtoken = item.add_Closed(delegate)
                 self.frame_pool = Direct3D11CaptureFramePool.CreateFreeThreaded(self.rtdevice,
                                                                                 DirectXPixelFormat.B8G8R8A8UIntNormalized,
-                                                                                2, item.Size)
+                                                                                1, item.Size)
                 self.session = self.frame_pool.CreateCaptureSession(item)
                 pool = self.frame_pool
                 pool.add_FrameArrived(

{ok-script-0.0.95 → ok-script-0.0.97}/ok/capture/windows/dump.py

@@ -27,12 +27,10 @@ def dump_threads():
     output_file = get_path_relative_to_exe(os.path.join('logs', "thread_dumps.txt"))
     print(f'Dumping threads to {output_file}')
 
-    with open(output_file, "w") as file:
-        # Open the file with UTF-8 encoding
-        with open('file.txt', 'w', encoding='utf-8') as f:
-            f.write("\n\n".join(thread_dumps))
+    with open(output_file, "w", encoding='utf-8') as f:
+        f.write("\n\n".join(thread_dumps))
 
-        # subprocess.Popen(r'explorer /select,"{}"'.format(output_file))
+    # subprocess.Popen(r'explorer /select,"{}"'.format(output_file))
 
 
 def kill_dump():

{ok-script-0.0.95 → ok-script-0.0.97}/ok/color/Color.py

@@ -104,6 +104,11 @@ def find_color_rectangles(image, color_range, min_width, min_height, max_width=-
         raise ValueError("Image not found or path is incorrect")
     if box is not None:
         image = image[box.y:box.y + box.height, box.x:box.x + box.width, :3]
+        x_offset = box.x
+        y_offset = box.y
+    else:
+        x_offset = 0
+        y_offset = 0
 
     # Convert color range to BGR format for OpenCV
     lower_bound = np.array([color_range['b'][0], color_range['g'][0], color_range['r'][0]], dtype="uint8")
@@ -139,7 +144,7 @@ def find_color_rectangles(image, color_range, min_width, min_height, max_width=-
             if percent >= threshold:
                 # Store the result
                 results.append(
-                    Box(x, y, w, h, confidence=percent)
+                    Box(x + x_offset, y + y_offset, w, h, confidence=percent)
                 )
 
     return results

{ok-script-0.0.95 → ok-script-0.0.97}/ok/feature/Box.py

@@ -95,6 +95,9 @@ class Box:
         return Box(self.x + x_offset, self.y + y_offset, self.width + width_offset, self.height + height_offset,
                    self.confidence, name or self.name)
 
+    def crop_frame(self, frame):
+        return frame[self.y:self.y + self.height, self.x:self.x + self.width]
+
     def center(self):
         return self.x + self.width / 2, self.y + self.height / 2
 

{ok-script-0.0.95 → ok-script-0.0.97}/ok/feature/CompressCoco.py

@@ -5,6 +5,7 @@ import sys
 import cv2
 import numpy as np
 from PIL import Image
+from PIL.PngImagePlugin import PngInfo
 
 from ok.feature.FeatureSet import read_from_json
 
@@ -51,19 +52,22 @@ def compress_coco(coco_json, canny_upper, canny_lower) -> None:
             }
 
             # Save the image with metadata
-            save_image_with_metadata(background, image_path, metadata)
+            save_image_with_metadata(background, image_path)
 
 
-def save_image_with_metadata(image, image_path, metadata):
+def save_image_with_metadata(image, image_path):
     # Convert OpenCV image (numpy array) to PIL Image
     pil_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
 
+    metadata = PngInfo()
+
     # Add metadata
-    for key, value in metadata.items():
-        pil_image.info[key] = value
+    metadata.add_text('ok_compressed', '1')
+    metadata.add_text("Author", "ok_compress")
+    metadata.add_text("Description", "This is a sample image")
 
     # Save the image with metadata
-    pil_image.save(image_path, 'PNG', optimize=True)
+    pil_image.save(image_path, 'PNG', optimize=True, pnginfo=metadata)
 
 
 if __name__ == '__main__':

{ok-script-0.0.95 → ok-script-0.0.97}/ok/feature/FeatureSet.py

@@ -172,8 +172,12 @@ class FeatureSet:
         if use_gray_scale:
             search_area = cv2.cvtColor(search_area, cv2.COLOR_BGR2GRAY)
             template = cv2.cvtColor(template, cv2.COLOR_BGR2GRAY)
-        # if category_name.startswith('edge_'):
-        #     search_area =
+        if category_name.startswith('edge_'):
+            search_area = cv2.cvtColor(search_area, cv2.COLOR_BGR2GRAY)
+            search_area = cv2.Canny(search_area, self.canny_lower, self.canny_higher)
+        elif category_name.startswith('gray_'):
+            search_area = cv2.cvtColor(search_area, cv2.COLOR_BGR2GRAY)
+
         result = cv2.matchTemplate(search_area, template, cv2.TM_CCOEFF_NORMED)
 
         # Define a threshold for acceptable matches
@@ -213,7 +217,7 @@ def read_from_json(coco_json, width=-1, height=-1, canny_lower=50, canny_upper=1
         image_path = str(os.path.join(coco_folder, file_name))
         if ok_compressed is None:
             image = Image.open(image_path)
-            ok_compressed = 'ok_compressed' in image.info.items()
+            ok_compressed = 'ok_compressed' in image.info.keys()
         whole_image = cv2.imread(image_path)
         if whole_image is None:
             logger.error(f'Could not read image {image_path}')
@@ -251,13 +255,14 @@ def read_from_json(coco_json, width=-1, height=-1, canny_lower=50, canny_upper=1
             if category_name in feature_dict:
                 raise ValueError(f"Multiple boxes found for category {category_name}")
             if not category_name.startswith('box_'):
-                if category_name.startswith('edge_'):
-                    image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-                    image = cv2.Canny(image, canny_lower, canny_upper)
-                elif category_name.startswith('gray_'):
-                    image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+                if ok_compressed:
+                    if category_name.startswith('edge_'):
+                        image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+                        image = cv2.Canny(image, canny_lower, canny_upper)
+                    elif category_name.startswith('gray_'):
+                        image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
                 feature_dict[category_name] = Feature(image, x, y)
-            box_dict[category_name] = Box(x, y, image.shape[1], image.shape[2], name=category_name)
+            box_dict[category_name] = Box(x, y, image.shape[1], image.shape[0], name=category_name)
 
     return feature_dict, box_dict, ok_compressed
 

{ok-script-0.0.95 → ok-script-0.0.97}/ok/feature/FindFeature.py

@@ -31,13 +31,24 @@ class FindFeature:
             setattr(self, feature, result)
         return ret
 
+    def wait_feature(self, feature, horizontal_variance=0, vertical_variance=0, threshold=0, wait_until_before_delay=-1,
+                     time_out=0, pre_action=None, post_action=None, use_gray_scale=False, box=None,
+                     raise_if_not_found=False):
+        return self.wait_until(
+            lambda: self.find_one(feature, horizontal_variance, vertical_variance, threshold,
+                                  use_gray_scale=use_gray_scale, box=box),
+            time_out,
+            pre_action,
+            post_action, wait_until_before_delay, raise_if_not_found)
+
     def wait_click_feature(self, feature, horizontal_variance=0, vertical_variance=0, threshold=0, relative_x=0.5,
                            relative_y=0.5,
-                           time_out=0, pre_action=None, post_action=None):
-        box = self.wait_until(lambda: self.find_one(feature, horizontal_variance, vertical_variance, threshold),
-                              time_out,
-                              pre_action,
-                              post_action)
+                           time_out=0, pre_action=None, post_action=None, box=None, raise_if_not_found=True):
+        box = self.wait_until(
+            lambda: self.find_one(feature, horizontal_variance, vertical_variance, threshold, box=box),
+            time_out,
+            pre_action,
+            post_action, raise_if_not_found)
         if box is not None:
             self.click_box(box, relative_x, relative_y)
             return True

{ok-script-0.0.95 → ok-script-0.0.97}/ok/firebase/Analytics.py

@@ -103,14 +103,14 @@ class Analytics:
 
         params.update(self.user_properties)
 
-        logger.debug(f'send report {params}')
+        logger.info(f'send report {params}')
 
         response = requests.post(self.report_url, json=params)
 
         if response.status_code == 200:
             logger.debug(f'Successfully send report')
         else:
-            logger.debug(f'Failed to send event: {response.status_code} - {response.text}')
+            logger.error(f'Failed to send event: {response.status_code} - {response.text}')
 
         self._handler.post(self.send_alive, 3600)
 

{ok-script-0.0.95 → ok-script-0.0.97}/ok/gui/App.py

@@ -47,7 +47,6 @@ class App:
         self.loading_window = None
         self.overlay_window = None
         self.main_window = None
-        self.log_window = None
         self.exit_event = exit_event
         self.icon = QIcon(self.config.get('gui_icon') or ":/icon/icon.ico")
         if self.config.get('analytics'):
@@ -98,12 +97,6 @@ class App:
         if self.overlay and ok.gui.device_manager.hwnd is not None:
             self.overlay_window = OverlayWindow(ok.gui.device_manager.hwnd)
 
-        if self.config.get('debug'):
-            from ok.gui.debug.LogWindow import LogWindow
-            self.log_window = LogWindow(self.config)
-            self.log_window.show()
-            logger.debug('showing log_window')
-
         self.main_window = MainWindow(self.config, self.icon, self.title, self.version, self.overlay, self.about,
                                       self.exit_event)
         # Set the window title here

{ok-script-0.0.95 → ok-script-0.0.97}/ok/gui/debug/DebugTab.py

@@ -29,6 +29,9 @@ class DebugTab(Tab):
 
         self.config = Config({'target_task': "", 'target_images': [], 'target_function': ""},
                              app_config.get('config_folder'), 'debug')
+        self.log_window_config = Config({'width': 800, 'height': 300, 'x': 0, 'y': 0, 'keyword': '',
+                                         'level': 'ALL', 'show': True},
+                                        app_config.get('config_folder'), 'log_window')
         tool_widget = QWidget()
         layout = FlowLayout(tool_widget, False)
 
@@ -53,7 +56,14 @@ class DebugTab(Tab):
         ocr_button.clicked.connect(lambda: self.handler.post(self.ocr))
         layout.addWidget(ocr_button)
 
-        open_log_folder = PushButton(self.tr("Open Logs"))
+        self.log_window = None
+
+        log_window_button = PushButton(self.tr("Open Logs"))
+        log_window_button.clicked.connect(self.toggle_log_window)
+        layout.addWidget(log_window_button)
+
+        if self.log_window_config.get('show'):
+            self.toggle_log_window()
 
         call_task_widget = QWidget()
         call_task_container = QVBoxLayout(call_task_widget)
@@ -98,6 +108,22 @@ class DebugTab(Tab):
         self.handler.post(self.bind_hot_keys)
         self.handler.post(self.check_hotkey, 0.1)
 
+        ok.gui.app.app.aboutToQuit.connect(self.unregister)
+
+    def toggle_log_window(self):
+        if self.log_window is None:
+            from ok.gui.debug.LogWindow import LogWindow
+            self.log_window = LogWindow(self.log_window_config)
+            self.log_window.show()
+            self.log_window_config['show'] = True
+        elif self.log_window.isVisible():
+            self.log_window.hide()
+            self.log_window_config['show'] = False
+        else:
+            self.log_window_config['show'] = True
+            self.log_window.show()
+        logger.debug('showing log_window')
+
     def check_hotkey(self):
         # Example event type, you should use the appropriate QEvent.Type for your case
         msg = wintypes.MSG()
@@ -110,7 +136,7 @@ class DebugTab(Tab):
                     logger.debug('dumping threads')
                     dump_threads()
                 elif msg.wParam == 2:
-                    self.handler.post(self.capture)
+                    self.handler.post(capture)
 
         # Repost the check_hotkey method to be called after 100 ms
         self.handler.post(self.check_hotkey, 0.1)
@@ -127,8 +153,10 @@ class DebugTab(Tab):
             logger.debug("Failed to register hotkey for Alt+Ctrl+S")
         logger.debug('bind_hot_keys')
 
-    def __del__(self):
+    @staticmethod
+    def unregister():
         # Unregister the hotkeys
+        logger.debug('Unregister the hotkeys')
         windll.user32.UnregisterHotKey(None, 1)
         windll.user32.UnregisterHotKey(None, 2)
 

{ok-script-0.0.95 → ok-script-0.0.97}/ok/gui/debug/LogWindow.py

@@ -4,7 +4,6 @@ from PySide6.QtWidgets import QWidget, \
     QStyledItemDelegate, QListView, QVBoxLayout, QHBoxLayout, QAbstractItemView, QComboBox, QLineEdit, QLabel, \
     QPushButton
 
-from ok.config.Config import Config
 from ok.gui.Communicate import communicate
 
 LOG_BG_TRANS = 80
@@ -70,9 +69,6 @@ class LogModel(QAbstractListModel):
         self.do_filter_logs()
         self.endInsertRows()
 
-        # self.layoutChanged.emit()
-        # self.log_list.scrollToBottom()
-
     def do_filter_logs(self):
 
         keyword = self.current_keyword.lower()
@@ -109,15 +105,13 @@ level_severity = {v: k for k, v in log_levels.items()}
 
 
 class LogWindow(QWidget):
-    def __init__(self, app_config):
+    def __init__(self, config):
         super().__init__()
         self.setWindowTitle('Log Viewer')
         self.setWindowFlags(Qt.FramelessWindowHint | Qt.WindowStaysOnTopHint | Qt.Tool)
         self.setAttribute(Qt.WA_TranslucentBackground)
 
-        self.config = Config({'width': 800, 'height': 300, 'x': 0, 'y': 0,
-                              'level': 'ALL'},
-                             app_config.get('config_folder'), 'log_window')
+        self.config = config
         self.setGeometry(self.config['x'], self.config['y'], self.config['width'], self.config['height'])
 
         self.old_pos = None
@@ -139,6 +133,8 @@ class LogWindow(QWidget):
 
         self.keyword_filter = QLineEdit()
         self.keyword_filter.setPlaceholderText("Filter by keyword")
+        if keyword := self.config.get('keyword'):
+            self.keyword_filter.setText(keyword)
         self.keyword_filter.textChanged.connect(self.filter_logs)
 
         self.drag_button = QLabel(self.tr("Drag"))
@@ -162,6 +158,11 @@ class LogWindow(QWidget):
         self.log_list.setModel(self.log_model)
 
         communicate.log.connect(self.add_log)
+        self.filter_logs()
+
+    def close(self):
+        super().close()
+        self.config['show'] = False
 
     def add_log(self, level_no, message):
         # print('add_log', level_no, message)
@@ -170,8 +171,9 @@ class LogWindow(QWidget):
 
     def filter_logs(self):
         level = self.level_filter.currentText()
-        self.config['level'] = level
         keyword = self.keyword_filter.text()
+        self.config['keyword'] = keyword
+        self.config['level'] = level
         self.log_model.filter_logs(level, keyword)
 
     def mousePressEvent(self, event):

ok-script-0.0.97/ok/gui/tasks/OneTimeTaskTab.py

@@ -0,0 +1,20 @@
+from typing import List
+
+import ok.gui
+from ok.gui.tasks.TaskCard import TaskCard
+from ok.gui.tasks.TaskTab import TaskTab
+from ok.logging.Logger import get_logger
+
+logger = get_logger(__name__)
+
+
+class OneTimeTaskTab(TaskTab):
+    def __init__(self):
+        super().__init__()
+
+        for task in ok.gui.executor.onetime_tasks:
+            task_card = TaskCard(task, True)
+            self.addWidget(task_card)
+
+    def in_current_list(self, task):
+        return task in ok.gui.executor.onetime_tasks

{ok-script-0.0.95 → ok-script-0.0.97}/ok/gui/tasks/TaskCard.py

@@ -7,19 +7,19 @@ from ok.gui.Communicate import communicate
 from ok.gui.common.OKIcon import OKIcon
 from ok.gui.tasks.ConfigItemFactory import config_widget
 from ok.task.BaseTask import BaseTask
-from ok.task.TriggerTask import TriggerTask
 
 
 class TaskCard(ExpandSettingCard):
-    def __init__(self, task: BaseTask):
+    def __init__(self, task: BaseTask, onetime):
         super().__init__(FluentIcon.INFO, task.name, task.description)
         self.task = task
+        self.onetime = onetime
         if task.default_config:
             self.reset_config = PushButton(FluentIcon.CANCEL, self.tr("Reset Config"), self)
             self.addWidget(self.reset_config)
             self.reset_config.clicked.connect(self.reset_clicked)
 
-        if not isinstance(task, TriggerTask):
+        if onetime:
             self.task_buttons = TaskButtons(self.task)
             self.addWidget(self.task_buttons)
         else:
@@ -49,10 +49,10 @@ class TaskCard(ExpandSettingCard):
 
     def update_buttons(self, task):
         if task == self.task:
-            if isinstance(task, TriggerTask):
-                self.enable_button.setChecked(task.enabled)
-            else:
+            if self.onetime:
                 self.task_buttons.update_buttons()
+            else:
+                self.enable_button.setChecked(task.enabled)
 
     def check_changed(self, checked):
         if checked:

ok-script-0.0.95/ok/gui/tasks/OneTimeTaskTab.py → ok-script-0.0.97/ok/gui/tasks/TaskTab.py

@@ -5,7 +5,6 @@ from PySide6.QtWidgets import QTableWidgetItem
 
 import ok.gui
 from ok.gui.Communicate import communicate
-from ok.gui.tasks.TaskCard import TaskCard
 from ok.gui.tasks.TooltipTableWidget import TooltipTableWidget
 from ok.gui.widget.Tab import Tab
 from ok.gui.widget.UpdateConfigWidgetItem import value_to_string
@@ -14,7 +13,7 @@ from ok.logging.Logger import get_logger
 logger = get_logger(__name__)
 
 
-class OneTimeTaskTab(Tab):
+class TaskTab(Tab):
     def __init__(self):
         super().__init__()
 
@@ -23,10 +22,6 @@ class OneTimeTaskTab(Tab):
         self.task_info_container = self.addCard(self.tr("Choose Window"), self.task_info_table)
         self.addWidget(self.task_info_container)
 
-        for task in ok.gui.executor.onetime_tasks:
-            task_card = TaskCard(task)
-            self.addWidget(task_card)
-
         self.task_info_labels = [self.tr('Info'), self.tr('Value')]
         self.task_info_table.setColumnCount(len(self.task_info_labels))  # Name and Value
         self.task_info_table.setHorizontalHeaderLabels(self.task_info_labels)
@@ -38,9 +33,12 @@ class OneTimeTaskTab(Tab):
     def task_update(self, task):
         self.update_info_table()
 
+    def in_current_list(self, task):
+        return True
+
     def update_info_table(self):
         task = ok.gui.executor.current_task
-        if task is None:
+        if task is None or not self.in_current_list(task):
             self.task_info_container.hide()
         else:
             self.task_info_container.show()

{ok-script-0.0.95 → ok-script-0.0.97}/ok/gui/tasks/TriggerTaskTab.py

@@ -2,16 +2,19 @@ from typing import List
 
 import ok.gui
 from ok.gui.tasks.TaskCard import TaskCard
-from ok.gui.widget.Tab import Tab
+from ok.gui.tasks.TaskTab import TaskTab
 from ok.logging.Logger import get_logger
 
 logger = get_logger(__name__)
 
 
-class TriggerTaskTab(Tab):
+class TriggerTaskTab(TaskTab):
     def __init__(self):
         super().__init__()
 
         for task in ok.gui.executor.trigger_tasks:
-            task_card = TaskCard(task)
+            task_card = TaskCard(task, False)
             self.addWidget(task_card)
+
+    def in_current_list(self, task):
+        return task in ok.gui.executor.trigger_tasks

{ok-script-0.0.95 → ok-script-0.0.97}/ok/interaction/Win32Interaction.py

@@ -20,7 +20,7 @@ class Win32Interaction(BaseInteraction):
         if not is_admin():
             logger.error(f"You must be an admin to use Win32Interaction")
 
-    def send_key(self, key, down_time=0.02):
+    def send_key(self, key, down_time=0.01):
         if not self.capture.clickable():
             logger.error(f"can't click on {key}, because capture is not clickable")
             return

{ok-script-0.0.95 → ok-script-0.0.97}/ok/ocr/OCR.py

@@ -12,7 +12,7 @@ logger = get_logger(__name__)
 
 class OCR:
     executor = None
-    ocr_default_threshold = 0.5
+    ocr_default_threshold = 0.8
     ocr_target_height = 0
 
     def ocr(self, x=0, y=0, to_x=1, to_y=1, match: str | List[str] | Pattern[str] | List[Pattern[str]] | None = None,
@@ -75,7 +75,7 @@ class OCR:
             communicate.emit_draw_box("ocr", detected_boxes, "red")
             communicate.emit_draw_box("ocr_zone", box, "blue")
             logger.debug(
-                f"ocr_zone {box} found result: {len(detected_boxes)}) time: {(time.time() - start):.2f} scale_factor: {scale_factor:.2f}")
+                f"ocr_zone {box} found result: {detected_boxes}) time: {(time.time() - start):.2f} scale_factor: {scale_factor:.2f}")
             if not detected_boxes and ocr_boxes:
                 logger.info(f'ocr detected but no match: {match} {ocr_boxes}')
             return sort_boxes(detected_boxes)

{ok-script-0.0.95 → ok-script-0.0.97}/ok/task/ExecutorOperation.py

@@ -1,4 +1,3 @@
-import queue
 from typing import List
 
 from ok.color.Color import calculate_color_percentage
@@ -26,10 +25,9 @@ class ExecutorOperation:
         else:
             return False
 
-    def new_queue(self):
-        q = queue.Queue()
-        self.executor.exit_event.bind_queue(q)
-        return q
+    @property
+    def debug(self):
+        return self.executor.debug
 
     def is_scene(self, the_scene):
         return isinstance(self.executor.current_scene, the_scene)
@@ -140,7 +138,7 @@ class ExecutorOperation:
     def move_relative(self, x, y):
         self.executor.interaction.move_relative(x, y)
 
-    def click_box(self, box: Box | List[Box] = None, relative_x=0.5, relative_y=0.5, raise_if_not_found=True,
+    def click_box(self, box: Box | List[Box] = None, relative_x=0.5, relative_y=0.5, raise_if_not_found=False,
                   move_back=False):
         if isinstance(box, list):
             if len(box) > 0:
@@ -158,7 +156,6 @@ class ExecutorOperation:
         return self.executor.wait_scene(scene_type, time_out, pre_action, post_action)
 
     def sleep(self, timeout):
-        self.executor.reset_scene()
         self.executor.sleep(timeout)
 
     def send_key(self, key, down_time=0.02):
@@ -176,10 +173,12 @@ class ExecutorOperation:
         self.executor.reset_scene()
         self.executor.interaction.send_key_up(key)
 
-    def wait_until(self, condition, time_out=0, pre_action=None, post_action=None, wait_until_before_delay=0):
-        return self.executor.wait_condition(condition, time_out, pre_action, post_action, wait_until_before_delay)
+    def wait_until(self, condition, time_out=0, pre_action=None, post_action=None, wait_until_before_delay=-1,
+                   raise_if_not_found=False):
+        return self.executor.wait_condition(condition, time_out, pre_action, post_action, wait_until_before_delay,
+                                            raise_if_not_found)
 
-    def wait_click_box(self, condition, time_out=0, pre_action=None, post_action=None, raise_if_not_found=True):
+    def wait_click_box(self, condition, time_out=0, pre_action=None, post_action=None, raise_if_not_found=False):
         target = self.wait_until(condition, time_out, pre_action, post_action)
         self.click_box(box=target, raise_if_not_found=raise_if_not_found)
         return target