Selenium實(shí)現(xiàn)錄屏的一種方法

由于UI層自動(dòng)化的不穩(wěn)定性,經(jīng)常會(huì)遇到執(zhí)行中斷或用例失敗的問題盹兢,以下是一些常見的措施邻梆。

  1. 詳細(xì)的日志
  2. 定位出錯(cuò)時(shí)截圖
  3. Pytest的緩存機(jī)制(可以記錄成功了哪些失敗了哪些)
  4. 自動(dòng)重試機(jī)制(如pytest-rerunfailures)
  5. 用例錄像

用例錄像是最直觀的一種方式,可以查看到實(shí)時(shí)的操作情況绎秒。很多云平臺(tái)如Saucelabs等都提供了用例錄像的功能浦妄。
然而很遺憾selenium本身沒有這種能力。除了借助三方軟件如ffmpeg同步錄屏外见芹,另外一種實(shí)現(xiàn)方法是另外啟動(dòng)一個(gè)線程剂娄,不斷截圖,最后將圖片拼接成git玄呛。
操作方法如下阅懦。

實(shí)時(shí)截圖

  1. 編寫一個(gè)循環(huán)截圖函數(shù)shot
def shot(dr):
    i = 0
    while True:
        img_file = os.path.join(img_dir, f'{i}.png')
        try:
            dr.save_screenshot(img_file)
        except:
            return
        i += 1

由于webdriver接口執(zhí)行截圖命令的速度限制,每一輪截圖無需再sleep徘铝。

  1. 操作web時(shí)耳胎,啟動(dòng)一個(gè)線程
img_dir = 'img'  # 臨時(shí)圖片目錄
dr = webdriver.Chrome()

t = threading.Thread(target=shot, args=(dr, img_dir))  # 新建線程
t.start()  # 啟動(dòng)截圖線程

dr.get('https://www.baidu.com')
dr.find_element('id', 'kw').send_keys('簡書 韓志超')
dr.find_element('id', 'su').click()
time.sleep(1)
dr.get('https://www.qq.com')
dr.back()
time.sleep(2)
dr.quit()
  1. 圖片拼接成gif

需要安裝pillow: pip install pillow

img_list = os.listdir(img_dir)  # 列出目錄所有圖片
img_list.sort(key=lambda x: int(x[:-4]))  # 排序

first_img = Image.open(os.path.join(img_dir, img_list[0]))  # 第一張圖片對(duì)象
else_imgs = [Image.open(os.path.join(img_dir, img)) for img in img_list[1:]]  # 剩余圖片對(duì)象

first_img.save("record.gif", append_images=else_imgs,
               duration=300,
               save_all=True) # 拼接保存

完整代碼

from selenium import webdriver
import threading
import os
import time
from PIL import Image


def clear_dir(path):
    """創(chuàng)建或清空目錄"""
    if not os.path.isdir(path):
        os.mkdir(path)  # 創(chuàng)建目錄
    else:  # 清空目錄
        [os.remove(os.path.join(path, file_name)) for file_name in os.listdir(path)]


def shot(dr, img_dir):
    """循環(huán)截圖函數(shù)"""
    i = 0
    clear_dir(img_dir)  # 清空目錄
    while True:
        img_file = os.path.join(img_dir, f'{i}.png')
        try:
            dr.save_screenshot(img_file)
        except:
            return
        i += 1


# Selenium操作
img_dir = 'img'  # 臨時(shí)圖片目錄
dr = webdriver.Chrome()

t = threading.Thread(target=shot, args=(dr, img_dir))  # 新建線程
t.start()  # 啟動(dòng)截圖線程

dr.get('https://www.baidu.com')
dr.find_element('id', 'kw').send_keys('簡書 韓志超')
dr.find_element('id', 'su').click()
time.sleep(1)
dr.get('https://www.qq.com')
dr.back()
time.sleep(2)
dr.quit()

# 圖片拼接成gif
img_list = os.listdir(img_dir)  # 列出目錄所有圖片
img_list.sort(key=lambda x: int(x[:-4]))  # 排序

first_img = Image.open(os.path.join(img_dir, img_list[0]))  # 第一張圖片對(duì)象
else_imgs = [Image.open(os.path.join(img_dir, img)) for img in img_list[1:]]  # 剩余圖片對(duì)象

first_img.save("record.gif", append_images=else_imgs,
               duration=300,  # 每張圖片的過過渡時(shí)間
               save_all=True) # 拼接保存,如果想要循環(huán)播放可以加上loop=0

最終效果--有點(diǎn)像幻燈片


image
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末惕它,一起剝皮案震驚了整個(gè)濱河市怕午,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌淹魄,老刑警劉巖郁惜,帶你破解...
    沈念sama閱讀 219,490評(píng)論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異揭北,居然都是意外死亡扳炬,警方通過查閱死者的電腦和手機(jī)吏颖,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,581評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來恨樟,“玉大人半醉,你說我怎么就攤上這事∪笆酰” “怎么了缩多?”我有些...
    開封第一講書人閱讀 165,830評(píng)論 0 356
  • 文/不壞的土叔 我叫張陵,是天一觀的道長养晋。 經(jīng)常有香客問我衬吆,道長,這世上最難降的妖魔是什么绳泉? 我笑而不...
    開封第一講書人閱讀 58,957評(píng)論 1 295
  • 正文 為了忘掉前任逊抡,我火速辦了婚禮,結(jié)果婚禮上零酪,老公的妹妹穿的比我還像新娘冒嫡。我一直安慰自己,他們只是感情好四苇,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,974評(píng)論 6 393
  • 文/花漫 我一把揭開白布孝凌。 她就那樣靜靜地躺著,像睡著了一般月腋。 火紅的嫁衣襯著肌膚如雪蟀架。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,754評(píng)論 1 307
  • 那天榆骚,我揣著相機(jī)與錄音片拍,去河邊找鬼。 笑死寨躁,一個(gè)胖子當(dāng)著我的面吹牛穆碎,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播职恳,決...
    沈念sama閱讀 40,464評(píng)論 3 420
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼方面!你這毒婦竟也來了放钦?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,357評(píng)論 0 276
  • 序言:老撾萬榮一對(duì)情侶失蹤恭金,失蹤者是張志新(化名)和其女友劉穎操禀,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體横腿,經(jīng)...
    沈念sama閱讀 45,847評(píng)論 1 317
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡颓屑,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,995評(píng)論 3 338
  • 正文 我和宋清朗相戀三年斤寂,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片揪惦。...
    茶點(diǎn)故事閱讀 40,137評(píng)論 1 351
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡遍搞,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出器腋,到底是詐尸還是另有隱情溪猿,我是刑警寧澤,帶...
    沈念sama閱讀 35,819評(píng)論 5 346
  • 正文 年R本政府宣布纫塌,位于F島的核電站诊县,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏措左。R本人自食惡果不足惜依痊,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,482評(píng)論 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望怎披。 院中可真熱鬧抗悍,春花似錦、人聲如沸钳枕。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,023評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽鱼炒。三九已至衔沼,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間昔瞧,已是汗流浹背指蚁。 一陣腳步聲響...
    開封第一講書人閱讀 33,149評(píng)論 1 272
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留自晰,地道東北人凝化。 一個(gè)月前我還...
    沈念sama閱讀 48,409評(píng)論 3 373
  • 正文 我出身青樓,卻偏偏與公主長得像酬荞,于是被迫代替她去往敵國和親搓劫。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,086評(píng)論 2 355