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

kotonebot/client/implements/adb_raw.py

@@ -1,163 +0,0 @@
-import os
-import time
-import subprocess
-import struct
-from threading import Thread, Lock
-from functools import cached_property
-from typing_extensions import override
-
-import cv2
-import numpy as np
-from cv2.typing import MatLike
-try:
-    from adbutils._utils import adb_path
-    from adbutils._device import AdbDevice as AdbUtilsDevice
-except ImportError as _e:
-    from kotonebot.errors import MissingDependencyError
-    raise MissingDependencyError(_e, 'android')
-
-from .adb import AdbImpl
-from kotonebot import logging
-
-logger = logging.getLogger(__name__)
-
-WAIT_TIMEOUT = 10
-MAX_RETRY_COUNT = 5
-SCRIPT: str = """#!/bin/sh
-while true; do
-    screencap
-    sleep 0.3
-done
-"""
-
-class AdbRawImpl(AdbImpl):
-    def __init__(self, adb_connection: AdbUtilsDevice):
-        super().__init__(adb_connection)
-        self.__worker: Thread | None = None
-        self.__process: subprocess.Popen | None = None
-        self.__data: MatLike | None = None
-        self.__retry_count = 0
-        self.__lock = Lock()
-        self.__stopping = False
-
-    def __cleanup_worker(self) -> None:
-        if self.__process:
-            try:
-                self.__process.kill()
-            except:
-                pass
-            self.__process = None
-        if self.__worker:
-            try:
-                self.__worker.join()
-            except:
-                pass
-            self.__worker = None
-        self.__data = None
-
-    def __start_worker(self) -> None:
-        self.__stopping = True
-        self.__cleanup_worker()
-        self.__stopping = False
-        self.__worker = Thread(target=self.__worker_thread_with_retry, daemon=True)
-        self.__worker.start()
-
-    def __worker_thread_with_retry(self) -> None:
-        try:
-            self.__worker_thread()
-        except Exception as e:
-            logger.error(f"Worker thread failed: {e}")
-            with self.__lock:
-                self.__retry_count += 1
-            raise
-
-    def __worker_thread(self) -> None:
-        with open('screenshot.sh', 'w', encoding='utf-8', newline='\n') as f:
-            f.write(SCRIPT)
-        self.adb.push('screenshot.sh', '/data/local/tmp/screenshot.sh')
-        self.adb.shell(f'chmod 755 /data/local/tmp/screenshot.sh')
-        os.remove('screenshot.sh')
-
-        cmd = fr'{adb_path()} -s {self.adb.serial} exec-out "sh /data/local/tmp/screenshot.sh"'
-        self.__process = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
-        
-        while not self.__stopping and self.__process.poll() is None:
-            if self.__process.stdout is None:
-                logger.error("Failed to get stdout from process")
-                continue
-            
-            # 解析 header  
-            # https://stackoverflow.com/questions/22034959/what-format-does-adb-screencap-sdcard-screenshot-raw-produce-without-p-f
-            if self.__api_level >= 26:
-                metadata = self.__process.stdout.read(16)
-                w, h, p, c = struct.unpack('<IIII', metadata)
-                # w=width, h=height, p=pixel_format, c=color_space
-                # 详见：https://android.googlesource.com/platform/frameworks/base/+/26a2b97dbe48ee45e9ae70110714048f2f360f97/cmds/screencap/screencap.cpp#209
-            else:
-                metadata = self.__process.stdout.read(12)
-                w, h, p = struct.unpack('<III', metadata)
-            if p == 1: # PixelFormat.RGBA_8888
-                channel = 4
-            else:
-                raise ValueError(f"Unsupported pixel format: {p}")
-            data_size = w * h * channel
-
-            if (data_size < 100 * 100 * 4) or (data_size > 3000 * 3000 * 4):
-                raise ValueError(f"Invaild data_size: {w}x{h}.")
-
-            # 读取图像数据
-            # logger.verbose(f"receiving image data: {w}x{h} {data_size} bytes")
-            image_data = self.__process.stdout.read(data_size)
-            if not isinstance(image_data, bytes) or len(image_data) != data_size:
-                logger.error(f"Failed to read image data, expected {data_size} bytes but got {len(image_data) if isinstance(image_data, bytes) else 'non-bytes'}")
-                raise RuntimeError("Failed to read image data")
-                
-            np_data = np.frombuffer(image_data, np.uint8)
-            np_data = np_data.reshape(h, w, channel)
-            self.__data = cv2.cvtColor(np_data, cv2.COLOR_RGBA2BGR)
-
-    @cached_property
-    def __api_level(self) -> int:
-        try:
-            output = self.adb.shell("getprop ro.build.version.sdk")
-            assert isinstance(output, str)
-            return int(output.strip())
-        except Exception as e:
-            logger.error(f"Failed to get API level: {e}")
-            return 0
-
-    @override
-    def screenshot(self) -> MatLike:
-        with self.__lock:
-            if self.__retry_count >= MAX_RETRY_COUNT:
-                raise RuntimeError(f"Maximum retry count ({MAX_RETRY_COUNT}) exceeded")
-
-            if not self.__worker or (self.__worker and not self.__worker.is_alive()):
-                self.__start_worker()
-        
-        start_time = time.time()
-        while self.__data is None:
-            time.sleep(0.01)
-            if time.time() - start_time > WAIT_TIMEOUT:
-                logger.warning("Screenshot timeout, cleaning up and restarting worker...")
-                with self.__lock:
-                    if self.__retry_count < MAX_RETRY_COUNT:
-                        self.__start_worker()
-                        start_time = time.time()  # 重置超时计时器
-                        continue
-                    else:
-                        raise RuntimeError(f"Maximum retry count ({MAX_RETRY_COUNT}) exceeded")
-            
-            # 检查 worker 是否还活着
-            if self.__worker and not self.__worker.is_alive():
-                with self.__lock:
-                    if self.__retry_count < MAX_RETRY_COUNT:
-                        logger.warning("Worker thread died, restarting...")
-                        self.__start_worker()
-                    else:
-                        raise RuntimeError(f"Maximum retry count ({MAX_RETRY_COUNT}) exceeded")
-
-        logger.verbose(f"adb raw screenshot wait time: {time.time() - start_time:.4f}s")
-        data = self.__data
-        self.__data = None
-        return data

kotonebot-0.5.0.dist-info/RECORD

@@ -1,71 +0,0 @@
-kotonebot/__init__.py,sha256=1xO-UaCDhT9sAdfKWV3KUwtvws-ONqI7QCTRnYsV48E,597
-kotonebot/errors.py,sha256=biNgoJdU1RUxs3RwqKqbdvt2aXsZmkr_wpPjpxUrmS0,2503
-kotonebot/util.py,sha256=Tiiwgxbj2YYeqFJtebafZlRfcK970bHBItn5SxUR9pA,13934
-kotonebot/backend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kotonebot/backend/bot.py,sha256=kmqWjxSXLHAbPXyKozBZv3s3sY6YSmkXDsIbq7QOWOE,11558
-kotonebot/backend/color.py,sha256=kmDtYaWkofbg9QtdWpbMu41a-jorabcffFmtIu9gDR0,19600
-kotonebot/backend/core.py,sha256=NAhveckuZLhu356WjpOUQOFXFjcLi5LwoV44IxbBeXY,3897
-kotonebot/backend/dispatch.py,sha256=4IzA7pfyH4VFkoA5TLvHLZn_Is7MV9-wx78uVxMe1YQ,7196
-kotonebot/backend/flow_controller.py,sha256=fNbZHiFVGkFoC0-MvLTJDKrEjtjDXfrJ7L9vYbTdnIU,6032
-kotonebot/backend/image.py,sha256=jXYGVXa9K1zBCJBG1Btsf2AG5XI9A0DMy5uYy6KOusk,28111
-kotonebot/backend/loop.py,sha256=e4STYQ6sbUp53xcYJXyhz4vOyEzXds1MKR0xebqU228,6638
-kotonebot/backend/ocr.py,sha256=kgnPWnB-Sbl7pAM6_RSaq0NrhlmArL7wzB574TIURMw,17844
-kotonebot/backend/preprocessor.py,sha256=ce2J70I4vbiyOly61KVPkfmsJf8Oeuj23Vnmbz_eCGk,3320
-kotonebot/backend/context/__init__.py,sha256=TR8WdJt92k62HskPFyfuubyIWXcnJpT_ItGKH1daevY,165
-kotonebot/backend/context/context.py,sha256=R91knq7Ry5MSWaFhOj3HIF13u8bomOOsGk59yPOG4aw,33780
-kotonebot/backend/context/task_action.py,sha256=HZeYzKFbC3mgOQ1g3rodwhYCncImhqM5FenbKHNsHnc,6261
-kotonebot/backend/debug/__init__.py,sha256=pcSpwzU2YwGrogOoHmsI035nkA_-kDLfm-lfBxHuQ-c,43
-kotonebot/backend/debug/entry.py,sha256=Ohf42wPs0C6P5ZGEyAJiol0h531Lss6pKhATJMBnl0w,2744
-kotonebot/backend/debug/mock.py,sha256=Owm86oYtKnlSfgwqb3n5undlshJjdgNTkFl7D33zWSQ,2040
-kotonebot/backend/debug/server.py,sha256=-xZRDy6BSExQi2Zcx_HjuQ8TYviQB_776kPtt16Nooc,7295
-kotonebot/backend/debug/vars.py,sha256=_Zj7s4CikU2XU3xKsKeLYhKmImH-vB9xwXYviuproZQ,10561
-kotonebot/client/__init__.py,sha256=VlseblhIlmDtNehlxLjbaaSKiXGZSVZSs_RlS7VeKcw,183
-kotonebot/client/device.py,sha256=83h2hqNjAyqpPfCnMGXmQMlHfyoVlwiYRP1TgyyEWTA,20828
-kotonebot/client/fast_screenshot.py,sha256=v_57ZkEKOv98S0y8FHs_n4de5-k5neSiBJU4CEXDEd4,13270
-kotonebot/client/protocol.py,sha256=H9SKCSVPuORn63GuSycFnRV4CMWAfqtEWXZF9YAjY7w,2231
-kotonebot/client/registration.py,sha256=xHaCUDNOtA3o8enmSRAVsb1hVoSuv1GstyeM8MgONQM,824
-kotonebot/client/host/__init__.py,sha256=1bWuQSObozMVF0JqtvtAm_Fn-A1MsEDaJHCMw42FbAY,1537
-kotonebot/client/host/adb_common.py,sha256=OOu1ik3VzoCx1DdGKQPnebByxkAUtBiBp5f6EkBl8eg,3790
-kotonebot/client/host/custom.py,sha256=dy49_-ZBosUoSM92CzaEJ7eEHkoooR4VFehDNnIoc98,4104
-kotonebot/client/host/leidian_host.py,sha256=ydRJpyArD0F1bDNFcxp0w3ryIOGSfwv9_fzm-IfRg1I,7353
-kotonebot/client/host/mumu12_host.py,sha256=lbLsSZJtyEYIyX2iFOVmBCBBMOEYIg1W8gj_5F-nqtU,13911
-kotonebot/client/host/protocol.py,sha256=N5kRm_IVubsQQlAOkZvAHPrREbSaSnVYrLqyz9U8inc,7959
-kotonebot/client/host/windows_common.py,sha256=kxCSachnMIyZEavl-0-MAwADTd0lqbZmgrUyvgjImXo,2373
-kotonebot/client/implements/__init__.py,sha256=b9eaN16EdWPPu1WERA2HysWI1YHPYU9jtQ2wuAA8BvU,2205
-kotonebot/client/implements/adb.py,sha256=nMGAIDlKTy9v2SKU3q5XSuE78b9M-wqEQPwN98eLtOU,3199
-kotonebot/client/implements/adb_raw.py,sha256=QkYCeLnVHevc6q4qyxW34Y4MTSrhxS9_d7qfBW-QgBI,6292
-kotonebot/client/implements/remote_windows.py,sha256=OhgpEcmTYhAsedqPY419BcJTlMvJW0ssJ8nHQjdlEuU,6568
-kotonebot/client/implements/uiautomator2.py,sha256=BN1XRZi9LYZrvM9JKpPgs2F1PgXTsLWLzDhkRl-t2nw,2637
-kotonebot/client/implements/windows.py,sha256=FduISikRDi72paLs5xpKcohXsC4HkM36AInp8j4Be10,6713
-kotonebot/client/implements/nemu_ipc/__init__.py,sha256=9Q7ovvO4qf3sQTHAUTHHuUOgijmd6B8OTKSsdQkYrG0,310
-kotonebot/client/implements/nemu_ipc/external_renderer_ipc.py,sha256=TEz7DykLLa30Rb5r5JGt2Odus_OJw4X2HYfiNGsvOyQ,10359
-kotonebot/client/implements/nemu_ipc/nemu_ipc.py,sha256=SV6JRShJ_z_f5tzh-0wRn6fhsrjmbsZKBl65j9CmGas,11755
-kotonebot/config/__init__.py,sha256=-jATUOdrpUrBRT9RiTRQho2-2zeet50qQggsVMVpqNE,35
-kotonebot/config/base_config.py,sha256=BwC3XWc78r8FhBsHhcSBlW2sjhomy5EhRksMDNRuuvA,3342
-kotonebot/config/manager.py,sha256=Ybfv92yhM_l2kKKEw1VsKTYxundALlurSpZge7XIDxA,1025
-kotonebot/interop/win/__init__.py,sha256=e6Pps5u7z88PabVFG8Vg11RHw3toCSa3GjuihgUcXtc,169
-kotonebot/interop/win/_mouse.py,sha256=cMeIA3SaxTJYRfPC1a7M1dkzw-pa2Ctg7pxKAUZdtLI,10131
-kotonebot/interop/win/message_box.py,sha256=YiT_CFS9SPCJjAXwrzHktHJ5elCLzmE83sUnPAC9s4o,8322
-kotonebot/interop/win/reg.py,sha256=jWf-px_-iK4oJOqu5x955zHQGVDEcunNvuAPXt_DqvY,1310
-kotonebot/interop/win/shortcut.py,sha256=ftt0ieKGWDgcozn63URUFBa5x5U8vdrRkUdt_7BnVcc,1721
-kotonebot/interop/win/task_dialog.py,sha256=KO8yCelqcoP4K6Jgw18IsXnM5FK1wTCvZ6y-lWsOyx8,19678
-kotonebot/logging/__init__.py,sha256=bpRRM3l35XLeFkD7Yxkh1DfzM5FTsfNl7pPjn0E9dW4,140
-kotonebot/logging/log.py,sha256=e3Nr8viVSc3wIkPaLQBDRpWbc6nzpwtTmCUPIqSGDaE,538
-kotonebot/primitives/__init__.py,sha256=V5NybWHUCIcIe_7m2_5tw_SRCLK7RnBVSUCjCO0QYZU,338
-kotonebot/primitives/geometry.py,sha256=qhgms19KIVabyu9nZOphNw1E0q4F-t_pj9vbTiB9ZU4,25749
-kotonebot/primitives/visual.py,sha256=eJyNKGynDJO_aI2b8zFcIGaYH7da9n5ICe0mwMpIe8Q,1965
-kotonebot/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kotonebot/tools/mirror.py,sha256=FnElklO_fKZtzKi72tnRCIuY0veQSU5jLfKv7_tObt0,13466
-kotonebot/ui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kotonebot/ui/user.py,sha256=RnswuO4EgFf2eTuOAcnfK9vcrW5F5hEjFyD-3fi5RXY,4337
-kotonebot/ui/file_host/sensio.py,sha256=G6L4NqZT6T7wWF9_ZVuH9exa-W9qJ2OaLoB5mxGP_LY,878
-kotonebot/ui/file_host/tmp_send.py,sha256=am7dLjbU9eb-7d1HBRp1eBI5ycWeeJamB3DlLELGna8,1748
-kotonebot/ui/pushkit/__init__.py,sha256=icm_WYjlfbFhbxxdnKadn-gzYfm1XguTsiiZauqyN0E,70
-kotonebot/ui/pushkit/image_host.py,sha256=7miupmjSZP_6krAcF9TxB0Ae7ZJf9OJljrytEaT3MzY,2503
-kotonebot/ui/pushkit/protocol.py,sha256=h_1YICkrvviEiZYLAJ1QMnizXMWvqQnIrUY7zIyNWFQ,253
-kotonebot/ui/pushkit/wxpusher.py,sha256=qqEmg17wJmHOt6EtKdfInaX22erkMgzkULt6T0pk1Ho,1696
-kotonebot-0.5.0.dist-info/licenses/LICENSE,sha256=ixuiBLtpoK3iv89l7ylKkg9rs2GzF9ukPH7ynZYzK5s,35148
-kotonebot-0.5.0.dist-info/METADATA,sha256=_Py_jZ16jYDtksdTSVRGHKL8OkXe8XVPQHszN40570E,3154
-kotonebot-0.5.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-kotonebot-0.5.0.dist-info/top_level.txt,sha256=QUWAZdbBndoojkrs6RcNytLAn7a0ns4YNF4tLx2Nc4s,10
-kotonebot-0.5.0.dist-info/RECORD,,