Selenium 封裝了現(xiàn)成的文件上傳操作。但是隨著現(xiàn)代前端框架的發(fā)展，文件上傳的方式越來(lái)越多樣。而有一些文件上傳的控件，要做自動(dòng)化控制會(huì)更復(fù)雜一些，這篇文章主要討論在復(fù)雜情況下，如何通過(guò)自動(dòng)化完成文件上傳。

1. input 元素上傳文件

如果頁(yè)面需要文件上傳，那么在大多數(shù)情況下，都能在頁(yè)面源代碼中找到一個(gè)input的元素。

<input type="file" name="file_name">

如果能直接在頁(yè)面當(dāng)中看到這個(gè)input元素，那么通過(guò) selenium 的 send_keys 方法就能完成文件的上傳，在參數(shù)中傳入本地文件的路徑。

driver.get('<https://testpages.herokuapp.com/styled/file-upload-test.html>')

el = driver.find_element('id', "fileinput")
el.send_keys('/path/of/file.png')

2. input 元素隱藏

通過(guò)修改元素屬性，把隱藏的元素屬性改掉。

el = driver.find_element('xpath', '//input[@type="file"]')
driver.execute_script('arguments[0].style.visibility=\\'visible\\'', el)
el.send_keys(r'C:\\Users\\muji\\Desktop\\avatar.png')

比如可以通過(guò)這種方式實(shí)現(xiàn)百度的以圖搜圖。

driver.get('<http://www.baidu.com>')
driver.find_element('css selector', '.soutu-btn').click()
time.sleep(3)
el = driver.find_element('xpath', '//input[@type="file"]')
driver.execute_script('arguments[0].style.visibility=\\'visible\\'', el)
el.send_keys(r'C:\\Users\\muji\\Desktop\\avatar.png')

3. 文件選擇對(duì)話框

對(duì)于有一些元素，直接通過(guò) selenium 自帶的 send_keys 方法上傳文件并不會(huì)成功。如果不想對(duì)input元素進(jìn)行過(guò)多的分析，那么比較直接的方式是采用文件上傳對(duì)話框來(lái)處理。

一般來(lái)說(shuō)，如果需要上傳文件，那么當(dāng)你點(diǎn)擊這個(gè)元素之后，會(huì)出現(xiàn)一個(gè)文件上傳的對(duì)話框，要求你選擇文件，并點(diǎn)擊確定。這個(gè)對(duì)話框是屬于系統(tǒng)的，因此selenium 無(wú)法直接控制它。我們可以采用系統(tǒng)的自動(dòng)化工具或者直接調(diào)用鍵盤來(lái)操作這種對(duì)話框。

在操作對(duì)話框之前，首先我們通過(guò) selenium 點(diǎn)擊文件上傳的元素。

el = driver.find_element('id', "fileinput")
ActionChains(driver).click(el).perform() 

input 元素并不能點(diǎn)擊，所以不能用元素的 el.click() 方法，需要使用 ActionChains 下面的 click 方法。他們之間的區(qū)別在于元素的 el.click 方法更加嚴(yán)格，會(huì)對(duì)元素是否可見，是否可點(diǎn)擊進(jìn)行檢測(cè)，在點(diǎn)擊事件完全生效后，再執(zhí)行下面的操作，如果不滿足這些條件，可能會(huì)報(bào)錯(cuò)。 而 Action 下的 click 方法則粗暴得多，它幾乎不對(duì)對(duì)元素進(jìn)行檢測(cè)，直接把鼠標(biāo)移動(dòng)到元素上方，執(zhí)行點(diǎn)擊操作，至于點(diǎn)擊是否生效，根本不管。

4. 使用 pywinauto 上傳文件

pywinauto 是Windows系統(tǒng)下的一個(gè)自動(dòng)化工具,它可以直接獲取到Windows系統(tǒng)下面的彈框,因此當(dāng)文件上傳的窗口出現(xiàn)之后，我們可以使用這個(gè)工具傳入文件的路徑,然后點(diǎn)擊打開按鈕。

from pywinauto import Desktop

app = Desktop()
dialog = app['打開'] ?# 根據(jù)名字找到彈出窗口
dialog["Edit"].type_keys('/path/of/file.md') ?# 在輸入框中輸入值
dialog["Button"].click()

另一種系統(tǒng)自動(dòng)化工具叫做 pyautogui。這個(gè)工具最大的特點(diǎn)在于使用坐標(biāo)系統(tǒng)定位元素，可以輕而易舉的做到跨平臺(tái)。不管你是Windows，mac還是Linux，都可以使用這款工具實(shí)現(xiàn)自動(dòng)化。

但是這款工具目前不支持中文的輸入，因此我們需要借助剪切板來(lái)實(shí)現(xiàn)中輸入。首先我們把對(duì)應(yīng)的中文復(fù)制到剪切板當(dāng)中，然后再通過(guò) ctrl + v 熱鍵粘貼到文件路徑輸入框當(dāng)中。

5. pyautogui

?import pyperclip

?pyperclip.copy('D:\\\\用戶.html')
?pyautogui.hotkey('ctrl', 'v')
?pyautogui.press('enter', presses=2)

鍵盤

keyboard.write('C:\\\\Users\\\\muji\\\\Desktop\\\\avatar.png')
time.sleep(1)
keyboard.press('enter')

注意：百度以圖搜圖禁用了爬蟲，所以在上傳文件時(shí)會(huì)提示「圖片上傳失敗，請(qǐng)重新上傳」。

6. 并發(fā)問(wèn)題

通過(guò)系統(tǒng)窗口上傳文件簡(jiǎn)單粗暴，但是當(dāng)你的程序需要并發(fā)執(zhí)行時(shí)，使用這種方式實(shí)現(xiàn)文件上傳就比較麻煩了。如果你的程序需要并發(fā)執(zhí)行，最好還是通過(guò)控制input元素，使用 send_keys 方法實(shí)現(xiàn)文件上傳。

到此這篇關(guān)于Python中Selenium上傳文件的幾種方式的文章就介紹到這了,更多相關(guān)Python Selenium上傳文件內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論