Selenium上传文件实战代码整合windows第三方包pywinauto

#! /usr/bin/env pyhton
# -*- coding:utf-8 -*-
# author:Jeff.xie
# datetime:2023/9/12 14:35
# software:PyCharm

from selenium import webdriver
from time import sleep
import os

from selenium.webdriver.common.by import By

url="https://hkwbl.sharepoint.com/sites/Tech/VB/Forms/AllItems.aspx?ct=1692956030163&or=OWA%2DNT&cid=13a37e7e%2D4ef6%2Dad2d%2D4edc%2Dc081b48edd23&ga=1&WSL=1&id=%2Fsites%2FTech%2FVB%2FQA%2FTest%20Automation%2FReport&viewid=7c6a613e%2Dd057%2D4c3f%2Db986%2Deeb2a4cd2e58"

def find_ele(by_type, loc):
    try:
        return driver.find_element(by_type, loc)
    except:
        print('did not find the element: ' + loc)
        return None

def click_ele(by_type, loc):
    find_ele(by_type, loc).click()


def intput_send_keys(by_type, loc, data):
    find_ele(by_type, loc).send_keys(data)

driver = webdriver.Chrome()
driver.get(url)
driver.maximize_window()
sleep(3)
account_xpath = '//input[@name="loginfmt"]'
# account_xpath = '//input[@aria-label="请输入电子邮件、电话或 Skype。"]'
driver.find_element(By.XPATH, account_xpath).send_keys("[email protected]")

ele_path = '//*[contains(text(),"登录")]'
# Python文本元素定位，与Java不同 '//*[contains(text(),"登录")]'
# Java文本定位方式：            '//*[contains(@text,"登录")]'
driver.find_element(By.XPATH, ele_path)
# driver.find_element(By.PARTIAL_LINK_TEXT,"登录")
next_button_xpath = '//input[@value="下一步"]'
click_ele(By.XPATH, next_button_xpath)
sleep(3)
password_input_xpath = '//input[@placeholder="密码"]'
placeholder="密码"
intput_send_keys(By.XPATH, password_input_xpath, "Welab1209!")

login_xpath = '//input[@value="登录"]'
click_ele(By.XPATH, login_xpath)
sleep(3)

yes_xpath = '//input[@value="是"]'
click_ele(By.XPATH, yes_xpath)
sleep(10)

upload_xpath = '//button[@name="上传"]'
click_ele(By.XPATH, upload_xpath)
sleep(2)

file_item_xpath = '//button[@name="文件"]'
click_ele(By.XPATH, file_item_xpath)
sleep(2)


#  参考文档 https://blog.csdn.net/YLF123456789000/article/details/131475893
# selenium上传文件

# pip installl pywinauto 安装第三方包
from pywinauto import Desktop

file_path = "D:\\"
filename = "0001.png"
app = Desktop()
dialog = app['打开']  # 根据名字找到弹出窗口
dialog["Edit"].type_keys(file_path + filename)  # 在输入框中输入值
dialog["Button"].click()

# upload_result_ele_xpath = '//button[@text='+filename+']'
upload_result_ele_xpath = '//*[contains(text(),"'+filename+'")]'  #根据文本定位元素

find_ele(By.XPATH, upload_result_ele_xpath)
for i in range(5):
    ele = find_ele(By.XPATH, upload_result_ele_xpath)
    if ele is not None:
        print('find the element success: ' + upload_result_ele_xpath)
        break
    sleep(5)
    # print('did not find the element: '+upload_result_ele_xpath)
    driver.refresh()
    if i == 4:
        raise Exception

# driver.get("file:///D:/0001.png")
file_path =r"D:/0001.png"
upload_ele_class = 'od-ItemsScopeItemContent-list od-ItemContent-list'
# ele = find_ele(By.CLASS_NAME, upload_ele_class).upload_by_drop(file_path) # 上传文件不生效
# driver.get("file:///D:/0001.png") # 上传文件不生效

sleep(5)

# Python文本元素定位，与Java不同 '//*[contains(text(),"登录")]'
# Java文本定位方式：            '//*[contains(@text,"登录")]'

Selenium 封装了现成的文件上传操作。但是随着现代前端框架的发展，文件上传的方式越来越多样。而有一些文件上传的控件，要做自动化控制会更复杂一些，这篇文章主要讨论在复杂情况下，如何通过自动化完成文件上传

1.input 元素上传文件

如果页面需要文件上传，那么在大多数情况下，都能在页面源代码中找到一个 input 的元素。

如果能直接在页面当中看到这个 input 元素，那么通过 selenium 的 send_keys 方法就能完成文件的上传，在参数中传入本地文件的路径。

driver.get('')
 
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('')
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 热键粘贴到文件路径输入框当中。

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 方法实现文件上传。

Selenium上传文件有多少种方式？不信你有我全_软件测试潇潇的博客-CSDN博客