使用python在selenium自动化中实现长截图

前言

在selenium自动化中，截图的功能是很简单的，一句代码就能实现。

driver.get_screenshot_as_file(pic_name)

问题

可当前却碰到另一个问题，一个页面在一个屏幕放不下，需要滑动才能逐步显示全，而我需要获取这个页面的所有显示作为一张图片，那只能使用长截图了

虽然长截图在很多工具上都提供了，但我自己怎么实现呢？

分析

所谓长截图，不过是当前页面在一个屏幕下显示不全，需要分几屏幕显示而已。那么我们要做的就是

1.算出页面的总长度

2.根据当前屏幕的高度，分析需要滑动几次才能在屏幕显示完

3.每次截取一屏，滑动后再截图

4.把所有截图进行拼接

思路如上，那么又该如何实现呢？这就需要用到我们的PIL库（python2），在python3中是pillow库

实现

我们使用一个函数来实现，在这个函数中，传入3个参数：

driver 是指webdriver

pic_name是指要传入的保存文件名

target_element是指到达什么元素停止截图，如果要截取页面所有，可以不用传入

bar_height表示顶部需要忽略的高度，以避免页面滑动时有些顶部元素不随页面滑动，需要忽略的情况。

from selenium.webdriver.chrome.options import Options
options = Options()
options.add_experimental_option('excludeSwitches', ['enable-logging'])   # 禁用 Chrome 的日志输出
driver = webdriver.Chrome(options=options)

def get_long_pic_url(driver, pic_name, target_element=None, bar_height=0):
    # 获取页面总高度
    if target_element:
        total_height = target_element.location['y']
    else:
        total_height = driver.execute_script("return document.body.parentNode.scrollHeight")

    viewport_height = driver.execute_script("return window.innerHeight")

    # 计算需要截取的次数
    viewport_height = viewport_height-bar_height
    scrolls = total_height // viewport_height
    if total_height % viewport_height != 0:
        scrolls += 1

    # 创建空白画布
    stitched_image = Image.new('RGB', (driver.get_window_size()['width'], total_height))
    current_position = 0

    # 滚动并截取每一部分
    for i in range(scrolls):
        # 滚动到当前位置
        driver.execute_script(f"window.scrollTo(0, {current_position})")

        # 等待滚动完成
        time.sleep(0.5)

        # 获取当前视口的截图
        screenshot = driver.get_screenshot_as_png()
        img = Image.open(io.BytesIO(screenshot))

        # 如果当前是第一次截图，获取窗口高度
        if i == 0:
            effective_img = img
        else:
            # 截取从消息栏下方开始的部分
            effective_img = img.crop((0, bar_height, img.width, img.height))

        # 将截图粘贴到画布上
        stitched_image.paste(effective_img, (0, current_position))

        # 更新位置
        current_position += effective_img.height

    # 保存拼接后的图片
    pic_name = os.path.dirname(os.path.abspath(__file__)) + "\\img\\" + pic_name + time.strftime("%Y%m%d%H%M%S", time.localtime()) + ".png"  # 截图文件名
    stitched_image.save(pic_name)
    # print(f"拼接后的完整截图已保存至: {pic_name}")
    pic_url = upload_file(pic_name)
    return pic_url

pic_url = get_long_pic_url(driver=driver, pic_name='cma', target_element=target_element, bar_height=100)