Python中Selenium上傳檔案的幾種方式

2022-07-22 18:01:30

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上傳檔案內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！