第python自動化測試selenium屏幕截圖示例WebDriver內(nèi)置了測試中捕獲屏幕并保存的方法。

示例腳本：

（1）save_screenshot(filename)：保存屏幕截圖

fromseleniumimportwebdriver

fromtimeimportsleep,strftime,localtime,time

importos

classTestScreenShot(object):

defsetup(self):

self.driver=webdriver.Chrome()

self.driver.get()

deftest_screen(self):

self.driver.save_screenshot(testbaidu.png)

defteardown(self):

self.driver.quit()

if__name__==__main__:

shot=TestScreenShot()

shot.test_screen()

（2）get_screenshot_as_base64()：獲取當前屏幕截圖base64編碼字符串

fromseleniumimportwebdriver

fromtimeimportsleep,strftime,localtime,time

importos

classTestScreenShot(object):

defsetup(self):

self.driver=webdriver.Chrome()

self.driver.get()

deftest_screen(self):

self.driver.save_screenshot(testbaidu.png)

print(self.driver.get_screenshot_as_base64())

defteardown(self):

self.driver.quit()

if__name__==__main__:

shot=TestScreenShot()

shot.test_screen()

運行結果：

（3）get_screenshot_as_file(filename)：獲取當前屏幕截圖，使用完整路徑

fromseleniumimportwebdriver

fromtimeimportsleep,strftime,localtime,time

importos

classTestScreenShot(object):

defsetup(self):

self.driver=webdriver.Chrome()

self.driver.get()

deftest_screen(self):

self.driver.get_screenshot_as_file(testbaidu2.png)

defteardown(self):

self.driver.quit()

if__name__==__main__:

shot=TestScreenShot()

shot.test_screen()

（4）get_screenshot_as_file(filename)：獲取當前屏幕截圖，使用完整路徑，文件名以時間命名

fromseleniumimportwebdriver

fromtimeimportsleep,strftime,localtime,time

importos

classTestScreenShot(object):

defsetup(self):

self.driver=webdriver.Chrome()

self.driver.get()

deftest_screen(self):

#print(self.driver.get_screenshot_as_png())

#self.driver.get_screenshot_as_file(testbaidu2.png)

#self.driver.save_screenshot(testbaidu.png)

#print(self.driver.get_screenshot_as_base64())

self.driver.find_element_by_id(kw).send_keys(seleniumtest)

self.driver.find_element_by_id(su).click()

str=strftime(%Y-%m-%d-%H-%M-%S,localtime(time()))

filename=str+.png

path=os.path.abspath(screenshots)

filepath=path+/+filename

self.driver.get_screenshot_as_file(filepath)

print(filepath)

defteardown(self):

self.driver.quit()

if__name__==__main__:

shot=TestScreenShot()

shot.test_screen()

（5）get_screenshot_as_png()：獲取當前屏幕截圖的二進制文件數(shù)據(jù)

fromseleniumimportwebdriver

fromtimeimportsleep,strftime,localtime,time

importos

classTestScreenShot(object):

defsetup(self):

self.driver=webdriver.Chrome()

self.driver.get()

deftest_screen(self):

print(self.driver.get_screenshot_as_png())

defteardown(self):

self.driver.quit()

if__name_