[py] Add type stubs for lazy imported classes and modules

py/generate_webdriver_stub.py

@@ -0,0 +1,71 @@
+# Licensed to the Software Freedom Conservancy (SFC) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The SFC licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+
+"""Generate selenium/webdriver/__init__.pyi from the runtime lazy imports."""
+
+from __future__ import annotations
+
+import importlib
+from pathlib import Path
+
+PACKAGE = "selenium.webdriver"
+OUTPUT_FILE = Path(__file__).resolve().parent / "selenium" / "webdriver" / "__init__.pyi"
+
+
+def main() -> None:
+    pkg = importlib.import_module(PACKAGE)
+
+    lazy_imports: dict[str, tuple[str, str]] = getattr(pkg, "_LAZY_IMPORTS")
+    lazy_submodules: dict[str, str] = getattr(pkg, "_LAZY_SUBMODULES")
+
+    lines: list[str] = []
+    lines.append("# This file is auto-generated. DO NOT EDIT.")
+    lines.append("# Generated by py/generate_webdriver_stub.py")
+    lines.append("# ruff: noqa: F401, I001")
+    lines.append("")
+
+    # Generate re-exported symbols
+    for public_name in sorted(lazy_imports):
+        module_path, attr_name = lazy_imports[public_name]
+
+        if public_name == attr_name:
+            line = f"from {module_path} import {attr_name}"
+        else:
+            line = f"from {module_path} import {attr_name} as {public_name}"
+
+        lines.append(line)
+
+    lines.append("")
+
+    # Generate submodule re-exports
+    if lazy_submodules:
+        lines.append("from . import (")
+        for name in sorted(lazy_submodules):
+            lines.append(f"    {name},")
+        lines.append(")")
+        lines.append("")
+
+    lines.append("__all__: list[str]")
+
+    OUTPUT_FILE.write_text("\n".join(lines) + "\n", encoding="utf-8")
+
+    print(f"Stub written to: {OUTPUT_FILE}")
+
+
+if __name__ == "__main__":
+    main()

py/selenium/webdriver/__init__.pyi

@@ -0,0 +1,48 @@
+# This file is auto-generated. DO NOT EDIT.
+# Generated by py/generate_webdriver_stub.py
+# ruff: noqa: F401, I001
+
+from selenium.webdriver.common.action_chains import ActionChains
+from selenium.webdriver.chrome.webdriver import WebDriver as Chrome
+from selenium.webdriver.chrome.options import Options as ChromeOptions
+from selenium.webdriver.chrome.service import Service as ChromeService
+from selenium.webdriver.edge.webdriver import WebDriver as ChromiumEdge
+from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
+from selenium.webdriver.edge.webdriver import WebDriver as Edge
+from selenium.webdriver.edge.options import Options as EdgeOptions
+from selenium.webdriver.edge.service import Service as EdgeService
+from selenium.webdriver.firefox.webdriver import WebDriver as Firefox
+from selenium.webdriver.firefox.options import Options as FirefoxOptions
+from selenium.webdriver.firefox.firefox_profile import FirefoxProfile
+from selenium.webdriver.firefox.service import Service as FirefoxService
+from selenium.webdriver.ie.webdriver import WebDriver as Ie
+from selenium.webdriver.ie.options import Options as IeOptions
+from selenium.webdriver.ie.service import Service as IeService
+from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.common.proxy import Proxy
+from selenium.webdriver.remote.webdriver import WebDriver as Remote
+from selenium.webdriver.safari.webdriver import WebDriver as Safari
+from selenium.webdriver.safari.options import Options as SafariOptions
+from selenium.webdriver.safari.service import Service as SafariService
+from selenium.webdriver.wpewebkit.webdriver import WebDriver as WPEWebKit
+from selenium.webdriver.wpewebkit.options import Options as WPEWebKitOptions
+from selenium.webdriver.wpewebkit.service import Service as WPEWebKitService
+from selenium.webdriver.webkitgtk.webdriver import WebDriver as WebKitGTK
+from selenium.webdriver.webkitgtk.options import Options as WebKitGTKOptions
+from selenium.webdriver.webkitgtk.service import Service as WebKitGTKService
+
+from . import (
+    chrome,
+    chromium,
+    common,
+    edge,
+    firefox,
+    ie,
+    remote,
+    safari,
+    support,
+    webkitgtk,
+    wpewebkit,
+)
+
+__all__: list[str]