在Pytest框架中使用Selenium Python自动化代码时遇AttributeError问题

错误本质 #

报错AttributeError: 'NoneType' 对象没有属性 'find_element'，核心问题是**self.driver的值为None**——你没有在调用setUserName方法前，正确初始化或赋值Selenium驱动实例，导致调用驱动方法时出错。

常见修复方案 #

• 确保driver实例提前初始化
  在Pytest的前置方法（比如setup_method）里创建驱动实例并赋值给self.driver，保证调用setUserName时driver已经存在：

  from selenium import webdriver
  from selenium.webdriver.common.by import By
  
  class TestLogin:
      def setup_method(self):
          # 测试方法执行前初始化Chrome驱动
          self.driver = webdriver.Chrome()
          self.textbox_Username_ID = "实际的用户名输入框ID"  # 替换为页面真实ID
  
      def setUserName(self, username):
          self.driver.find_element(By.ID, self.textbox_Username_ID).send_keys(username)
  
      def test_login_flow(self):
          self.driver.get("你的登录页面URL")
          self.setUserName("测试账号")
  

• 排查是否有代码意外置空driver
  检查测试类的后置方法（比如teardown_method）或其他逻辑，是否错误地将self.driver赋值为None，导致后续调用时驱动实例丢失。

• 跨类调用时传递driver实例
  如果setUserName所在类和初始化driver的类不是同一个，要通过构造函数或参数把driver实例传递过去，避免出现未初始化的情况。

额外优化建议 #

• 确认self.textbox_Username_ID是页面上真实存在的元素ID，防止后续出现元素找不到的错误。
• 加入显式等待，确保元素加载完成后再操作，提升稳定性：

  from selenium.webdriver.support.ui import WebDriverWait
  from selenium.webdriver.support import expected_conditions as EC
  
  def setUserName(self, username):
      # 等待10秒，直到元素可见
      username_input = WebDriverWait(self.driver, 10).until(
          EC.visibility_of_element_located((By.ID, self.textbox_Username_ID))
      )
      username_input.send_keys(username)
  

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