如何让Selenium脚本根据Chrome/Safari/Firefox浏览器差异化执行？

针对你的需求，这里有几种实用且易维护的实现方式，按推荐度排序：

1. 工厂模式 + 策略模式（最佳实践） #

这种方式把驱动创建和业务逻辑完全分离，扩展性极强，适合中大型项目。

步骤示例： #

• 先定义不同浏览器的逻辑策略类：

from selenium.webdriver.chrome.webdriver import WebDriver as ChromeDriver
from selenium.webdriver.firefox.webdriver import WebDriver as FirefoxDriver
from selenium.webdriver.safari.webdriver import WebDriver as SafariDriver

class ChromeBrowserStrategy:
    def execute_logic(self, driver):
        # Chrome专属逻辑，比如设置特定选项、处理Chrome独有的弹窗
        driver.execute_script("console.log('Chrome专属操作')")
        driver.maximize_window()

class FirefoxBrowserStrategy:
    def execute_logic(self, driver):
        # Firefox专属逻辑，比如处理Firefox的权限弹窗或安装插件
        driver.install_addon("/path/to/firefox/addon.xpi")
        print("Firefox专属操作执行完成")

class SafariBrowserStrategy:
    def execute_logic(self, driver):
        # Safari专属逻辑，比如适配浏览器特定的交互逻辑
        driver.execute_script("document.body.style.backgroundColor='lightblue'")
        print("Safari专属操作执行完成")

• 再写驱动工厂类，负责创建驱动并绑定对应策略：

class WebDriverFactory:
    @staticmethod
    def create_driver(browser_type):
        if browser_type == "chrome":
            driver = ChromeDriver()
            strategy = ChromeBrowserStrategy()
        elif browser_type == "firefox":
            driver = FirefoxDriver()
            strategy = FirefoxBrowserStrategy()
        elif browser_type == "safari":
            driver = SafariDriver()
            strategy = SafariBrowserStrategy()
        else:
            raise ValueError(f"不支持的浏览器类型：{browser_type}")
        # 把策略绑定到驱动实例，方便后续调用
        driver.strategy = strategy
        return driver

# 使用示例
driver = WebDriverFactory.create_driver("chrome")
driver.strategy.execute_logic(driver)

2. 直接判断WebDriver类型（小型脚本首选） #

如果你的脚本逻辑简单，直接用isinstance判断驱动类型即可，代码简洁直观：

def execute_browser_specific_logic(driver):
    if isinstance(driver, ChromeDriver):
        # Chrome专属逻辑
        driver.add_cookie({"name": "test", "value": "chrome"})
        print("执行Chrome逻辑")
    elif isinstance(driver, FirefoxDriver):
        # Firefox专属逻辑
        driver.set_page_load_timeout(30)
        print("执行Firefox逻辑")
    elif isinstance(driver, SafariDriver):
        # Safari专属逻辑
        driver.refresh()
        print("执行Safari逻辑")

# 使用示例
chrome_driver = ChromeDriver()
execute_browser_specific_logic(chrome_driver)

3. 字典映射逻辑函数（灵活轻量） #

用字典把驱动类型和对应的逻辑函数做映射，调用时直接根据类型取函数执行：

def chrome_logic(driver):
    print("Chrome专属操作")
    driver.get("https://www.google.com")

def firefox_logic(driver):
    print("Firefox专属操作")
    driver.get("https://www.mozilla.org")

def safari_logic(driver):
    print("Safari专属操作")
    driver.get("https://www.apple.com")

# 映射字典
browser_logic_map = {
    ChromeDriver: chrome_logic,
    FirefoxDriver: firefox_logic,
    SafariDriver: safari_logic
}

# 使用示例
driver = SafariDriver()
browser_logic_map[type(driver)](driver)

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