今天学习下使用selenium自动化测试工具获取网页截图。
1，如果是简单获取当前屏幕截图只需要使用方法：
driver.get_screenshot_as_file('screenshot.png')
2，如果想获取完整网页长宽的截图需要设置参数后使用该方法：
首先打开驱动方式设置为无界面显示模式
# 打开驱动
def open_driver():
try:
# 连接浏览器web驱动全局变量
global driver
# Linux系统下浏览器驱动无界面显示，需要设置参数
# “–no-sandbox”参数是让Chrome在root权限下跑
# “–headless”参数是不用打开图形界面
chrome_options = Options()
# 设为无头模式
chrome_options.add_argument('--headless')
chrome_options.add_argument('--no-sandbox')
chrome_options.add_argument('--disable-gpu')
chrome_options.add_argument('--disable-dev-shm-usage')
# 连接Chrome浏览器驱动，获取驱动
driver = webdriver.Chrome(chrome_options=chrome_options)
'''
# 此步骤很重要，设置chrome为开发者模式，防止被各大网站识别出来使用了Selenium
options = Options()
# 静默模式（加载浏览器的静默模式，让它在后台偷偷运行）
# options.add_argument('headless')
# 去掉提示：Chrome正收到自动测试软件的控制
options.add_argument('disable-infobars')
# 以键值对的形式加入参数,打开浏览器开发者模式
# options.add_experimental_option('excludeSwitches', ['enable-automation'])
# 打开浏览器开发者模式
# options.add_argument("--auto-open-devtools-for-tabs")
driver = webdriver.Chrome(chrome_options=options)
'''
# driver = webdriver.Chrome()
print('连接Chrome浏览器驱动')
# 浏览器窗口最大化
driver.maximize_window()
'''
1, 隐式等待方法
driver.implicitly_wait(最大等待时间, 单位: 秒)
2, 隐式等待作用
在规定的时间内等待页面所有元素加载;
3,使用场景:
在有页面跳转的时候, 可以使用隐式等待。
'''
driver.implicitly_wait(3)
# 强制等待，随机休眠 暂停0-3秒的整数秒，时间区间：[0,3]
time.sleep(random.randint(0, 3))
except Exception as e:
driver = None
print(str(e))
然后设置网页长宽最大化，保证截图是完整的，不会出现滚动条
S = lambda X: driver.execute_script('return document.body.parentNode.scroll' + X)
driver.set_window_size(S('Width'), S('Height'))
driver.get_screenshot_as_file('screenshot.png')
3，编写代码
依旧采用拆分步骤细化功能模块封装方法编写代码，便于后续扩展功能模块，代码中缺少的封装方法代码，详情参考之前的《selenium自动化测试》文章。
def spider_screenshot_image(req_dict):
'''
@方法名称: 爬取网页内容截图文件
@中文注释: 爬取网页内容截图文件
@入参:
@param req_dict dict 请求容器
@出参:
@返回状态:
@return 0 失败或异常
@return 1 成功
@返回错误码
@返回错误信息
@param rsp_dict dict 响应容器
@作 者: PandaCode辉
@weixin公众号: PandaCode辉
@创建时间: 2023-09-26
@使用范例: spider_screenshot_image(req_dict)
'''
try:
if (not type(req_dict) is dict):
return [0, "111111", "请求容器参数类型错误,不为字典", [None]]
# 截图目录
screenshot_dir = os.path.join(os.path.dirname(__file__), 'screenshot')
if not os.path.exists(screenshot_dir):
os.makedirs(screenshot_dir)
print('打开浏览器驱动')
open_driver()
# 打开网址网页
print('打开网址网页')
driver.get(req_dict['url'])
# 等待6秒启动完成
driver.implicitly_wait(6)
print('随机休眠')
# 随机休眠 暂停0-2秒的整数秒
time.sleep(random.randint(0, 2))
# 保存当前网页屏幕快照PNG图像文件，截图不保证完整网页内容都截取到
page_file_1 = os.path.join(screenshot_dir, 'page1.png')
isprint = driver.get_screenshot_as_file(page_file_1)
print(isprint)
# 网页长宽最大化，保证截图是完整的，不会出现滚动条
S = lambda X: driver.execute_script('return document.body.parentNode.scroll' + X)
driver.set_window_size(S('Width'), S('Height'))
# 保存当前网页屏幕快照PNG图像文件
page_file_2 = os.path.join(screenshot_dir, 'page2.png')
isprint = driver.get_screenshot_as_file(page_file_2)
print(isprint)
# 章节内容截图
image_file = os.path.join(screenshot_dir, 'content.png')
# 元素定位
elem = driver.find_element(By.ID, req_dict['elem_id'])
print(elem)
# 元素截图
isprint = elem.screenshot(image_file)
print(isprint)
print('关闭浏览器驱动')
close_driver()
print("爬取网页内容截图文件成功")
# 返回容器
return [1, '000000', '爬取网页内容截图文件成功', [None]]
except Exception as e:
print('关闭浏览器驱动')
close_driver()
print("爬取网页内容截图文件异常," + str(e))
return [0, '999999', "爬取网页内容截图文件异常," + str(e), [None]]
4，运行效果
最后说明：上述文章仅供学习参考，请勿用于商业用途，感谢阅读。
