一、Selenium之-文件上傳
通過input標(biāo)簽實現(xiàn)的上傳功能,可以將其看作是一個輸入框,即通過send_keys()指定本地文件路徑的方式實現(xiàn)文件上傳。
#!/usr/bin/env python
# -*- encoding: utf-8 -*-
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()
二、Selenium之-Cookie操作
網(wǎng)站為了辨別用戶身份、進(jìn)行 session 跟蹤而存儲在用戶本地終端上的數(shù)據(jù),也可以叫做瀏覽器緩存。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 信息
#!/usr/bin/env python
# -*- encoding: utf-8 -*-
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()
"""
cookies = browser.get_cookies() # 獲得所有的cookie
with open("cookies.txt", "w") as fp:
json.dump(cookies, fp)
browser.get("https://www.baidu.com")
browser.delete_all_cookies()
with open("cookies.txt", "r") as fp:
cookies = json.load(fp)
for cookie in cookies:
browser.add_cookie(cookie)
"""
# 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()
三、Selenium之-調(diào)用 JavaScript
對于 webdriver 中無法操作的動作(例如:滾動瀏覽器的側(cè)邊欄),可以調(diào)用 webdriver 進(jìn)行瀏覽器的控制。webdriver 提供了execute_script()方法來執(zhí)行 JavaScript 代碼。
#!/usr/bin/env python
# -*- encoding: utf-8 -*-
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()方法用于設(shè)置瀏覽器窗口滾動條的水平和垂直位置。方法的第一個參數(shù)表示水平的左間距,第二個參數(shù)表示垂直的上邊距。
browser.set_window_size(500, 500)
browser.find_element_by_id('kw').send_keys('百度')
browser.find_element_by_id('su').click()
sleep(2)
# 通過javascript設(shè)置瀏覽器窗口的滾動條位置
js = "window.scrollTo(100, 450);"
browser.execute_script(js)
sleep(2)
browser.quit()
四、Selenium之-窗口截圖
自動化用例是由程序去執(zhí)行的,因此有時候打印的錯誤信息并不十分明確。如果在腳本執(zhí)行出錯的時候能對當(dāng)前窗口截圖保存,那么通過圖片就可以非常直觀地看出出錯的原因。webdriver 提供了截圖函數(shù) get_screenshot_as_file() 來截取當(dāng)前窗口。
#!/usr/bin/env python
# -*- encoding: utf-8 -*-
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)
# 截取當(dāng)前窗口并指定報錯截圖的位置
# browser.get_screenshot_as_file('ScreenShot/14_screenShot.jpg')
browser.get_screenshot_as_file('ScreenShot/14_screenShot.png')
browser.quit()
五、Selenium之-關(guān)閉窗口
close() 關(guān)閉單個窗口
quit() 關(guān)閉所有窗口