ok-script 0.0.32__tar.gz → 0.0.34__tar.gz

{ok-script-0.0.32 → ok-script-0.0.34}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ok-script
-Version: 0.0.32
+Version: 0.0.34
 Summary: Automation with Computer Vision for Python
 Home-page: https://github.com/ok-oldking/ok-script
 Author: ok-oldking
@@ -11,14 +11,14 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: Microsoft :: Windows
 Requires-Python: >=3.6
 Description-Content-Type: text/markdown
-Requires-Dist: adbutils~=2.6.2
-Requires-Dist: numpy~=1.26.4
-Requires-Dist: PyDirectInput~=1.0.4
+Requires-Dist: adbutils==2.7.1
+Requires-Dist: numpy==1.26.4
+Requires-Dist: PyDirectInput==1.0.4
 Requires-Dist: pywin32==306
-Requires-Dist: typing-extensions~=4.11.0
-Requires-Dist: PySide6~=6.7.0
-Requires-Dist: PySide6-Fluent-Widgets~=1.5.6
-Requires-Dist: opencv-python~=4.9.0.80
+Requires-Dist: typing-extensions==4.11.0
+Requires-Dist: PySide6==6.7.0
+Requires-Dist: PySide6-Fluent-Widgets==1.5.6
+Requires-Dist: opencv-python==4.9.0.80
 
 # ok-script
 

{ok-script-0.0.32 → ok-script-0.0.34}/ok/capture/adb/DeviceManager.py

@@ -1,11 +1,7 @@
-import os
-import subprocess
 import threading
 
 import adbutils
 import win32gui
-from adbutils import adb_path
-from adbutils.errors import AdbConnectionError, AdbTimeout
 
 from ok.capture.HwndWindow import HwndWindow, find_hwnd_by_title_and_exe
 from ok.capture.adb.ADBCaptureMethod import ADBCaptureMethod, do_screencap
@@ -21,19 +17,6 @@ from ok.logging.Logger import get_logger
 logger = get_logger(__name__)
 
 
-def fix_adbutils_safe_connect(self):
-    try:
-        return self._create_socket()
-    except (AdbConnectionError, AdbTimeout):
-        flags = subprocess.CREATE_NO_WINDOW if os.name == 'nt' else 0
-        subprocess.run([adb_path(), "start-server"], timeout=20.0,
-                       creationflags=flags)  # 20s should enough for adb start
-        return self._create_socket()
-
-
-adbutils._adb.AdbConnection._safe_connect = fix_adbutils_safe_connect
-
-
 class DeviceManager:
 
     def __init__(self, config_folder, hwnd_title=None, exe_name=None, debug=False, exit_event=None):

ok-script-0.0.34/ok/capture/windows/dump.py

@@ -0,0 +1,27 @@
+import sys
+import traceback
+import threading
+
+from ok.util.path import get_path_relative_to_exe
+
+def get_thread_name(thread_id):
+    for thread in threading.enumerate():
+        if thread.ident == thread_id:
+            return thread.name
+    return ""  # Return empty string if the thread name is not found
+
+def dump_threads():
+    # Get the stack trace for each thread
+    thread_dumps = []
+    for thread_id, frame in sys._current_frames().items():
+        thread_name = get_thread_name(thread_id)
+        thread_dump = f"Stack for thread {thread_id} (Name: {thread_name}):\n"
+        thread_dump += "".join(traceback.format_stack(frame))
+        thread_dumps.append(thread_dump)
+
+    # Write the thread dumps to a file
+    output_file = get_path_relative_to_exe("thread_dumps.txt")
+    print(f'Dumping threads to {output_file}')
+
+    with open(output_file, "w") as file:
+        file.write("\n\n".join(thread_dumps))

{ok-script-0.0.32 → ok-script-0.0.34}/ok/gui/MainWindow.py

@@ -1,7 +1,9 @@
 from PySide6.QtCore import QObject, Signal, Qt
+from PySide6.QtGui import QKeySequence, QShortcut
 from qfluentwidgets import FluentIcon, NavigationItemPosition, MSFluentWindow, InfoBar, InfoBarPosition, MessageBox
 
 import ok.gui
+from ok.capture.windows.dump import dump_threads
 from ok.gui.Communicate import communicate
 from ok.gui.about.AboutTab import AboutTab
 from ok.gui.start.StartTab import StartTab
@@ -48,6 +50,9 @@ class MainWindow(MSFluentWindow):
         communicate.executor_paused.connect(self.executor_paused)
         communicate.tab.connect(self.navigate_tab)
 
+        self.shortcut = QShortcut(QKeySequence("Ctrl+Alt+D"), self)
+        self.shortcut.activated.connect(dump_threads)
+
     def navigate_tab(self, index):
         if index == "start":
             self.switchTo(self.start_tab)

ok-script-0.0.34/ok/gui/about/AboutTab.py

@@ -0,0 +1,24 @@
+from PySide6.QtWidgets import QTextEdit
+from qfluentwidgets import TextEdit, PushButton
+
+import ok.gui
+from ok.capture.windows.dump import dump_threads
+from ok.gui.widget.Tab import Tab
+
+
+class AboutTab(Tab):
+    def __init__(self, text=""):
+        super().__init__()
+        # Create a QTextEdit instance
+        text_edit = TextEdit()
+        text_edit.setHtml(text)
+        text_edit.setReadOnly(True)
+
+        # Set the layout on the widget
+        self.addWidget(text_edit)
+        if ok.gui.ok.debug:
+            dump_button = PushButton(self.tr("Dump Threads(HotKey:Ctrl+Alt+D)"))
+            dump_button.clicked.connect(dump_threads)
+            self.addWidget(dump_button)
+
+

{ok-script-0.0.32 → ok-script-0.0.34}/ok/gui/tasks/StartButton.py

@@ -1,6 +1,7 @@
 from PySide6.QtCore import Slot, QPropertyAnimation
 from PySide6.QtGui import QColor
 from PySide6.QtWidgets import QPushButton
+from qfluentwidgets import PushButton
 
 import ok
 from ok.gui.Communicate import communicate
@@ -10,7 +11,7 @@ from ok.logging.Logger import get_logger
 logger = get_logger(__name__)
 
 
-class StartButton(QPushButton):
+class StartButton(PushButton):
     def __init__(self):
         super().__init__("Start")
         self.setCheckable(True)

{ok-script-0.0.32 → ok-script-0.0.34}/ok/task/TaskExecutor.py

@@ -196,14 +196,13 @@ class TaskExecutor:
         processing_time = 0
         while not self.exit_event.is_set():
             try:
-                task, cycled = self.next_task()
-                if not task:
+                self.current_task, cycled = self.next_task()
+                if not self.current_task:
                     time.sleep(1)
                     continue
-                self.current_task = task
-                if isinstance(task, OneTimeTask):
-                    task.running = True
-                    communicate.task.emit(task)
+                if isinstance(self.current_task, OneTimeTask):
+                    self.current_task.running = True
+                    communicate.task.emit(self.current_task)
                 if cycled or self._frame is None:
                     self.next_frame()
                     processing_time = 0
@@ -213,14 +212,14 @@ class TaskExecutor:
                     self.next_frame()
                 self.detect_scene()
                 start = time.time()
-                result = task.run()
+                result = self.current_task.run()
                 processing_time += time.time() - start
-                if result and isinstance(task, TriggerTask):
-                    task.trigger_count += 1
-                    communicate.task.emit(task)
-                elif isinstance(task, OneTimeTask):
-                    task.set_done()
-                    communicate.task.emit(task)
+                if result and isinstance(self.current_task, TriggerTask):
+                    self.current_task.trigger_count += 1
+                    communicate.task.emit(self.current_task)
+                elif isinstance(self.current_task, OneTimeTask):
+                    self.current_task.set_done()
+                    communicate.task.emit(self.current_task)
                 self.current_task = None
             except TaskDisabledException:
                 logger.info(f"task is disabled, go to next task")
@@ -228,13 +227,17 @@ class TaskExecutor:
                 logger.info(f"FinishedException, breaking")
                 break
             except Exception as e:
+                name = ""
+                if self.current_task is not None:
+                    self.current_task.disable()
+                    name = self.current_task.name
                 traceback.print_exc()
                 stack_trace_str = traceback.format_exc()
-                logger.error(f"{task.name} exception: {e}, traceback: {stack_trace_str}")
+                logger.error(f"{name} exception: {e}, traceback: {stack_trace_str}")
             self.current_task = None
-            if isinstance(task, OneTimeTask):
-                task.running = False
-                communicate.task.emit(task)
+            if isinstance(self.current_task, OneTimeTask):
+                self.current_task.running = False
+                communicate.task.emit(self.current_task)
         logger.debug(f'exit_event is set, destroy all tasks')
         for task in self.onetime_tasks:
             task.on_destroy()

{ok-script-0.0.32 → ok-script-0.0.34}/ok_script.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ok-script
-Version: 0.0.32
+Version: 0.0.34
 Summary: Automation with Computer Vision for Python
 Home-page: https://github.com/ok-oldking/ok-script
 Author: ok-oldking
@@ -11,14 +11,14 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: Microsoft :: Windows
 Requires-Python: >=3.6
 Description-Content-Type: text/markdown
-Requires-Dist: adbutils~=2.6.2
-Requires-Dist: numpy~=1.26.4
-Requires-Dist: PyDirectInput~=1.0.4
+Requires-Dist: adbutils==2.7.1
+Requires-Dist: numpy==1.26.4
+Requires-Dist: PyDirectInput==1.0.4
 Requires-Dist: pywin32==306
-Requires-Dist: typing-extensions~=4.11.0
-Requires-Dist: PySide6~=6.7.0
-Requires-Dist: PySide6-Fluent-Widgets~=1.5.6
-Requires-Dist: opencv-python~=4.9.0.80
+Requires-Dist: typing-extensions==4.11.0
+Requires-Dist: PySide6==6.7.0
+Requires-Dist: PySide6-Fluent-Widgets==1.5.6
+Requires-Dist: opencv-python==4.9.0.80
 
 # ok-script
 

{ok-script-0.0.32 → ok-script-0.0.34}/ok_script.egg-info/SOURCES.txt

@@ -14,6 +14,7 @@ ok/capture/adb/bin/__init__.py
 ok/capture/windows/WindowsGraphicsCaptureMethod.py
 ok/capture/windows/__init__.py
 ok/capture/windows/d3d11.py
+ok/capture/windows/dump.py
 ok/capture/windows/utils.py
 ok/capture/windows/window.py
 ok/color/Color.py

ok-script-0.0.34/ok_script.egg-info/requires.txt

@@ -0,0 +1,8 @@
+adbutils==2.7.1
+numpy==1.26.4
+PyDirectInput==1.0.4
+pywin32==306
+typing-extensions==4.11.0
+PySide6==6.7.0
+PySide6-Fluent-Widgets==1.5.6
+opencv-python==4.9.0.80

{ok-script-0.0.32 → ok-script-0.0.34}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setuptools.setup(
     name="ok-script",
-    version="0.0.32",
+    version="0.0.34",
     author="ok-oldking",
     author_email="firedcto@gmail.com",
     description="Automation with Computer Vision for Python",
@@ -20,14 +20,14 @@ setuptools.setup(
         "Operating System :: Microsoft :: Windows",
     ],
     install_requires=[
-        'adbutils~=2.6.2',
-        'numpy~=1.26.4',
-        'PyDirectInput~=1.0.4',
+        'adbutils==2.7.1',
+        'numpy==1.26.4',
+        'PyDirectInput==1.0.4',
         'pywin32==306',
-        'typing-extensions~=4.11.0',
-        'PySide6~=6.7.0',
-        'PySide6-Fluent-Widgets~=1.5.6',
-        'opencv-python~=4.9.0.80'
+        'typing-extensions==4.11.0',
+        'PySide6==6.7.0',
+        'PySide6-Fluent-Widgets==1.5.6',
+        'opencv-python==4.9.0.80'
     ],
     python_requires='>=3.6',
 )

ok-script-0.0.32/ok/gui/about/AboutTab.py

@@ -1,15 +0,0 @@
-from PySide6.QtWidgets import QTextEdit
-
-from ok.gui.widget.Tab import Tab
-
-
-class AboutTab(Tab):
-    def __init__(self, text=""):
-        super().__init__()
-        # Create a QTextEdit instance
-        text_edit = QTextEdit()
-        text_edit.setHtml(text)
-        text_edit.setReadOnly(True)
-
-        # Set the layout on the widget
-        self.addWidget(text_edit)

ok-script-0.0.32/ok_script.egg-info/requires.txt

@@ -1,8 +0,0 @@
-adbutils~=2.6.2
-numpy~=1.26.4
-PyDirectInput~=1.0.4
-pywin32==306
-typing-extensions~=4.11.0
-PySide6~=6.7.0
-PySide6-Fluent-Widgets~=1.5.6
-opencv-python~=4.9.0.80