一，背景
1，在实际业务中，执行自动化任务时，会将执行结果截图并返回，但是selenium原生不支持长截图。
二，解决
先将网页缩小后，将网页分割成两份进行截图然后拼接。
def screenshotByElement(self):
try:
"""首先生成最终结果图文件"""
self.screenshotPicName = self.screenshotPicName_tail = f"C:\\tools\\02201\\{datetime.datetime.now().strftime('%Y-%m-%d')}\\{datetime.datetime.now().strftime('%Y-%m-%d-%H-%M-%S')}_{self.bookingreq_id}.png"
self.driver.get_screenshot_as_file(self.screenshotPicName)
"""将网页缩小"""
self.driver.execute_script("document.body.style.zoom = 0.5;")
actions = ActionChains(self.driver)
"""将鼠标焦点移动至主元素"""
div_element = self.driver.find_element(By.ID, 'app')
actions.move_to_element(div_element).perform()
"""将页面拉至最顶端"""
actions.key_down(Keys.PAGE_UP).key_up(Keys.PAGE_UP).perform()
actions.move_to_element(div_element).perform()
actions.key_down(Keys.PAGE_UP).key_up(Keys.PAGE_UP).perform()
actions.move_to_element(div_element).perform()
actions.key_down(Keys.PAGE_UP).key_up(Keys.PAGE_UP).perform()
actions.move_to_element(div_element).perform()
actions.key_down(Keys.PAGE_UP).key_up(Keys.PAGE_UP).perform()
sleep(0.5)
"""截图第一张"""
self.screenshotPicName_top = f"C:\\tools\\02201\\{datetime.datetime.now().strftime('%Y-%m-%d')}\\{datetime.datetime.now().strftime('%Y-%m-%d-%H-%M-%S')}_{self.bookingreq_id}_top.png"
self.driver.get_screenshot_as_file(self.screenshotPicName_top)
"""算好位置 向下拉动 """
actions.move_to_element(div_element).perform()
actions.key_down(Keys.PAGE_DOWN).key_up(Keys.PAGE_DOWN).perform()
sleep(0.5)
"""算好位置 截图第二张 """
self.screenshotPicName_tail = f"C:\\tools\\02201\\{datetime.datetime.now().strftime('%Y-%m-%d')}\\{datetime.datetime.now().strftime('%Y-%m-%d-%H-%M-%S')}_{self.bookingreq_id}_tail.png"
self.driver.get_screenshot_as_file(self.screenshotPicName_tail)
"""拼接图片"""
self.joinPic(self.screenshotPicName_top,self.screenshotPicName_tail)
except Exception as e:
self.logger.info(str(e))
self.screenshotPicName=None
def joinPic(self,pic1,pic2):
"""将两张图片上下 拼接在一起"""
image1 = Image.open(pic1)
image2 = Image.open(pic2)
new_image = Image.new('RGB', (image1.width, image1.height + image2.height))
new_image.paste(image1, (0, 0))
new_image.paste(image2, (0, image1.height))
new_image.save(self.screenshotPicName)
