[py] Use lazy imports in webdriver __init__.py

titusfortner · claude · titusfortner · commit acfa8ca89384 · 2026-01-23T22:32:08.000-06:00
Replace eager imports of all browser modules with lazy loading via
__getattr__. This enables future modularization of the Bazel build
and improves import performance for users who only use a subset of
browsers.

The public API is fully backwards compatible - all existing import
patterns continue to work identically.

Co-Authored-By: Claude Opus 4.5 &lt;noreply@anthropic.com&gt;
diff --git a/py/selenium/webdriver/__init__.py b/py/selenium/webdriver/__init__.py
@@ -15,6 +15,7 @@
 # specific language governing permissions and limitations
 # under the License.
 
+import importlib
 import logging
 import os
 
@@ -25,38 +26,62 @@
     if not logger.handlers:
         logger.addHandler(logging.StreamHandler())
 
-from selenium.webdriver.chrome.options import Options as ChromeOptions
-from selenium.webdriver.chrome.service import Service as ChromeService
-from selenium.webdriver.chrome.webdriver import WebDriver as Chrome
-from selenium.webdriver.common.action_chains import ActionChains
-from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
-from selenium.webdriver.common.keys import Keys
-from selenium.webdriver.common.proxy import Proxy
-from selenium.webdriver.edge.options import Options as EdgeOptions
-from selenium.webdriver.edge.service import Service as EdgeService
-from selenium.webdriver.edge.webdriver import WebDriver as ChromiumEdge
-from selenium.webdriver.edge.webdriver import WebDriver as Edge
-from selenium.webdriver.firefox.firefox_profile import FirefoxProfile
-from selenium.webdriver.firefox.options import Options as FirefoxOptions
-from selenium.webdriver.firefox.service import Service as FirefoxService
-from selenium.webdriver.firefox.webdriver import WebDriver as Firefox
-from selenium.webdriver.ie.options import Options as IeOptions
-from selenium.webdriver.ie.service import Service as IeService
-from selenium.webdriver.ie.webdriver import WebDriver as Ie
-from selenium.webdriver.remote.webdriver import WebDriver as Remote
-from selenium.webdriver.safari.options import Options as SafariOptions
-from selenium.webdriver.safari.service import Service as SafariService
-from selenium.webdriver.safari.webdriver import WebDriver as Safari
-from selenium.webdriver.webkitgtk.options import Options as WebKitGTKOptions
-from selenium.webdriver.webkitgtk.service import Service as WebKitGTKService
-from selenium.webdriver.webkitgtk.webdriver import WebDriver as WebKitGTK
-from selenium.webdriver.wpewebkit.options import Options as WPEWebKitOptions
-from selenium.webdriver.wpewebkit.service import Service as WPEWebKitService
-from selenium.webdriver.wpewebkit.webdriver import WebDriver as WPEWebKit
-
 __version__ = "4.41.0.202601181916"
 
-# We need an explicit __all__ because the above won't otherwise be exported.
+# Lazy import mapping: name -> (module_path, attribute_name)
+_LAZY_IMPORTS = {
+    # Chrome
+    "Chrome": ("selenium.webdriver.chrome.webdriver", "WebDriver"),
+    "ChromeOptions": ("selenium.webdriver.chrome.options", "Options"),
+    "ChromeService": ("selenium.webdriver.chrome.service", "Service"),
+    # Edge
+    "Edge": ("selenium.webdriver.edge.webdriver", "WebDriver"),
+    "ChromiumEdge": ("selenium.webdriver.edge.webdriver", "WebDriver"),
+    "EdgeOptions": ("selenium.webdriver.edge.options", "Options"),
+    "EdgeService": ("selenium.webdriver.edge.service", "Service"),
+    # Firefox
+    "Firefox": ("selenium.webdriver.firefox.webdriver", "WebDriver"),
+    "FirefoxOptions": ("selenium.webdriver.firefox.options", "Options"),
+    "FirefoxProfile": ("selenium.webdriver.firefox.firefox_profile", "FirefoxProfile"),
+    "FirefoxService": ("selenium.webdriver.firefox.service", "Service"),
+    # IE
+    "Ie": ("selenium.webdriver.ie.webdriver", "WebDriver"),
+    "IeOptions": ("selenium.webdriver.ie.options", "Options"),
+    "IeService": ("selenium.webdriver.ie.service", "Service"),
+    # Safari
+    "Safari": ("selenium.webdriver.safari.webdriver", "WebDriver"),
+    "SafariOptions": ("selenium.webdriver.safari.options", "Options"),
+    "SafariService": ("selenium.webdriver.safari.service", "Service"),
+    # Remote
+    "Remote": ("selenium.webdriver.remote.webdriver", "WebDriver"),
+    # WebKitGTK
+    "WebKitGTK": ("selenium.webdriver.webkitgtk.webdriver", "WebDriver"),
+    "WebKitGTKOptions": ("selenium.webdriver.webkitgtk.options", "Options"),
+    "WebKitGTKService": ("selenium.webdriver.webkitgtk.service", "Service"),
+    # WPEWebKit
+    "WPEWebKit": ("selenium.webdriver.wpewebkit.webdriver", "WebDriver"),
+    "WPEWebKitOptions": ("selenium.webdriver.wpewebkit.options", "Options"),
+    "WPEWebKitService": ("selenium.webdriver.wpewebkit.service", "Service"),
+    # Common utilities
+    "ActionChains": ("selenium.webdriver.common.action_chains", "ActionChains"),
+    "DesiredCapabilities": ("selenium.webdriver.common.desired_capabilities", "DesiredCapabilities"),
+    "Keys": ("selenium.webdriver.common.keys", "Keys"),
+    "Proxy": ("selenium.webdriver.common.proxy", "Proxy"),
+}
+
+
+def __getattr__(name):
+    if name in _LAZY_IMPORTS:
+        module_path, attr_name = _LAZY_IMPORTS[name]
+        module = importlib.import_module(module_path)
+        return getattr(module, attr_name)
+    raise AttributeError(f"module 'selenium.webdriver' has no attribute {name!r}")
+
+
+def __dir__():
+    return list(__all__) + list(globals().keys())
+
+
 __all__ = [
     "ActionChains",
     "Chrome",
