### 在 Flask 应用中多个接口之间共享 Selenium 创建的 driver 对象 在 Flask 应用中，多个接口之间共享由 Selenium 创建的 `driver` 对象可以通过使用全局变量、上下文管理器或更复杂的依赖注入框架来实现。以下是一个详细的解决方案： #### 1. 使用全局变量存储 `driver` 对象 通过将 `driver` 对象存储为全局变量，可以在多个接口中访问它。需要确保 `driver` 对象仅初始化一次，并且在应用关闭时正确关闭。 ```python from flask import Flask from selenium import webdriver app = Flask(__name__) driver = None @app.before_first_request def initialize_driver(): global driver if not driver: options = webdriver.ChromeOptions() options.add_argument("--headless") # 可选：无头模式 driver = webdriver.Chrome(options=options) @app.teardown_appcontext def close_driver(exception=None): global driver if driver: driver.quit() @app.route("/test1") def test1(): global driver driver.get("https://example.com") title = driver.title return f"Page Title: {title}" @app.route("/test2") def test2(): global driver element = driver.find_element_by_id("element-id") text = element.text return f"Element Text: {text}" ``` 此方法简单直接，但不适合大型项目，因为全局变量可能导致代码难以维护[^1]。 #### 2. 使用上下文管理器和 Flask 的扩展 Flask 提供了多种扩展机制，可以使用 `flask.g` 来存储请求范围内的数据。然而，由于 `driver` 是整个应用生命周期内共享的对象，建议结合上下文管理器和全局变量。 ```python from flask import Flask, g from selenium import webdriver app = Flask(__name__) def get_driver(): if "driver" not in g: options = webdriver.ChromeOptions() options.add_argument("--headless") g.driver = webdriver.Chrome(options=options) return g.driver @app.teardown_appcontext def close_driver(exception=None): driver = g.pop("driver", None) if driver: driver.quit() @app.route("/test1") def test1(): driver = get_driver() driver.get("https://example.com") title = driver.title return f"Page Title: {title}" @app.route("/test2") def test2(): driver = get_driver() element = driver.find_element_by_id("element-id") text = element.text return f"Element Text: {text}" ``` 这种方法利用了 Flask 的上下文管理功能，同时避免了直接使用全局变量[^3]。 #### 3. 使用依赖注入框架（如 Injector） 对于更复杂的项目，可以引入依赖注入框架来管理 `driver` 对象的生命周期。 ```python from flask import Flask from injector import Injector, Module, provider, singleton from selenium import webdriver app = Flask(__name__) class SeleniumModule(Module): @singleton @provider def provide_driver(self) -> webdriver.Chrome: options = webdriver.ChromeOptions() options.add_argument("--headless") return webdriver.Chrome(options=options) injector = Injector(SeleniumModule()) @app.route("/test1") def test1(): driver = injector.get(webdriver.Chrome) driver.get("https://example.com") title = driver.title return f"Page Title: {title}" @app.route("/test2") def test2(): driver = injector.get(webdriver.Chrome) element = driver.find_element_by_id("element-id") text = element.text return f"Element Text: {text}" @app.teardown_appcontext def close_driver(exception=None): driver = injector.get(webdriver.Chrome) driver.quit() ``` 这种方法适合大规模项目，能够清晰地管理对象的依赖关系和生命周期[^2]。 ### 注意事项 - 确保 `driver` 对象在多线程环境中正确同步。 - 使用 `--headless` 模式可以避免启动实际的浏览器窗口，适用于服务器端运行环境。 - 如果 `driver` 对象长时间未被使用，可以考虑自动销毁并重新创建以节省资源。