Python Selenium使用EC判断按钮可点击时机早于视频上传完成问题求解

问题成因 #

• Selenium内置的EC.element_to_be_clickable判断逻辑仅校验两个条件：元素在DOM中可见、原生disabled属性为false。大部分业务场景下，上传过程中前端不会长时间给按钮加原生disabled属性，可能用自定义属性（比如aria-disabled="true"、自定义类名disabled）控制实际可用性，也可能按钮本身一直处于可点击状态，仅点击时会校验上传是否完成，这两种情况都会让内置判断提前通过。
• 文件上传是浏览器与服务端的异步交互过程，send_keys执行完成仅代表本地文件路径传入了上传控件，不代表上传到服务端完成，此时前端按钮状态可能还没同步更新，内置的状态判断无法感知后台上传请求的进度。
• 代码中while True + 捕获所有异常的逻辑存在漏洞：就算等待超时触发异常，也会立刻重试，一旦按钮临时满足可点击条件就会直接跳出循环，不会等上传完全结束。

解决方案 #

方案1：监听上传完成的专属标识（最稳定） #

先在前端页面定位到上传完成后才会出现/变化的特征，比如上传进度条消失、上传成功的提示图标出现、按钮新增upload-success类、隐藏域upload_status的值变为success等，优先等待这个标识出现，再判断按钮状态。

方案2：自定义按钮可点击的判断逻辑 #

如果按钮用自定义属性控制可用性，不要用内置的EC.element_to_be_clickable，自己写判断逻辑校验对应的自定义属性即可。

优化后的代码示例 #

以下示例默认上传完成后会出现类为upload-success-tip的成功提示，且按钮通过aria-disabled属性控制实际可用性，可根据自己的页面实际情况调整判断条件：

from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By

file_uploader = driver.find_element(By.XPATH, 'xpath_input')
file_uploader.send_keys(myvideo_file)

# 第一步：等待上传完成的标识出现
WebDriverWait(driver, 300).until(EC.presence_of_element_located((By.CLASS_NAME, 'upload-success-tip')))

# 第二步：自定义规则判断按钮实际可用
def button_is_available(driver):
    btn = driver.find_element(By.XPATH, 'button_xpath')
    return btn.is_displayed() and btn.get_attribute('aria-disabled') == 'false'

WebDriverWait(driver, 10).until(button_is_available)

print("Button ready to be clicked!")

内容的提问来源于stack exchange，提问作者EL-AJI Oussama