[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

@@ -1,20 +1,26 @@
-# 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
+# The MIT License(MIT)
 #
-#   http://www.apache.org/licenses/LICENSE-2.0
+# Copyright(c) 2018 Hyperion Gray
 #
-# 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.
-
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files(the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and / or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+# This code comes from https://github.com/HyperionGray/trio-chrome-devtools-protocol/tree/master/trio_cdp
 
 import contextvars
 import importlib
@@ -54,7 +60,11 @@ def import_devtools(ver):
         # because cdp has been updated but selenium python has not been released yet.
         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()))
+        available_versions = tuple(
+            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:
             raise
@@ -65,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")
 
 
@@ -120,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):
@@ -217,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
 
@@ -233,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
@@ -274,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
@@ -285,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_
@@ -299,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:
@@ -425,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
@@ -438,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()
@@ -451,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"])