UI自動化測試中，如果需要截圖方便后續(xù)確認(rèn)操作結(jié)果的話，可以利用PIL的ImageGrab進(jìn)行截圖保存。
注意，如果應(yīng)用python3，下載模塊時，需要下載pillow。

pip3 install pillow

下載完成后，首先要引入需要的模塊

# screenshot.py
from PIL import ImageGrab
import os
import time

接下來，定義screen方法，包括進(jìn)行屏幕鏡像的截取，創(chuàng)建截屏的存儲路徑以及圖像保存。

def screen(name):
    # 截取鏡像
    png = ImageGrab.grab()
    # 創(chuàng)建存儲路徑
    screen_path = os.path.join(os.path.dirname(__file__), 'screen')
    if not os.path.exists(screen_path):
        os.makedirs(screen_path)
    # 保存圖片到存儲路徑
    image_name = os.path.join(screen_path, name)
    t = time.strftime('%Y%m%d_%H%M%S', time.localtime())
    png.save('%s_%s.png' % (image_name, t))  # 文件名name后面加了個時間戳，避免重名

后續(xù)調(diào)用，引入from screenshot import screen后，在文件中直接調(diào)用screen方法即可。

截圖保存位置

自我記錄，有錯誤歡迎指正~~