Python+Selenium嵌套页面对象优化：如何提升代码可读性？

核心优化思路 #

• 提取独立类，消除嵌套：把原来嵌套在函数、类内部的页面操作类拆成顶级类，每个类对应一个页面/视图，职责单一。
• 通过构造函数传递Driver：每个页面类创建时接收浏览器驱动实例，让不同页面操作共用同一个浏览器窗口，避免靠嵌套作用域共享变量的混乱。
• 分离初始化与业务逻辑：把驱动启动、页面打开的逻辑和页面元素操作逻辑分开，让页面类只专注于页面交互。
• 安全管理Driver生命周期：用try-finally或上下文管理器确保浏览器驱动能正确关闭，避免资源泄漏。

优化后代码示例 #

基础版（清晰易读）

from selenium import webdriver

# 任务列表页面操作类
class TasksView:
    def __init__(self, driver):
        self.driver = driver
        # 初始化时打开任务列表页面（也可移到外部统一处理）
        self.driver.get("你的任务列表页面URL")

    def get_task_names(self):
        # 实际逻辑：用driver.find_elements定位并提取任务名称
        print("读取所有任务名称并返回")
        return ["buy_cheese", "clean_room"]

    def open_task(self, name):
        # 实际逻辑：定位对应任务并点击打开
        print(f"打开任务: {name}")
        # 返回任务详情页面实例，传递同一个驱动
        return TaskView(self.driver)

# 任务详情页面操作类
class TaskView:
    def __init__(self, driver):
        self.driver = driver

    def get_details(self):
        # 实际逻辑：定位并提取任务详情
        print("返回任务详情")
        return "任务详情内容"

    def get_creation_date(self):
        # 实际逻辑：定位并提取创建日期
        print("返回任务创建日期")
        return "2024-05-20"

    def close(self):
        # 实际逻辑：关闭详情页，回到任务列表
        print("关闭任务详情，返回任务列表")

# 驱动初始化函数
def init_driver():
    # 根据你使用的浏览器选择对应驱动
    return webdriver.Chrome()

# 主执行逻辑
if __name__ == "__main__":
    driver = init_driver()
    try:
        tasks_view = TasksView(driver)
        # 获取任务列表
        task_names = tasks_view.get_task_names()
        print(f"当前任务列表: {task_names}")
        
        # 打开并操作单个任务
        task_view = tasks_view.open_task("buy_cheese")
        print(task_view.get_details())
        print(task_view.get_creation_date())
        task_view.close()
    finally:
        # 确保无论是否出错，驱动都会关闭
        driver.quit()

进阶版（用上下文管理器简化资源管理）

from selenium import webdriver
from contextlib import contextmanager

# 上下文管理器：自动管理驱动的启动与关闭
@contextmanager
def selenium_driver():
    driver = webdriver.Chrome()
    try:
        yield driver
    finally:
        driver.quit()

# 页面操作类同基础版，此处省略...

# 主执行逻辑
if __name__ == "__main__":
    with selenium_driver() as driver:
        tasks_view = TasksView(driver)
        task_view = tasks_view.open_task("buy_cheese")
        print(task_view.get_details())
        task_view.close()

优化后的优势 #

• 可读性大幅提升：每个类独立存在，结构清晰，一眼就能区分不同页面的操作逻辑。
• 扩展性更强：后续新增页面（比如任务编辑页），只需要新增对应的操作类即可，不会影响现有代码。
• 维护成本降低：每个方法职责单一，修改某页面的操作逻辑时，只需改动对应类的方法，不会牵连其他部分。
• 资源更安全：通过try-finally或上下文管理器，避免浏览器驱动因代码异常而无法关闭的问题。

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