改造 uiautomator2 的 screenshot 方法
uiautomator2 不支持 base64 格式截图,需要对截图的方法进行改造。
在uiautomator2/__init__.py中找到 screenshot 方法,添加生成 base64 格式图片的代码,改造后的方法如下:
def screenshot(self, filename=None, format='pillow'):
"""
Image format is JPEG
Args:
filename (str): saved filename
format (string): used when filename is empty. one of "pillow" or "opencv"
Raises:
IOError, SyntaxError
Examples:
screenshot("saved.jpg")
screenshot().save("saved.png")
cv2.imwrite('saved.jpg', screenshot(format='opencv'))
"""
r = requests.get(self.server.screenshot_uri, timeout=10)
if filename:
with open(filename, 'wb') as f:
f.write(r.content)
return filename
elif format &