Python类中使用Selenium报错'driver'未定义，求解决方法

问题根源 #

你在__init__方法里定义的driver是局部变量，仅在__init__内部有效，check_email方法无法访问到它。要让类的其他方法能复用这个浏览器实例，必须把它定义为实例属性（用self.driver替代driver）。

额外优化点 #

• 类名CheckRegistation拼写错误，修正为CheckRegistration
• 用WebDriverWait显式等待替代time.sleep，提升代码稳定性
• 增加浏览器自动关闭逻辑，避免残留进程

修改后的完整代码 #

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

class CheckRegistration:
    def __init__(self, mail):
        self.mail = mail
        options = ChromeOptions()
        # 最大化窗口
        options.add_argument("--start-maximized")
        # 禁用自动化提示栏
        options.add_experimental_option("useAutomationExtension", False)
        options.add_experimental_option("excludeSwitches", ["enable-automation"])
        # 将driver绑定为实例属性，供类内其他方法调用
        self.driver = webdriver.Chrome(options=options)
        # 初始化显式等待对象，超时时间10秒
        self.wait = WebDriverWait(self.driver, 10)

    def check_email(self):
        try:
            self.driver.get("https://appleid.apple.com/account")
            # 显式等待邮箱输入框加载完成
            email_input = self.wait.until(EC.presence_of_element_located((By.XPATH, '//input[@type="email"]')))
            email_input.send_keys(self.mail)
            # 点击密码框触发邮箱可用性验证
            pwd_input = self.wait.until(EC.presence_of_element_located((By.XPATH, "//input[@id='password']")))
            pwd_input.click()
            # 等待验证提示加载
            time.sleep(1)
            page_source = self.driver.page_source
            search_text = " This email address is not available. Choose a different address. "
            if search_text in page_source:
                print("该邮箱已被注册，请更换其他邮箱！")
            else:
                print("该邮箱未被注册，可以用来创建Apple ID！")
        finally:
            # 无论操作成功与否，确保浏览器关闭
            self.driver.quit()

# 实例化并执行检测
p1 = CheckRegistration("xyz@gmail.com")
p1.check_email()

关键改动说明 #

1. 实例属性绑定：把driver改为self.driver，让整个类的方法都能访问浏览器实例
2. 显式等待：用WebDriverWait等待元素加载，避免因页面加载延迟导致的元素查找失败
3. 资源释放：try...finally块确保浏览器一定会被关闭，防止进程残留

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