废话不多说，直接进入正题

一、直接截取网页全屏

截全屏的时候，我们用到的内置方法为save_screenshot(“demo1.png”)

from selenium import webdriver
from time import sleep

class test:
    driver = webdriver.Chrome()
    driver.maximize_window()
    driver.get('https://www.baidu.com/')
    sleep(2)
    # 截取全屏
    driver.save_screenshot("demo1.png")

AI写代码c运行

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10

运行成功之后，打开截图文件，如下所示：

二、截取页面上固定元素的图片

咱们以截取百度首页“百度一下”的按钮为例。

from selenium import webdriver
from time import sleep

class test:
    # 打开谷歌浏览器
    driver = webdriver.Chrome()
    # 网页全屏显示
    driver.maximize_window()
    # 访问百度首页
    driver.get('https://www.baidu.com/')
    # 强制等待2秒
    sleep(2)
    # 使用ID定位的方法获取"百度一下"的元素
    element = driver.find_element_by_id("su")
    # 截图
    element.screenshot("element.png")
    # 关闭浏览器
    driver.quit()

AI写代码c运行

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11
• 12
• 13
• 14
• 15
• 16
• 17
• 18

运行之后，打开图片文件，如下图所示：

三、截取指定位置的图片

from selenium import webdriver
from time import sleep
from PIL import Image

class test:

    driver = webdriver.Chrome()
    driver.maximize_window()
    driver.get('https://www.baidu.com/')
    sleep(2)
    driver.save_screenshot("baidu1.png")
    element = driver.find_element_by_id("kw")
    print("获取元素的坐标")
    location = element.location
    print(location)
    print(element.size)
    # 计算元素上、下、左、右的位置
    top = element.location['y']
    bottom = element.location['y'] + element.size['height']
    left = element.location['x']
    right = element.location['x'] + element.size['width']
    print(top, bottom, left, right)
    im = Image.open('baidu1.png')
    im = im.crop((left, top, right,bottom))
    im.show()
    im.save("baidu2.png")

AI写代码c运行

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11
• 12
• 13
• 14
• 15
• 16
• 17
• 18
• 19
• 20
• 21
• 22
• 23
• 24
• 25
• 26

四、图片拼接

from selenium import webdriver
from PIL import Image

driver = webdriver.Chrome()
# 设置浏览器窗口最大化
driver.maximize_window()  # 设置打开页面最大化,目的是更好的截取错误图
# 打开税网
driver.get("https://www.csdn.net/")
# 1. 截取当前页面
driver.save_screenshot('result.png')


def join_images(png1, png2, size=0, output='result.png'):
    """
    图片拼接
    :param png1: 图片1
    :param png2: 图片2
    :param size: 两个图片重叠的距离
    :param output: 输出的图片文件
    :return:
    """
    # 图片拼接
    img1, img2 = Image.open(png1), Image.open(png2)
    size1, size2 = img1.size, img2.size  # 获取两张图片的大小
    joint = Image.new('RGB', (size1[0], size1[1]+size2[1]-size))    # 创建一个空白图片
    # 设置两张图片要放置的初始位置
    loc1, loc2 = (0, 0), (0, size1[1] - size)
    # 分别放置图片
    joint.paste(img1, loc1)
    joint.paste(img2, loc2)
    # 保存结果
    joint.save(output)


JS = {
    '滚动到页尾': "window.scroll({top:document.body.clientHeight,left:0,behavior:'auto'});",
    '滚动到': "window.scroll({top:%d,left:0,behavior:'auto'});",
}
# 获取body大小
body_h = int(driver.find_element_by_xpath('//body').size.get('height'))
# 计算当前页面截图的高度
# （使用driver.get_window_size()也可以获取高度，但有误差，推荐使用图片高度计算）
current_h = Image.open('result.png').size[1]
image_list = ['result.png']  # 储存截取到的图片路径

for i in range(1, int(body_h/current_h)):
    # 1. 滚动到指定锚点
    driver.execute_script(JS['滚动到'] % (current_h * i))
    # 2. 截图
    driver.save_screenshot(f'test_{i}.png')
    join_images('result.png', f'test_{i}.png')
# 处理最后一张图
driver.execute_script(JS['滚动到页尾'])
driver.save_screenshot('test_end.png')
# 拼接图片
join_images('result.png', 'test_end.png', size=current_h-int(body_h % current_h))

AI写代码c运行

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11
• 12
• 13
• 14
• 15
• 16
• 17
• 18
• 19
• 20
• 21
• 22
• 23
• 24
• 25
• 26
• 27
• 28
• 29
• 30
• 31
• 32
• 33
• 34
• 35
• 36
• 37
• 38
• 39
• 40
• 41
• 42
• 43
• 44
• 45
• 46
• 47
• 48
• 49
• 50
• 51
• 52
• 53
• 54
• 55
• 56

运行之后的结果为：

总结：

感谢每一个认真阅读我文章的人！！！

作为一位过来人也是希望大家少走一些弯路，如果你不想再体验一次学习时找不到资料，没人解答问题，坚持几天便放弃的感受的话，在这里我给大家分享一些自动化测试的学习资源，希望能给你前进的路上带来帮助。

软件测试面试文档

我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

 

          视频文档获取方式：
这份文档和视频资料，对于想从事【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你！以上均可以分享，点下方小卡片即可自行领取。