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

Author: AutomatedTester

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/private/cdp.py

@@ -61,7 +61,9 @@ def import_devtools(ver):
         devtools_path = pathlib.Path(__file__).parents[1].joinpath("devtools")
         versions = tuple(f.name for f in devtools_path.iterdir() if f.is_dir())
         available_versions = tuple(
-            x for x in versions if x == "latest" or (x.startswith("v") and x[1:].isdigit())
+            x
+            for x in versions
+            if x == "latest" or (x.startswith("v") and x[1:].isdigit())
         )
         numeric_versions = tuple(x[1:] for x in available_versions if x.startswith("v"))
         if not numeric_versions:
@@ -73,7 +75,9 @@ def import_devtools(ver):
         return devtools
 
 
-_connection_context: contextvars.ContextVar = contextvars.ContextVar("connection_context")
+_connection_context: contextvars.ContextVar = contextvars.ContextVar(
+    "connection_context"
+)
 _session_context: contextvars.ContextVar = contextvars.ContextVar("session_context")
 
 
@@ -128,7 +132,9 @@ def set_global_connection(connection):
     certain use cases such as running inside Jupyter notebook.
     """
     global _connection_context
-    _connection_context = contextvars.ContextVar("_connection_context", default=connection)
+    _connection_context = contextvars.ContextVar(
+        "_connection_context", default=connection
+    )
 
 
 def set_global_session(session):
@@ -225,7 +231,9 @@ async def execute(self, cmd: Generator[dict, T, Any]) -> T:
             logger.debug(f"Received CDP message: {response}")
         if isinstance(response, Exception):
             if logger.isEnabledFor(logging.DEBUG):
-                logger.debug(f"Exception raised by {cmd_event} message: {type(response).__name__}")
+                logger.debug(
+                    f"Exception raised by {cmd_event} message: {type(response).__name__}"
+                )
             raise response
         return response
 
@@ -241,7 +249,9 @@ def listen(self, *event_types, buffer_size=10):
         return receiver
 
     @asynccontextmanager
-    async def wait_for(self, event_type: type[T], buffer_size=10) -> AsyncGenerator[CmEventProxy, None]:
+    async def wait_for(
+        self, event_type: type[T], buffer_size=10
+    ) -> AsyncGenerator[CmEventProxy, None]:
         """Wait for an event of the given type and return it.
 
         This is an async context manager, so you should open it inside
@@ -282,7 +292,9 @@ def _handle_cmd_response(self, data: dict):
         try:
             cmd, event = self.inflight_cmd.pop(cmd_id)
         except KeyError:
-            logger.warning("Got a message with a command ID that does not exist: %s", data)
+            logger.warning(
+                "Got a message with a command ID that does not exist: %s", data
+            )
             return
         if "error" in data:
             # If the server reported an error, convert it to an exception and do
@@ -293,7 +305,9 @@ def _handle_cmd_response(self, data: dict):
             # into a CDP object.
             try:
                 _ = cmd.send(data["result"])
-                raise InternalError("The command's generator function did not exit when expected!")
+                raise InternalError(
+                    "The command's generator function did not exit when expected!"
+                )
             except StopIteration as exit:
                 return_ = exit.value
             self.inflight_result[cmd_id] = return_
@@ -307,15 +321,19 @@ def _handle_event(self, data: dict):
         """
         global devtools
         if devtools is None:
-            raise RuntimeError("CDP devtools module not loaded. Call import_devtools() first.")
+            raise RuntimeError(
+                "CDP devtools module not loaded. Call import_devtools() first."
+            )
         event = devtools.util.parse_json_event(data)
         logger.debug("Received event: %s", event)
         to_remove = set()
         for sender in self.channels[type(event)]:
             try:
                 sender.send_nowait(event)
             except trio.WouldBlock:
-                logger.error('Unable to send event "%r" due to full channel %s', event, sender)
+                logger.error(
+                    'Unable to send event "%r" due to full channel %s', event, sender
+                )
             except trio.BrokenResourceError:
                 to_remove.add(sender)
         if to_remove:
@@ -433,8 +451,12 @@ async def connect_session(self, target_id) -> "CdpSession":
         """Returns a new :class:`CdpSession` connected to the specified target."""
         global devtools
         if devtools is None:
-            raise RuntimeError("CDP devtools module not loaded. Call import_devtools() first.")
-        session_id = await self.execute(devtools.target.attach_to_target(target_id, True))
+            raise RuntimeError(
+                "CDP devtools module not loaded. Call import_devtools() first."
+            )
+        session_id = await self.execute(
+            devtools.target.attach_to_target(target_id, True)
+        )
         session = CdpSession(self.ws, session_id, target_id)
         self.sessions[session_id] = session
         return session
@@ -446,7 +468,9 @@ async def _reader_task(self):
         """
         global devtools
         if devtools is None:
-            raise RuntimeError("CDP devtools module not loaded. Call import_devtools() first.")
+            raise RuntimeError(
+                "CDP devtools module not loaded. Call import_devtools() first."
+            )
         while True:
             try:
                 message = await self.ws.get_message()
@@ -459,7 +483,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"])