appium截取手机长图(滚动截屏)

在做APP的UI自动化，想要通过简单的截取长图的方式，简单的对比下页面元素是否全部可见且无截断，所以写了个将appium截图进行拼接的脚本。各位可以自行优化。

from PIL import Image
import os

#打开2个图像
image1 = os.getcwd() +'/pics/1.png'
image2 = os.getcwd() +'/pics/2.png'
img_info1 = Image.open(image1)
img_data1 = img_info1.getdata()
img_info2 = Image.open(image2)
img_data2 = img_info2.getdata()
#获取第一张图最后一行和倒数第五行的像素信息
last_pix = []
for w in range(img_info1.width):
    last_pix.append(img_data1.getpixel((w, img_info1.height-1)))
last_five_px = []
for w in range(img_info1.width):
    last_five_px.append(img_data1.getpixel((w, img_info1.height-6)))

#寻找在第二张图中和第一张图最后一行相等的行
#如果相等行存在，去看往前数5行是否也相等
#这个判断可能要根据页面的复杂度多校验几次，约简单的图，越要多校验几行
row = 0
print('图片高度是:{}'.format(img_info2.height))
for h in range(img_info2.height):
    current = []
    current_five = []
    for w in range(img_info2.width):
        current.append(img_data2.getpixel((w,