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

kotonebot/interop/win/window.py

@@ -0,0 +1,89 @@
+from typing import Literal
+from typing_extensions import assert_never
+
+import win32gui
+import win32con
+import win32api
+
+from kotonebot.primitives import Rect
+
+FindWindowMethod = Literal['title']
+
+class Win32Window:
+    def __init__(self, hwnd: int) -> None:
+        self.hwnd = hwnd
+
+    @staticmethod
+    def find_window(method: FindWindowMethod, title: str) -> 'Win32Window | None':
+        """查找窗口。
+
+        :param method: 查找依据。
+        :param title: 窗口标题
+        :return: 若找到窗口则返回 Win32Window 实例，否则返回 None。
+        """
+        match method:
+            case 'title':
+                hwnd = win32gui.FindWindow(None, title)
+                if hwnd == 0:
+                    return None
+                return Win32Window(hwnd)
+            case _:
+                assert_never(method)
+    
+    @staticmethod
+    def require_window(method: FindWindowMethod, title: str) -> 'Win32Window':
+        """查找窗口，未找到则抛出异常。
+        
+        参数同 :ref:`find_window`。
+        """
+        window = Win32Window.find_window(method, title)
+        if window is None:
+            raise RuntimeError(f'Window not found: {title}')
+        return window
+
+    def get_rect(self) -> Rect:
+        """取得窗口范围"""
+        left, top, right, bottom = win32gui.GetWindowRect(self.hwnd)
+        return Rect(left, top, right - left, bottom - top)
+    
+    def get_client_rect(self) -> Rect:
+        """取得窗口客户区域范围"""
+        left, top, right, bottom = win32gui.GetClientRect(self.hwnd)
+        return Rect(left, top, right - left, bottom - top)
+    
+    def is_active(self) -> bool:
+        """检查窗口是否为前台窗口"""
+        active_hwnd = win32gui.GetForegroundWindow()
+        return active_hwnd == self.hwnd
+    
+    def is_minimized(self) -> bool:
+        """检查窗口是否最小化"""
+        return win32gui.IsIconic(self.hwnd) != 0
+    
+    def restore(self) -> None:
+        """还原最小化的窗口"""
+        win32gui.ShowWindow(self.hwnd, win32con.SW_RESTORE)
+
+    def set_position(self, x: int, y: int, *, flags: int | None = None) -> None:
+        """
+        设置窗口位置。
+        
+        :param flags: SetWindowPos 的 `flags` 参数。默认参数为 SWP_NOSIZE | SWP_NOZORDER | SWP_NOACTIVATE。
+        """
+        if flags is None:
+            flags = win32con.SWP_NOSIZE | win32con.SWP_NOZORDER | win32con.SWP_NOACTIVATE
+        win32gui.SetWindowPos(
+            self.hwnd, None, x, y, 0, 0,
+            flags
+        )
+    
+    def bring_foreground(self) -> None:
+        """将窗口置于前台"""
+        win32gui.SetForegroundWindow(self.hwnd)
+
+    def send_message(self, msg: int, wparam: int, lparam: int) -> int:
+        return win32gui.SendMessage(self.hwnd, msg, wparam, lparam)
+    
+    def post_message(self, msg: int, wparam: int, lparam: int) -> bool:
+        win32gui.PostMessage(self.hwnd, msg, wparam, lparam)
+        return win32api.GetLastError() == 0

kotonebot/logging/__init__.py

@@ -1,2 +1,2 @@
-from logging import DEBUG, INFO, WARNING, ERROR, CRITICAL, basicConfig
-from .log import getLogger, enable_verbose, LEVEL_VERBOSE as VERBOSE
+from logging import DEBUG, INFO, WARNING, ERROR, CRITICAL, basicConfig
+from .log import getLogger, enable_verbose, LEVEL_VERBOSE as VERBOSE

kotonebot/logging/log.py

@@ -1,18 +1,18 @@
-import logging
-from typing import cast
-
-LEVEL_VERBOSE = 5
-
-def enable_verbose():
-    logging.addLevelName(LEVEL_VERBOSE, 'VERBOSE')
-
-class VerboseLogger(logging.Logger):
-    def verbose(self, msg, *args, **kwargs):
-        if self.isEnabledFor(LEVEL_VERBOSE):
-            self._log(LEVEL_VERBOSE, msg, args, **kwargs)
-
-def getLogger(name: str | None = None) -> VerboseLogger:
-    logger = logging.getLogger(name)
-    if not isinstance(logger, VerboseLogger):
-        logger.__class__ = VerboseLogger
+import logging
+from typing import cast
+
+LEVEL_VERBOSE = 5
+
+def enable_verbose():
+    logging.addLevelName(LEVEL_VERBOSE, 'VERBOSE')
+
+class VerboseLogger(logging.Logger):
+    def verbose(self, msg, *args, **kwargs):
+        if self.isEnabledFor(LEVEL_VERBOSE):
+            self._log(LEVEL_VERBOSE, msg, args, **kwargs)
+
+def getLogger(name: str | None = None) -> VerboseLogger:
+    logger = logging.getLogger(name)
+    if not isinstance(logger, VerboseLogger):
+        logger.__class__ = VerboseLogger
     return cast(VerboseLogger, logger)

kotonebot/primitives/__init__.py

@@ -1,17 +1,19 @@
-from .visual import Image, Template
-from .geometry import Point, Rect, Size, Vector2D, Vector3D, is_point, is_rect
-from .geometry import  RectTuple, PointTuple
-
-__all__ = [
-    'Image',
-    'Template',
-    'Point',
-    'Rect',
-    'Size',
-    'Vector2D',
-    'Vector3D',
-    'is_point',
-    'is_rect',
-    'RectTuple',
-    'PointTuple'
-]
+from .visual import Image, Template, ImageSlice
+from .geometry import Point, Rect, Size, Vector2D, Vector3D, is_point, is_rect
+from .geometry import  RectTuple, PointTuple
+
+__all__ = [
+    'Image',
+    'Template',
+    'ImageSlice',
+    
+    'Point',
+    'Rect',
+    'Size',
+    'Vector2D',
+    'Vector3D',
+    'is_point',
+    'is_rect',
+    'RectTuple',
+    'PointTuple'
+]