python+selenium的web自動化上傳操作的實現

2022-08-08 22:02:37

一、關於上傳操作

上傳有兩種情況：

如果是input可以直接輸入路徑的，那麼直接使用send_keys(檔案路徑)輸入路徑即可；
非input標籤的上傳，則需要藉助第三方工具：第三方庫 pywin32、第三方工具pyautogui等等。

那這裡針對以上兩種情況分別介紹一下具體的解決方法。

二、input標籤

定位到元素，然後直接使用send_keys(檔案路徑)輸入路徑，比較簡單。

from selenium import webdriver
driver = webdriver.Chrome()
driver.get("http://xxxx")
driver.maximize_window()
 
# 定位上傳檔案元素
input_file = driver.find_element_by_id("xxx")
# 上傳檔案
input_file.send_keys(r"D:learnxx.txt")

三、第三方庫pywin32

非input標籤的上傳，則需要藉助第三方工具。pywin32就是這樣的一個第三方庫，主要是方便python開發者快速呼叫windows API的一個模組庫，可以識別上傳對話方塊控制程式碼進而操作。

安裝命令：pip install pywin32。

 從上面的上傳視窗，及WinSpy工具視窗，可以得出控制關係圖如下：

 從上圖可以看出：

檔名輸入框的路徑為：上傳視窗 - > comboBoxEx32 - >comboBox - > Edit，在第四級edit；
按鈕的路徑為：上傳視窗 - > button(開啟按鈕)，在第二級。

我們需要一步一步定位，從第一層開始往下去找，找到對應的視窗然後進行操作，步驟說明如下：

 第一步：先定位到第一層大視窗（即整個上傳視窗），使用win32gui模組下的FindWindow()，它是專門用於查詢大視窗的；
 第二步：在大視窗下查詢子視窗，使用win32gui模組下的FindWindowEx()，查詢大視窗以下的子視窗都是用這個方法；
第三步：定位到最後要操作的視窗後，就開始要執行編輯、點選這個操作了，不管是編輯還是點選都是使用win32gui模組下的SendMessage()。

以上三步的相關函數說明如下：

win32gui.FindWindow(IpClassName,IpWindowName)：自頂層視窗開始尋找匹配條件的視窗，並返回這個視窗的控制程式碼IpClassName：類名，在Spy++裡能夠看到IpWindowName：視窗名，標題列上能看到的名字
win32gui.FindWindowEx(hwndParent=0,hwndChildAfter=O,IpClassName=None,IpWindowName=None)：搜尋類名和表單名匹配的表單，並返回這個表單的控制程式碼，找不到就返回0hwndParent：若不為0，則搜尋控制程式碼為hwndParent表單的子表單hwndChildAfter：若不為0，則按照z-index的順序從hwndChildAfter向後開始搜尋子表單，否則從第一個子表單開始搜尋IpClassName：字元型，是表單的類名，這個可以在Spy++裡能夠看到IpWindowName：字元型，是視窗名，就是標題列上能看到的名字
win32gui.SendMessage(hWnd, Msg, wParam, IParam)：hWnd：整型，接收訊息的表單控制程式碼Msg：整型，要傳送的訊息，這些訊息都是windows預先定義好的wParam：整型，訊息的wParam引數IParam：整型，訊息的IParam引數

以下是封裝好的上傳函數，因為window的上傳視窗是固定不變的，我們只需要封裝好一個類知道如何去呼叫即可，不需要理解它的處理細節，但如果你有興趣可以深入瞭解一下。

import win32gui
import win32con
 
# edit - combobox - comboBoxEx32 - #32770  編輯框在第四級
# button - #32770  開啟按鈕在第二級
 
# 前提：window上傳視窗已經出現，最好sleep 1-2秒等待視窗出現
 
def upload(filepath, browser_type="chrome"):
    if browser_type == "chrome":
        title = "開啟"
    else:
        title = ""
    # 找元素
    # 從一級開始找，一級視窗「#32770」，「開啟」
    dialog = win32gui.FindWindow("#32770", title)  # FindWindow用於找大視窗
 
    # 二級之後都用FindWindowEx，需要四個引數，
    # 1、元素的父親，2、從第一個子代開始找元素，3、元素的型別名（class），4、元素的文字值
    comboBoxEx32 = win32gui.FindWindowEx(dialog, 0, "ComboBoxEx32", None)  # 二級
    comBox = win32gui.FindWindowEx(comboBoxEx32, 0, "ComboBox", None)  # 三級
    # 編輯框
    edit = win32gui.FindWindowEx(comBox, 0, 'Edit', None)  # 四級
    # 開啟按鈕
    button = win32gui.FindWindowEx(dialog, 0, 'Button', '開啟(&0)')  # 二級
 
    # 往編輯框輸入檔案路徑
    win32gui.SendMessage(edit, win32con.WM_SETTEXT, None, filepath)  # 傳送檔案路徑
    win32gui.SendMessage(dialog, win32con.WM_COMMAND, 1, button)  # 點選開啟按鈕
if __name__ == '__main__':
    upload('D:\test.txt')

四、第三方工具pyautogui

還有一種更方便的方法，就是藉助第三方工具pyautogui，pyautogui是一個純Python的GUI自動化工具，其目的是可以用程式自動控制滑鼠和鍵盤操作，並且支援多平臺（Windows，OS X，Linux）。強烈推薦！

下面以舉例的html為例，需要先把該程式碼儲存成html檔案，後續直接存取這個本地html檔案：

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>
<h1> hello world </h1>
<input type="file" name="mfile">
</body>
</html>

安裝模組： pip install pyautogui，

操作只需要兩步：

第一步：將要上傳的檔案路徑寫入到檔名的輸入框內；
第二步：敲回車，相當於點選【開啟】按鈕，注意回車presses要2次才能生效。

import os
import time
import pyautogui as ui
from selenium import webdriver
 
driver = webdriver.Chrome()
driver.implicitly_wait(20)      # 隱式等待
dir = os.getcwd()
html_dir = os.path.join(dir, 'demo.html')
driver.get(html_dir)
 
file_elem = driver.find_element_by_name('mfile')
file_elem.click()
 
# 系統之間要等待
time.sleep(1)
file_dir = os.path.join(dir, 'test.txt')	# 要上傳的檔案路徑
# 寫入檔案
ui.write(file_dir)	# 直接呼叫寫入
time.sleep(0.5)
# 輸入確認鍵, presses表示按的次數，按一次不會生效，有可能是執行太快，建議presses=2
ui.press('enter', presses=2)
time.sleep(2)
driver.quit()

不管是使用哪種上傳方式，只要是作業系統的上傳視窗，操作之前記得先等待一下。