kotonebot 0.5.0__py3-none-any.whl → 0.7.0__py3-none-any.whl

kotonebot/backend/debug/entry.py

@@ -1,89 +1,89 @@
-import os
-import runpy
-import shutil
-import argparse
-import importlib
-from pathlib import Path
-from threading import Thread
-
-from . import debug
-from kotonebot import logging
-from kotonebot.backend.context import init_context
-
-logger = logging.getLogger(__name__)
-
-def _task_thread(task_module: str):
-    """任务线程。"""
-    runpy.run_module(task_module, run_name="__main__")
-
-def _parse_args():
-    """解析命令行参数。"""
-    parser = argparse.ArgumentParser(description='KotoneBot visual debug tool')
-    parser.add_argument(
-        '-s', '--save', 
-        help='Save dump image and results to the specified folder',
-        type=str,
-        metavar='PATH'
-    )
-    parser.add_argument(
-        '-c', '--clear',
-        help='Clear the dump folder before running',
-        action='store_true'
-    )
-    parser.add_argument(
-        '-t', '--config-type',
-        help='The full path of the config data type. e.g. `kotonebot.tasks.common.BaseConfig`',
-        type=str,
-        metavar='TYPE',
-        required=True
-    )
-    parser.add_argument(
-        'input_module',
-        help='The module to run'
-    )
-    return parser.parse_args()
-
-def _start_task_thread(module: str):
-    """启动任务线程。"""
-    thread = Thread(target=_task_thread, args=(module,))
-    thread.start()
-
-if __name__ == "__main__":
-    args = _parse_args()
-    debug.enabled = True
-    
-    # 设置保存路径
-    if args.save:
-        save_path = Path(args.save)
-        debug.auto_save_to_folder = str(save_path)
-        if not os.path.exists(save_path):
-            os.makedirs(save_path)
-    if args.clear:
-        if debug.auto_save_to_folder:
-            try:
-                logger.info(f"Removing {debug.auto_save_to_folder}")
-                shutil.rmtree(debug.auto_save_to_folder)
-            except PermissionError:
-                logger.warning(f"Failed to remove {debug.auto_save_to_folder}. Trying to remove all contents instead.")
-                for root, dirs, files in os.walk(debug.auto_save_to_folder):
-                    for file in files:
-                        try:
-                            os.remove(os.path.join(root, file))
-                        except PermissionError:
-                            raise
-                
-
-    # 初始化上下文
-    module_name, class_name = args.config_type.rsplit('.', 1)
-    class_ = importlib.import_module(module_name).__getattribute__(class_name)
-    init_context(config_type=class_)
-    
-    # 启动服务器
-    from .server import app
-    import uvicorn
-    
-    # 启动任务线程
-    _start_task_thread(args.input_module)
-    
-    # 启动服务器
-    uvicorn.run(app, host="127.0.0.1", port=8000, log_level='critical' if debug.hide_server_log else None)
+import os
+import runpy
+import shutil
+import argparse
+import importlib
+from pathlib import Path
+from threading import Thread
+
+from . import debug
+from kotonebot import logging
+from kotonebot.backend.context import init_context
+
+logger = logging.getLogger(__name__)
+
+def _task_thread(task_module: str):
+    """任务线程。"""
+    runpy.run_module(task_module, run_name="__main__")
+
+def _parse_args():
+    """解析命令行参数。"""
+    parser = argparse.ArgumentParser(description='KotoneBot visual debug tool')
+    parser.add_argument(
+        '-s', '--save', 
+        help='Save dump image and results to the specified folder',
+        type=str,
+        metavar='PATH'
+    )
+    parser.add_argument(
+        '-c', '--clear',
+        help='Clear the dump folder before running',
+        action='store_true'
+    )
+    parser.add_argument(
+        '-t', '--config-type',
+        help='The full path of the config data type. e.g. `kotonebot.tasks.common.BaseConfig`',
+        type=str,
+        metavar='TYPE',
+        required=True
+    )
+    parser.add_argument(
+        'input_module',
+        help='The module to run'
+    )
+    return parser.parse_args()
+
+def _start_task_thread(module: str):
+    """启动任务线程。"""
+    thread = Thread(target=_task_thread, args=(module,))
+    thread.start()
+
+if __name__ == "__main__":
+    args = _parse_args()
+    debug.enabled = True
+    
+    # 设置保存路径
+    if args.save:
+        save_path = Path(args.save)
+        debug.auto_save_to_folder = str(save_path)
+        if not os.path.exists(save_path):
+            os.makedirs(save_path)
+    if args.clear:
+        if debug.auto_save_to_folder:
+            try:
+                logger.info(f"Removing {debug.auto_save_to_folder}")
+                shutil.rmtree(debug.auto_save_to_folder)
+            except PermissionError:
+                logger.warning(f"Failed to remove {debug.auto_save_to_folder}. Trying to remove all contents instead.")
+                for root, dirs, files in os.walk(debug.auto_save_to_folder):
+                    for file in files:
+                        try:
+                            os.remove(os.path.join(root, file))
+                        except PermissionError:
+                            raise
+                
+
+    # 初始化上下文
+    module_name, class_name = args.config_type.rsplit('.', 1)
+    class_ = importlib.import_module(module_name).__getattribute__(class_name)
+    init_context(config_type=class_)
+    
+    # 启动服务器
+    from .server import app
+    import uvicorn
+    
+    # 启动任务线程
+    _start_task_thread(args.input_module)
+    
+    # 启动服务器
+    uvicorn.run(app, host="127.0.0.1", port=8000, log_level='critical' if debug.hide_server_log else None)

kotonebot/backend/debug/mock.py

@@ -1,79 +1,79 @@
-import time
-from typing_extensions import override
-
-import cv2
-from cv2.typing import MatLike
-
-from kotonebot import sleep
-from kotonebot.client.device import Device
-
-class Video:
-    def __init__(self, path: str, fps: int):
-        self.path = path
-        self.fps = fps
-        self.paused = False
-        """是否暂停"""
-        self.__cap = cv2.VideoCapture(path)
-        self.__last_frame = None
-        self.__last_time = 0
-
-    def __iter__(self):
-        return self
-
-    def __next__(self):
-        if self.paused:
-            return self.__last_frame
-        ret, frame = self.__cap.read()
-        if not ret:
-            raise StopIteration
-        self.__last_frame = frame
-        self.__last_time = time.time()
-        if self.__last_time - time.time() < 1 / self.fps:
-            sleep(1 / self.fps)
-        return frame
-
-    def pause(self):
-        self.paused = True
-
-    def resume(self):
-        self.paused = False
-
-class MockDevice(Device):
-    def __init__(
-        self
-    ):
-        super().__init__()
-        self.__video_stream = None
-        self.__image = None
-        self.__screen_size = None
-
-    def load_video(self, path: str, fps: int):
-        self.__video_stream = Video(path, fps)
-        return self.__video_stream
-
-    def load_image(self, img: str | MatLike):
-        if isinstance(img, str):
-            self.__image = cv2.imread(img)
-        else:
-            self.__image = img
-        return self.__image
-
-    def set_screen_size(self, width: int, height: int):
-        self.__screen_size = (width, height)
-
-    @override
-    def screenshot(self):
-        if self.__image is not None:
-            return self.__image
-        elif self.__video_stream is not None:
-            return next(self.__video_stream)
-        else:
-            raise RuntimeError('No video stream loaded')
-        
-    @property
-    @override
-    def screen_size(self):
-        if self.__screen_size is not None:
-            return self.__screen_size
-        else:
+import time
+from typing_extensions import override
+
+import cv2
+from cv2.typing import MatLike
+
+from kotonebot import sleep
+from kotonebot.client.device import Device
+
+class Video:
+    def __init__(self, path: str, fps: int):
+        self.path = path
+        self.fps = fps
+        self.paused = False
+        """是否暂停"""
+        self.__cap = cv2.VideoCapture(path)
+        self.__last_frame = None
+        self.__last_time = 0
+
+    def __iter__(self):
+        return self
+
+    def __next__(self):
+        if self.paused:
+            return self.__last_frame
+        ret, frame = self.__cap.read()
+        if not ret:
+            raise StopIteration
+        self.__last_frame = frame
+        self.__last_time = time.time()
+        if self.__last_time - time.time() < 1 / self.fps:
+            sleep(1 / self.fps)
+        return frame
+
+    def pause(self):
+        self.paused = True
+
+    def resume(self):
+        self.paused = False
+
+class MockDevice(Device):
+    def __init__(
+        self
+    ):
+        super().__init__()
+        self.__video_stream = None
+        self.__image = None
+        self.__screen_size = None
+
+    def load_video(self, path: str, fps: int):
+        self.__video_stream = Video(path, fps)
+        return self.__video_stream
+
+    def load_image(self, img: str | MatLike):
+        if isinstance(img, str):
+            self.__image = cv2.imread(img)
+        else:
+            self.__image = img
+        return self.__image
+
+    def set_screen_size(self, width: int, height: int):
+        self.__screen_size = (width, height)
+
+    @override
+    def screenshot(self):
+        if self.__image is not None:
+            return self.__image
+        elif self.__video_stream is not None:
+            return next(self.__video_stream)
+        else:
+            raise RuntimeError('No video stream loaded')
+        
+    @property
+    @override
+    def screen_size(self):
+        if self.__screen_size is not None:
+            return self.__screen_size
+        else:
             raise RuntimeError('No screen size set')