1. 程式人生 > 實用技巧 >Selenium Web自動化之檔案上傳,Cookie,呼叫 js,錯誤截圖

Selenium Web自動化之檔案上傳,Cookie,呼叫 js,錯誤截圖

一、檔案上傳

from selenium import webdriver
import os
 
 
driver = webdriver.Chrome('../tools/chromedriver.exe')
file_path = "file:///" + os.path.abspath('upfile.html')
driver.get(file_path)
 
# 定位上傳按鈕的位置
driver.find_element_by_name('file').send_keys(os.path.abspath('upfile.txt'))
driver.quit()

二、Cookie操作

網站為了辨別使用者身份、進行 session 跟蹤而儲存在使用者本地終端上的資料,也可以叫做瀏覽器快取。webdriver 對 cookie 的常用操作有新增、刪除、讀取。

(1) get_cookies()-----獲得所有的 cookie 資訊
(2) get_cookie(name)-----活得 key 值為 name 的 cookie 的資訊
(3) add_cookie(cookie_dict)----新增 cookie。"cookie_dict" 指字典物件,必須有 name 和 value 值
(4) delete_cookie(name,optionsString):刪除cookie資訊。“name”是要刪除的cookie的名稱,“optionsString”是該cookie的選項,目前支援的選項包括“路徑”,“域”
(5) delete_all_cookies()----刪除所有 cookie 資訊

from selenium import webdriver
from time import sleep
 
 
base_url = 'https://www.baidu.com/'
browser = webdriver.Chrome('../tools/chromedriver.exe')
browser.get(base_url)
 
# 1. 獲取 cookie 資訊
cookies = browser.get_cookies()
print(cookies)
sleep(2)
browser.quit()
 
# 2. cookie 寫入 browser.add_cookie( { 'name': 'add-cookie', 'value': 'add-cookie-value' } ) # 遍歷cookies列印cookie資訊 for cookie in browser.get_cookies(): print("%s ---> %s" % (cookie['name'], cookie['value'])) sleep(2) browser.quit()

三、呼叫 JavaScript

對於 webdriver 中無法操作的動作(例如:滾動瀏覽器的側邊欄),可以呼叫 webdriver 進行瀏覽器的控制。webdriver 提供了execute_script()方法來執行 JavaScript 程式碼。

from selenium import webdriver
from time import sleep
 
 
base_url = 'https://www.baidu.com'
browser = webdriver.Chrome('../tools/chromedriver.exe')
browser.get(base_url)
 
# window.scrollTo()方法用於設定瀏覽器視窗滾動條的水平和垂直位置。方法的第一個引數表示水平的左間距,第二個引數表示垂直的上邊距。
browser.set_window_size(500, 500)
browser.find_element_by_id('kw').send_keys('百度')
browser.find_element_by_id('su').click()
sleep(2)
 
# 通過javascript設定瀏覽器視窗的滾動條位置
js = "window.scrollTo(100, 450);"
browser.execute_script(js)
sleep(2)
 
browser.quit()

四、視窗截圖

自動化用例是由程式去執行的,因此有時候列印的錯誤資訊並不十分明確。如果在指令碼執行出錯的時候能對當前視窗截圖儲存,那麼通過圖片就可以非常直觀地看出出錯的原因。webdriver 提供了截圖函式 get_screenshot_as_file() 來擷取當前視窗。

from selenium import webdriver
from time import sleep
 
base_url = 'http://www.baidu.com/'
browser = webdriver.Chrome('../tools/chromedriver.exe')
 
browser.get(base_url)
 
browser.find_element_by_id('kw').send_keys('python selenium')
browser.find_element_by_id('su').click()
sleep(2)
 
# 擷取當前視窗並指定報錯截圖的位置
# browser.get_screenshot_as_file('ScreenShot/14_screenShot.jpg')
browser.get_screenshot_as_file('ScreenShot/14_screenShot.png')
 
browser.quit()