[py] Use generated Bidi files instead of hand curated ones

.gitignore

@@ -67,6 +67,7 @@ __pycache__
 .tox
 *.pyc
 dist/
+py/selenium/webdriver/common/bidi/
 py/selenium/webdriver/common/devtools/**/*
 !py/selenium/webdriver/common/devtools/util.py
 py/selenium/webdriver/common/linux/

py/BUILD.bazel

@@ -262,8 +262,11 @@ py_library(
 # BiDi protocol support
 py_library(
     name = "bidi",
-    srcs = glob(["selenium/webdriver/common/bidi/**/*.py"]),
-    data = [":mutation-listener"],
+    srcs = [],
+    data = [
+        ":create-bidi-src",
+        ":mutation-listener",
+    ],
     imports = ["."],
     visibility = ["//visibility:public"],
     deps = [
@@ -617,7 +620,7 @@ generate_devtools_latest(
     browser_versions = BROWSER_VERSIONS,
 )
 
-# Pilot BiDi code generation from CDDL specification
+# Generate BiDi source files from CDDL specification
 generate_bidi(
     name = "create-bidi-src",
     cddl_file = "//common/bidi/spec:all.cddl",

py/generate_bidi.py

@@ -1699,9 +1699,12 @@ def main(
 
     logger.info(f"Parsed {len(modules)} modules")
 
-    # Clean up existing generated files
+    # Clean up existing generated files.
+    # Keep static helper modules that are staged by Bazel (for example cdp.py)
+    # as part of create-bidi-src.extra_srcs.
+    preserved_python_files = {"py.typed", "cdp.py"}
     for file_path in output_path.glob("*.py"):
-        if file_path.name != "py.typed" and not file_path.name.startswith("_"):
+        if file_path.name not in preserved_python_files and not file_path.name.startswith("_"):
             file_path.unlink()
             logger.debug(f"Removed: {file_path}")
 

py/private/cdp.py

@@ -451,7 +451,13 @@ async def _reader_task(self):
             try:
                 data = json.loads(message)
             except json.JSONDecodeError:
-                raise BrowserError({"code": -32700, "message": "Client received invalid JSON", "data": message})
+                raise BrowserError(
+                    {
+                        "code": -32700,
+                        "message": "Client received invalid JSON",
+                        "data": message,
+                    }
+                )
             logger.debug("Received message %r", data)
             if "sessionId" in data:
                 session_id = devtools.target.SessionID(data["sessionId"])