Selenium封装find_element到函数中出现NoSuchElementError问题求助

问题根源 #

• 变量名不匹配：getuserbox函数接收的参数是name，但函数内部判断用的是未定义的webpage变量，导致逻辑错误，触发else分支使用错误的元素定位规则。
• 函数参数与调用不统一：getpwdbox定义的参数是webpage，但调用时传入的是name，参数名混淆容易引发逻辑误解。
• 依赖全局driver对象：函数直接使用全局的driver变量，代码耦合性高，多实例场景下易出错。
• 固定sleep不可靠：使用sleep(2)等固定等待时间，无法保证页面元素完全加载，可能导致元素未渲染就执行定位操作。
• 未定义变量pwd：代码中使用pwd但未初始化，会引发额外的NameError。

修复方案 #

1. 修正函数内的变量名：将getuserbox内的webpage替换为参数name，确保判断逻辑基于传入的标识名。
2. 统一函数参数名：将getpwdbox的参数名改为name，与调用时传入的参数一致，避免混淆。
3. 传入driver参数：将driver作为参数传入定位函数，消除全局变量依赖。
4. 使用显式等待：替换固定sleep为WebDriverWait，等待元素可被定位后再操作，提升稳定性。
5. 初始化pwd变量：通过getpass()获取密码，避免未定义错误。

修正后的代码 #

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

webpages = dict(zip(['name','name2', 'name3'],['webpage','weboage2', 'werbpage3']))
usr = 'user@gmail.com'
pwd = getpass("Enter password: ")

def getuserbox(driver, name): 
    if name == 'name': 
        foundelement = WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.ID, "mat-input-0"))
        )
    elif name == 'name2': 
        foundelement = WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.NAME, 'userId'))
        )
    else: 
        foundelement = WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.NAME, 'userid'))
        )
    return foundelement

def getpwdbox(driver, name): 
    if name == 'name': 
        foundelement = WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.ID, "mat-input-1"))
        )
    elif name == 'name2': 
        foundelement = WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.NAME, 'password'))
        )
    else: 
        foundelement = WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.ID, 'password'))
        )
    return foundelement

for name, page in webpages.items(): 
    driver = webdriver.Firefox() 
    driver.get(page) 
    print(f"Opened page {name}\n{page}")

    username_box = getuserbox(driver, name) 
    print('found box and trying to send usr') 
    username_box.send_keys(usr) 
    print("Email Id entered") 

    password_box = getpwdbox(driver, name) 
    password_box.send_keys(pwd) 
    # 可添加提交操作，比如 password_box.submit() 或者定位提交按钮点击

内容的提问来源于stack exchange，提问作者afk_philip