perf(Python): 完善 Win32Controller 类型注释; 调整 AlgorithmEnum 继承方式 (#550)

MaaXYZ · Feb 21, 2025 · 13d57de · 13d57de
1 parent 84cf6e2
commit 13d57de
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/source/binding/Python/maa/controller.py b/source/binding/Python/maa/controller.py
@@ -349,7 +349,7 @@ class Win32Controller(Controller):
 
     def __init__(
         self,
-        hWnd: Optional[ctypes.c_void_p],
+        hWnd: Union[ctypes.c_void_p, int, None],
         screencap_method: int = MaaWin32ScreencapMethodEnum.DXGI_DesktopDup,
         input_method: int = MaaWin32InputMethodEnum.Seize,
         notification_handler: Optional[NotificationHandler] = None,

diff --git a/source/binding/Python/maa/define.py b/source/binding/Python/maa/define.py
@@ -1,10 +1,11 @@
 import ctypes
 import platform
 from dataclasses import dataclass
-from enum import IntEnum, Enum
+from enum import IntEnum
 from typing import List, Tuple, Union, Dict, Optional
 
 import numpy
+from strenum import StrEnum  # For Python 3.9/3.10
 
 MaaBool = ctypes.c_uint8
 MaaSize = ctypes.c_size_t
@@ -446,7 +447,7 @@ def roi(self):
 ]
 
 
-class AlgorithmEnum(str, Enum):
+class AlgorithmEnum(StrEnum):
     DirectHit = "DirectHit"
     TemplateMatch = "TemplateMatch"
     FeatureMatch = "FeatureMatch"

diff --git a/source/binding/Python/requirements.txt b/source/binding/Python/requirements.txt
@@ -1 +1,2 @@
 numpy
+strenum