appium+opencv获取元素颜色

省略appium前置配置步骤 

import cv2
import numpy as np

#根据文本获取button
text_element = driver.find_element(By.XPATH, f"//*[@text='{arg}']/..")
target_button = text_element.find_element(By.ID, 'switchButton')

#获取button位置信息和大小
button_location = target_button.location
button_size = target_button.size

#截图并保存
screenshot_path = r"D:\Coding\imgTest\screenshot.png"
driver.get_screenshot_as_file(screenshot_path)
        
#提取按钮区域，并计算平均颜色
screenshot = cv2.imread(screenshot_path)
button_x, button_y = button_location['x'], button_location['y']
button_width, button_height = button_size['width'], button_size['height']
button_area = screenshot[button_y:button_y + button_height, button_x:button_x + button_width]

average_color_per_row = np.average(button_area, axis=0)
average_color = np.average(average_color_per_row, axis=0)
        
#这里可使用sum对RGB通道进行划分，并做对应操作
print("Average color for the button:", average_color)