windows-mcp 0.5.7__py3-none-any.whl → 0.5.8__py3-none-any.whl

windows_mcp/tree/utils.py

@@ -1,22 +1,22 @@
-import random
-from uiautomation import Control
-
-def random_point_within_bounding_box(node: Control, scale_factor: float = 1.0) -> tuple[int, int]:
-    """
-    Generate a random point within a scaled-down bounding box.
-
-    Args:
-        node (Control): The node with a bounding rectangle
-        scale_factor (float, optional): The factor to scale down the bounding box. Defaults to 1.0.
-
-    Returns:
-        tuple: A random point (x, y) within the scaled-down bounding box
-    """
-    box = node.BoundingRectangle
-    scaled_width = int(box.width() * scale_factor)
-    scaled_height = int(box.height() * scale_factor)
-    scaled_left = box.left + (box.width() - scaled_width) // 2
-    scaled_top = box.top + (box.height() - scaled_height) // 2
-    x = random.randint(scaled_left, scaled_left + scaled_width)
-    y = random.randint(scaled_top, scaled_top + scaled_height)
+import random
+from windows_mcp.uia import Control
+
+def random_point_within_bounding_box(node: Control, scale_factor: float = 1.0) -> tuple[int, int]:
+    """
+    Generate a random point within a scaled-down bounding box.
+
+    Args:
+        node (Control): The node with a bounding rectangle
+        scale_factor (float, optional): The factor to scale down the bounding box. Defaults to 1.0.
+
+    Returns:
+        tuple: A random point (x, y) within the scaled-down bounding box
+    """
+    box = node.BoundingRectangle
+    scaled_width = int(box.width() * scale_factor)
+    scaled_height = int(box.height() * scale_factor)
+    scaled_left = box.left + (box.width() - scaled_width) // 2
+    scaled_top = box.top + (box.height() - scaled_height) // 2
+    x = random.randint(scaled_left, scaled_left + scaled_width)
+    y = random.randint(scaled_top, scaled_top + scaled_height)
     return (x, y)

windows_mcp/tree/views.py

@@ -1,116 +1,116 @@
-from dataclasses import dataclass,field
-from tabulate import tabulate
-from typing import Optional
-
-@dataclass
-class DOMInfo:
-    horizontal_scrollable: bool
-    horizontal_scroll_percent: float
-    vertical_scrollable: bool
-    vertical_scroll_percent: float
-
-@dataclass
-class TreeState:
-    interactive_nodes:list['TreeElementNode']=field(default_factory=list)
-    scrollable_nodes:list['ScrollElementNode']=field(default_factory=list)
-    dom_informative_nodes:list['TextElementNode']=field(default_factory=list)
-    dom_info:Optional['DOMInfo']=None
-
-    def interactive_elements_to_string(self) -> str:
-        if not self.interactive_nodes:
-            return "No interactive elements"
-        headers = ["Label", "App Name", "ControlType", "Name", "Value", "Shortcut", "Coordinates" ,"IsFocused"]
-        rows = [node.to_row(idx) for idx, node in enumerate(self.interactive_nodes)]
-        return tabulate(rows, headers=headers, tablefmt="simple")
-
-    def scrollable_elements_to_string(self) -> str:
-        if not self.scrollable_nodes:
-            return "No scrollable elements"
-        headers = [
-            "Label", "App Name", "ControlType", "Name", "Coordinates",
-            "Horizontal Scrollable", "Horizontal Scroll Percent(%)", "Vertical Scrollable", "Vertical Scroll Percent(%)", "IsFocused"
-        ]
-        base_index = len(self.interactive_nodes)
-        rows = [node.to_row(idx, base_index) for idx, node in enumerate(self.scrollable_nodes)]
-        return tabulate(rows, headers=headers, tablefmt="simple")
-    
-@dataclass
-class BoundingBox:
-    left:int
-    top:int
-    right:int
-    bottom:int
-    width:int
-    height:int
-
-    def get_center(self)->'Center':
-        return Center(x=self.left+self.width//2,y=self.top+self.height//2)
-
-    def xywh_to_string(self):
-        return f'({self.left},{self.top},{self.width},{self.height})'
-    
-    def xyxy_to_string(self):
-        x1,y1,x2,y2=self.convert_xywh_to_xyxy()
-        return f'({x1},{y1},{x2},{y2})'
-    
-    def convert_xywh_to_xyxy(self)->tuple[int,int,int,int]:
-        x1,y1=self.left,self.top
-        x2,y2=self.left+self.width,self.top+self.height
-        return x1,y1,x2,y2
-
-@dataclass
-class Center:
-    x:int
-    y:int
-
-    def to_string(self)->str:
-        return f'({self.x},{self.y})'
-
-@dataclass
-class TreeElementNode:
-    name: str
-    control_type: str
-    app_name: str
-    value:str
-    shortcut: str
-    bounding_box: BoundingBox
-    center: Center
-    xpath:str
-    is_focused:bool
-
-    def to_row(self, index: int):
-        return [index, self.app_name, self.control_type, self.name, self.value, self.shortcut, self.center.to_string(),self.is_focused]
-
-@dataclass
-class ScrollElementNode:
-    name: str
-    control_type: str
-    xpath:str
-    app_name: str
-    bounding_box: BoundingBox
-    center: Center
-    horizontal_scrollable: bool
-    horizontal_scroll_percent: float
-    vertical_scrollable: bool
-    vertical_scroll_percent: float
-    is_focused: bool
-
-    def to_row(self, index: int, base_index: int):
-        return [
-            base_index + index,
-            self.app_name,
-            self.control_type,
-            self.name,
-            self.center.to_string(),
-            self.horizontal_scrollable,
-            self.horizontal_scroll_percent,
-            self.vertical_scrollable,
-            self.vertical_scroll_percent,
-            self.is_focused
-        ]
-
-@dataclass
-class TextElementNode:
-    text:str
-
+from dataclasses import dataclass,field
+from tabulate import tabulate
+from typing import Optional
+
+@dataclass
+class DOMInfo:
+    horizontal_scrollable: bool
+    horizontal_scroll_percent: float
+    vertical_scrollable: bool
+    vertical_scroll_percent: float
+
+@dataclass
+class TreeState:
+    interactive_nodes:list['TreeElementNode']=field(default_factory=list)
+    scrollable_nodes:list['ScrollElementNode']=field(default_factory=list)
+    dom_informative_nodes:list['TextElementNode']=field(default_factory=list)
+    dom_info:Optional['DOMInfo']=None
+
+    def interactive_elements_to_string(self) -> str:
+        if not self.interactive_nodes:
+            return "No interactive elements"
+        headers = ["Label", "App Name", "ControlType", "Name", "Value", "Shortcut", "Coordinates" ,"IsFocused"]
+        rows = [node.to_row(idx) for idx, node in enumerate(self.interactive_nodes)]
+        return tabulate(rows, headers=headers, tablefmt="simple")
+
+    def scrollable_elements_to_string(self) -> str:
+        if not self.scrollable_nodes:
+            return "No scrollable elements"
+        headers = [
+            "Label", "App Name", "ControlType", "Name", "Coordinates",
+            "Horizontal Scrollable", "Horizontal Scroll Percent(%)", "Vertical Scrollable", "Vertical Scroll Percent(%)", "IsFocused"
+        ]
+        base_index = len(self.interactive_nodes)
+        rows = [node.to_row(idx, base_index) for idx, node in enumerate(self.scrollable_nodes)]
+        return tabulate(rows, headers=headers, tablefmt="simple")
+    
+@dataclass
+class BoundingBox:
+    left:int
+    top:int
+    right:int
+    bottom:int
+    width:int
+    height:int
+
+    def get_center(self)->'Center':
+        return Center(x=self.left+self.width//2,y=self.top+self.height//2)
+
+    def xywh_to_string(self):
+        return f'({self.left},{self.top},{self.width},{self.height})'
+    
+    def xyxy_to_string(self):
+        x1,y1,x2,y2=self.convert_xywh_to_xyxy()
+        return f'({x1},{y1},{x2},{y2})'
+    
+    def convert_xywh_to_xyxy(self)->tuple[int,int,int,int]:
+        x1,y1=self.left,self.top
+        x2,y2=self.left+self.width,self.top+self.height
+        return x1,y1,x2,y2
+
+@dataclass
+class Center:
+    x:int
+    y:int
+
+    def to_string(self)->str:
+        return f'({self.x},{self.y})'
+
+@dataclass
+class TreeElementNode:
+    name: str
+    control_type: str
+    app_name: str
+    value:str
+    shortcut: str
+    bounding_box: BoundingBox
+    center: Center
+    xpath:str
+    is_focused:bool
+
+    def to_row(self, index: int):
+        return [index, self.app_name, self.control_type, self.name, self.value, self.shortcut, self.center.to_string(),self.is_focused]
+
+@dataclass
+class ScrollElementNode:
+    name: str
+    control_type: str
+    xpath:str
+    app_name: str
+    bounding_box: BoundingBox
+    center: Center
+    horizontal_scrollable: bool
+    horizontal_scroll_percent: float
+    vertical_scrollable: bool
+    vertical_scroll_percent: float
+    is_focused: bool
+
+    def to_row(self, index: int, base_index: int):
+        return [
+            base_index + index,
+            self.app_name,
+            self.control_type,
+            self.name,
+            self.center.to_string(),
+            self.horizontal_scrollable,
+            self.horizontal_scroll_percent,
+            self.vertical_scrollable,
+            self.vertical_scroll_percent,
+            self.is_focused
+        ]
+
+@dataclass
+class TextElementNode:
+    text:str
+
 ElementNode=TreeElementNode|ScrollElementNode

windows_mcp/uia/__init__.py

@@ -0,0 +1,4 @@
+from .enums import *
+from .core import *
+from .patterns import *
+from .controls import *