selenium自动化测试-获取网页截图
今天学习下使用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，运行效果
最后说明：上述文章仅供学习参考，请勿用于商业用途，感谢阅读。
