Selenium网页长截图

最新推荐文章于 2024-04-29 09:15:33 发布

西红市杰出青年

最新推荐文章于 2024-04-29 09:15:33 发布

阅读量437

点赞数 7

文章标签： python selenium 缓存测试工具开发语言

本文链接：https://blog.csdn.net/qq_48441158/article/details/137785248

版权

本文介绍了如何在Selenium中处理执行自动化任务时的长截图问题，通过先缩小网页，然后分割截图并使用Python代码进行上下拼接，以实现完整的屏幕截图功能。

摘要由CSDN通过智能技术生成

一，背景

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)