Skip to content

Commit

Permalink
Merge branch 'master' of https://github.com/AirtestProject/Airtest in…
Browse files Browse the repository at this point in the history
…to v1.2.1
  • Loading branch information
yimelia committed Jul 26, 2021
2 parents 2564013 + b9cdd55 commit 07bb05f
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions airtest/core/win/win.py
Expand Up @@ -118,10 +118,12 @@ def snapshot(self, filename=None, quality=10, max_size=None):
screen = screenshot(filename)
if self.app:
rect = self.get_rect()
rect = self._fix_image_rect(rect)
screen = aircv.crop_image(screen, [rect.left, rect.top, rect.right, rect.bottom])
if not screen.any():
if self.app:
rect = self.get_rect()
rect = self._fix_image_rect(rect)
screen = aircv.crop_image(screenshot(filename), [rect.left, rect.top, rect.right, rect.bottom])
if self._focus_rect != (0, 0, 0, 0):
height, width = screen.shape[:2]
Expand All @@ -131,6 +133,16 @@ def snapshot(self, filename=None, quality=10, max_size=None):
aircv.imwrite(filename, screen, quality, max_size=max_size)
return screen


def _fix_image_rect(self, rect):
"""Fix rect in image."""
# 将rect 转换为左上角为(0,0), 与图片坐标对齐
rect.left = rect.left - self.monitor["left"]
rect.right = rect.right - self.monitor["left"]
rect.top = rect.top - self.monitor["top"]
rect.bottom = rect.bottom - self.monitor["top"]
return rect

def keyevent(self, keyname, **kwargs):
"""
Perform a key event
Expand Down

0 comments on commit 07bb05f

Please sign in to comment.