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

1. input 元素上傳文件

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

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

如果能直接在頁面當中看到這個input元素，那么通過 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 元素隱藏

通過修改元素屬性，把隱藏的元素屬性改掉。

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')

比如可以通過這種方式實現(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. 文件選擇對話框

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

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

在操作對話框之前，首先我們通過 selenium 點擊文件上傳的元素。

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

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

4. 使用 pywinauto 上傳文件

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

from pywinauto import Desktop

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

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

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

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')

注意：百度以圖搜圖禁用了爬蟲，所以在上傳文件時會提示「圖片上傳失敗，請重新上傳」。

6. 并發(fā)問題

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

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

最新評論