python3+unittest+appium 做ui自动化截图对比之后返回对比图

from PIL import Image
from PIL import ImageChops

def compare_images(driver, path_one, lu):
	img_folder = os.fspath('D:\\screenshots') + '\\' + str(lu) + "\ERROR" + lu
	Screentime = time.strftime('%Y%m%d%H%M%S', time.localtime(time.time()))
	screen_save_path = img_folder + Screentime + '.png'
	driver.get_screenshot_as_file(screen_save_path)
	# myprint(screen_save_path)
	"""
	比较图片，如果有不同则生成展示不同的图片

	@参数一: path_one: 第一张图片的路径
	@参数二: path_two: 第二张图片的路径
	@参数三: diff_save_location: 不同图的保存路径
	"""
	# diff_save_location = 'D:\\screenshots\\mumu\\' + str(department) + Screentime + '.png'
	image_one = Image.open(path_one)
	image_two = Image.open(screen_save_path)
	try:
		diff = ImageChops.difference(image_one, image_two)
		if diff.getbbox() is None:
			# 图片间没有任何不同则删除图片
			os.remove(screen_save_path)
			return True
		else:
			myprint(diff)
			# diff.save(diff_save_location)
			# image_two.save(diff_save_location)
			return screen_save_path
	except ValueEr