Python 3 Behave WebDriver OOP模式下Driver跨文件传递问题

先直接戳核心：你遇到的'WebDriver' object has no attribute 'driver'异常，大概率是在AutoTools.py的Navigation类里，错误地在已经是WebDriver实例的对象上再去访问.driver属性。咱们一步步拆解排查和解决：

1. 先检查Navigation类的实现逻辑 #

假设你的AutoTools.py里是类似下面的错误写法：

from selenium.webdriver.support.ui import WebDriverWait

class Navigation:
    def __init__(self, driver):
        self.driver = driver

    def wait_for_page_to_load(self):
        # 错误：这里写了self.driver.driver，但self.driver本身就是WebDriver实例
        WebDriverWait(self.driver.driver, 10).until(...)

正确的写法应该直接用self.driver（因为你传入的就是WebDriver实例本身）：

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

class Navigation:
    def __init__(self, driver):
        # 保存传入的WebDriver实例
        self.driver = driver

    def wait_for_page_to_load(self):
        # 直接用self.driver调用WebDriver的方法
        try:
            # 等待页面刷新完成
            WebDriverWait(self.driver, 15).until(
                EC.staleness_of(self.driver.find_element("tag name", "html"))
            )
        except:
            # 兼容无刷新的单页应用，等待页面加载状态完成
            WebDriverWait(self.driver, 15).until(
                lambda d: d.execute_script("return document.readyState") == "complete"
            )

2. 确认environment.py中实例传递的正确性 #

你在environment.py初始化Driver后，要确保传递给Navigation的是原始的WebDriver实例，而不是封装它的上下文对象或其他容器。比如：

正确的调用方式：

from selenium import webdriver
from Tools.AutoTools import Navigation

def before_scenario(context, scenario):
    # 初始化WebDriver实例
    context.driver = webdriver.Chrome()
    # 把WebDriver实例直接传给Navigation
    nav = Navigation(context.driver)
    # 调用等待方法
    nav.wait_for_page_to_load()

错误的调用方式（会引发类似问题）：

# 错误：传入的是context，而不是context.driver
nav = Navigation(context)
# 这时候要访问context.driver才是WebDriver实例，但如果方法里写了self.driver就会报错

3. 调试小技巧 #

如果还是不确定问题出在哪，可以在wait_for_page_to_load方法开头加一行打印，确认传入对象的类型：

def wait_for_page_to_load(self):
    print(f"当前driver的类型：{type(self.driver)}")
    # 后续代码...

如果输出是<class 'selenium.webdriver.chrome.webdriver.WebDriver'>，那绝对不能再用.driver去访问它的属性，直接用self.driver调用方法即可。

4. 优化OOP封装建议 #

如果想更规范地管理Driver实例，可以自己封装一个Driver类，避免传递混乱：

# Tools/AutoTools.py
from selenium import webdriver

class CustomDriver:
    def __init__(self):
        self.driver = webdriver.Chrome()

    def get_webdriver(self):
        return self.driver

    def quit(self):
        self.driver.quit()

class Navigation:
    def __init__(self, custom_driver):
        self.custom_driver = custom_driver
        self.driver = custom_driver.get_webdriver()

    def wait_for_page_to_load(self):
        # 这里用self.driver就不会出错
        ...

然后在environment.py里这样使用：

from Tools.AutoTools import CustomDriver, Navigation

def before_scenario(context, scenario):
    context.custom_driver = CustomDriver()
    nav = Navigation(context.custom_driver)
    nav.wait_for_page_to_load()

def after_scenario(context, scenario):
    context.custom_driver.quit()

这样整个结构更清晰，也不容易搞混实例类型。

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