Python中Selenium上傳文件的幾種方式

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

1. input 元素上傳文件

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

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

如果能直接在頁面當中看到這個input元素,那麼通過 selenium 的 send_keys 方法就能完成文件的上傳,在參數中傳入本地文件的路徑。

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

比如可以通過這種方式實現百度的以圖搜圖。

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元素進行過多的分析,那麼比較直接的方式是采用文件上傳對話框來處理。

一般來說,如果需要上傳文件,那麼當你點擊這個元素之後,會出現一個文件上傳的對話框,要求你選擇文件,並點擊確定。這個對話框是屬於系統的,因此selenium 無法直接控制它。我們可以采用系統的自動化工具或者直接調用鍵盤來操作這種對話框。

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

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

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

4. 使用 pywinauto 上傳文件

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

from pywinauto import Desktop

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

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

但是這款工具目前不支持中文的輸入,因此我們需要借助剪切板來實現中輸入。首先我們把對應的中文復制到剪切板當中,然後再通過 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. 並發問題

通過系統窗口上傳文件簡單粗暴,但是當你的程序需要並發執行時,使用這種方式實現文件上傳就比較麻煩瞭。如果你的程序需要並發執行,最好還是通過控制input元素,使用 send_keys 方法實現文件上傳。

到此這篇關於Python中Selenium上傳文件的幾種方式的文章就介紹到這瞭,更多相關Python Selenium上傳文件內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: