selenium + python實現截圖並且保存圖片

webdriver的截圖功能十分強悍，無論頁面多長，webdriver都能比較完美的截到完整的頁面。

python代碼：

# -*- coding: utf-8 -*-
from selenium import webdriver
import unittest
import os, sys, time
from pathlib import Path

# 初始化實例
driver = webdriver.Firefox()

current_time = time.strftime("%Y-%m-%d-%H_%M_%S", time.localtime(time.time()))
current_time1 = time.strftime("%Y-%m-%d", time.localtime(time.time()))
print(current_time)
print(current_time1)

# 設置存儲圖片路徑，測試結果圖片可以按照每天進行區分

# 通過if進行斷言判斷
# driver.get("https://www.csdn.net/")
driver.get("https://www.processon.com/view/link/5a815fafe4b064e9ddc3e018")
# 新創建路徑“.”表示當前整個.py文件的路徑所在的位置，“\\”路徑分割符，其中的一個是"\"轉義符
pfilename = u'.\\image'
pic_path = pfilename + '\\' + current_time1 + '_' + current_time + '.png'
# 判斷文件夾是否存在，不存在就新建一個新的
if Path(pfilename).is_dir():
    pass
else:
    Path(pfilename).mkdir()

print(pic_path)
time.sleep(2)
print(driver.title)
# 截取當前url頁面的圖片，並且將截取的圖片保存在指定的路徑（pic_path）
driver.save_screenshot(pic_path)

if u'CSDN-專業IT技術社區' == driver.title:
    print('Assertion test pass')
else:
    print('Assertion test fail')

driver.quit()
備注：瀏覽器驅動放到python路徑下（可以就放到腳本當前路徑）