Selenium(十一)：設置元素等待、上傳文件、下載文件

1. 設置元素等待

前面咱們接觸了幾個元素等待方法，sleep、implicitly_wait方法，這一章咱們就來總體學一下。

如今大多數Web應用程序使用的都是AJAX技術。當瀏覽器加載頁面時，頁面上的元素可能並非同時被加載完成的，這給元素的定位增長了困難。若是由於在加載某個元素時延遲而形成ElementNotVisibleException的狀況出現，那麼就會下降自動化腳本的穩定性，咱們能夠經過設置元素等待，來改善這種問題形成的不穩定。

WebDriver提供了兩種類型的等待：顯示等待和隱式等待。

1.1 顯示等待

顯式等待使WebDriver等待某個條件處理時繼續執行，不然在達到最大時長時拋棄超時異常(TimeoutException)。

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

wd = webdriver.Chrome()

wd.get('https://www.baidu.com/')
a = WebDriverWait(wd,10).until(EC.presence_of_element_located((By.ID,"kw")))
a.send_keys('selenium')

WebDriverWait類是由WebDriver提供的等待方法。在設置時間內，默認每隔一段時間檢測一次當前頁面元素是否存在，若是超過設置時間檢測不到則拋出異常。

WebDriverWait(driver, timeout, poll_frequency=POLL_FREQUENCY, ignored_exceptions=None)

driver：瀏覽器驅動

timeout：最長超時時間，默認以秒爲單位

poll_frequency：檢測的間隔(步長)時間，默認爲0.5S

ignored_exceptions：超時後的異常信息，默認狀況下拋NoSuchElementException異常

WebDriverWait()通常和until()或until_not()方法配合使用。

until(method, message='')

調用該方法提供的驅動程序做爲一個參數，直到返回值爲True。

until_not(method, message='')

調用該方法提供的驅動程序做爲一個參數，直到返回值爲False。

在上面那個例子中，經過as關鍵字將expected_conditions重命名爲EC，並調用presence_of_element_located()方法判斷元素是否存在。

expected_conditions類提供的預期條件判斷的方法以下所示：

title_is：判斷當前頁面的標題是否等於預期

title_contains：判斷當前頁面的標題是否包含預期字符串

presence_of_element_located：判斷元素是否被加在DOM樹裏，並不表明元素必定可見

visibility_of_element_located：判斷元素是否可見(可見表明元素非隱藏，而且元素的寬和高都不等於0)

visibility_of：與上一個方法做用相同，只是上一個方法參數爲，該方法接收的參數爲定位後的元素

presence_of_all_elements_located：判斷是否至少有一個元素存在於DOM樹中。例如，在頁面中有n個元素的class爲「wp」，那麼只要有一個存在就返回True

text_to_be_present_in_element：判斷某個元素中的text是否包含了預期的字符串

text_to_be_present_in_element_value：判斷某個元素的value屬性是否包含了預期的字符串

frame_to_be_available_and_switch_to_it：判斷該表單是否可用切換進去，若是可用，返回True而且switch進去，不然返回False

invisibility_of_element_located：判斷某個元素是否不存在於DOM樹或不可見

element_to_be_clickable：判斷元素是否可見而且是能夠點擊的

staleness_of：等到一個元素從DOM樹中移除

element_to_be_selected：判斷某個元素是否被選中，通常用在下拉列表

element_selection_state_to_be：判斷某個元素的選擇狀態是否符合預期

element_located_selection_state_to_be：與上一個方法做用相同，只是上一個方法參數爲單位後的元素，該方法接收的參數爲定位

alert_is_present：判斷頁面上是否存在alert

除expected_conditions所提供的豐富的預期條件判斷方法外，還可使用is_displayed()方法來判斷元素是否可見。 

from selenium import webdriver
from time import  sleep,ctime

wd = webdriver.Chrome()

wd.get('https://www.baidu.com/')

print(ctime())
for i in range(10):
    try:
        el = wd.find_element_by_id("kw22")
        if el.is_displayed():
            break
    except:pass
    sleep(1)
else:
    print("time out")
wd.close()
print(ctime())

相對來講，這種方式更容易理解，經過for循環10次，每次循環判斷元素的is_displayed()狀態是否爲True。若是爲True，則break跳出循環；不然sleep(1)後繼續循環判斷，直到10次循環結束後，打印「time out」信息。

執行結果以下：

 

1.2 隱式等待

隱式等待是經過必定的時長等待頁面上某元素加載完成。若是超出了設置的時長元素尚未被加載，則拋出NoSuchElementException異常。WebDriver提功力implicitly_wait()方法來實現隱式等待，默認設置爲0。它的用法相對來講要簡單得多。 

from selenium import webdriver
from selenium.common.exceptions import  NoSuchElementException
from time import ctime

wd = webdriver.Chrome()

#設置隱式等待爲10秒
wd.implicitly_wait(10)
wd.get('https://www.baidu.com/')

try:
    print(ctime())
    wd.find_element_by_id("kw22").send_keys('selenium')
except NoSuchElementException as e:
    print(e)
finally:
    print(ctime())

Implicitly_wait()默認參數的單位爲秒，本例中設置等待時長爲10秒。首先這10秒並不是一個固定的等待時間。它並不影響腳本的執行速度。其次，它並不針對頁面上的某一個元素進行等待。當腳本執行到某個元素定位時，若是元素能夠定位，則繼續執行；若是元素定位不到，則它將以循環查詢的方式不斷地判斷元素是否被定位到。假設在第3秒定位到了元素，則繼續執行，若知道超出設置時長(10秒)尚未定位到元素，則拋出異常。

在上面的例子中，顯然百度輸入框的定位id=kw22是有誤的，經過打印的兩次時間能夠看出，當執行對百度輸入框的操做時，超過了10秒的等待。

1.3 sleep休眠方法

在前面咱們就使用過sleep方法了，只不過沒有詳細的介紹，如今就把這個納入到這個大標題中來介紹下。

有時候咱們但願腳本在執行到某一個位置時作固定時間的休眠，尤爲是在腳本調試中。這時可使用sleep()方法，須要說明的是，sleep()方法由python的time模塊提供。 

from selenium import webdriver
from time import sleep

wd = webdriver.Chrome()

wd.get('https://www.baidu.com/')

sleep(2)
wd.find_element_by_id("kw").send_keys("selenium")
wd.find_element_by_id("su").click()
sleep(5)

wd.quit()

當執行到sleep()方法時會固定休眠必定的時長，而後再繼續執行。sleep()方法默認參數以秒爲單位，若是設置時長小於1秒，則能夠用小數表示，若是sleep(0.5)表示休眠0.5秒。

2. 上傳文件

上傳文件是比較常見的Web功能之一，但WebDriver並無提供專門用於上傳的方法，如何實現上傳操做關鍵在於上傳文件的思路。

通常Web頁面的上傳功能的操做須要點擊「上傳」按鈕後打開本地的Window窗口，從窗口中選擇本地文件進行上傳。而WebDriver是沒法操做Windows控件的，因此，對於初學者來講，通常思路會卡在如何識別Window控件這個問題上。

對於Web頁面的上傳功能實現通常有一下兩種方式。

普通上傳：普通的附件上傳是將本地文件的路徑做爲一個值放在input標籤中，經過form表單將這個值提交給服務器。

插件上傳：通常是指基於Flash、JavaScript或Ajax等技術所實現的上傳功能。

插件上傳不適合放在自動化裏面講解，可能會放在python高級編程裏面講解。對於經過input標籤實現的上傳功能，能夠將其看做是一個輸入框，經過send_keys()指定本地文件路徑的方式實現文件上傳。

html代碼：

<!DOCTYPE html>
<html lang="en">
    <head>
        <meta charset="UTF-8">
        <title></title>
    </head>
    <body>
        <div class="row-fluid">
            file：<input type="file" name="file" />
        </div>
    </body>
</html>

經過瀏覽器打開，效果以下圖所示：

 

from selenium import webdriver

wd = webdriver.Chrome()

wd.get('http://127.0.0.1:8020/day01/index.html')

#定位上傳按鈕，添加本地文件
wd.find_element_by_name("file").send_keys('E:\\1.png')

經過這種方法上傳，就避免了操做Windows控件的步驟。若是能找到上傳的input標籤，那麼基本上就能夠經過send_keys()方法向其輸入一個文件地址來實現上傳。

3. 下載文件 

WebDriver容許咱們設置默認的文件下載路徑，也就是說，文件會自動下載而且存放到設置的目錄中。下面以谷歌瀏覽器爲例，執行文件的下載。

from selenium import webdriver
import os

options = webdriver.ChromeOptions()

prefs = {'profile.default_content_settings.popups':0,'download.default_directory':'E:\\'}
options.add_experimental_option('prefs',prefs)

wd = webdriver.Chrome(executable_path=r'E:\webdrivers\chromedriver.exe',chrome_options=options)

wd.get('')#輸入要下載的文件所在網頁
wd.find_element_by_xpath('').click()#點擊文件所在位置的元素

download.default_directory：指定路徑
profile.default_content_settings.popups：0 爲屏蔽彈窗，1 爲開啓彈窗

不一樣的瀏覽器設置方法也不一樣，以上例子值針對谷歌瀏覽器。火狐瀏覽器的下載設置就是其餘類型的方法了。