Refactor with black/isort

Author: jman-sketch

py/selenium/webdriver/remote/network.py

@@ -15,15 +15,16 @@
 # specific language governing permissions and limitations
 # under the License.
 from dataclasses import fields
+
 import trio
+
 from selenium.webdriver.common.bidi import network
-from selenium.webdriver.common.bidi.browsing_context import Navigate, NavigateParameters
-from selenium.webdriver.common.bidi.network import (
-    AddInterceptParameters,
-    BeforeRequestSent,
-    BeforeRequestSentParameters,
-    ContinueRequestParameters,
-)
+from selenium.webdriver.common.bidi.browsing_context import Navigate
+from selenium.webdriver.common.bidi.browsing_context import NavigateParameters
+from selenium.webdriver.common.bidi.network import AddInterceptParameters
+from selenium.webdriver.common.bidi.network import BeforeRequestSent
+from selenium.webdriver.common.bidi.network import BeforeRequestSentParameters
+from selenium.webdriver.common.bidi.network import ContinueRequestParameters
 
 
 def default_request_handler(params: BeforeRequestSentParameters):
@@ -37,13 +38,17 @@ def __init__(self, driver):
         self.intercept = None
         self.scope = None
 
-    async def add_request_handler(self, request_filter=lambda _: True, handler=default_request_handler, conn=None):
+    async def add_request_handler(
+        self, request_filter=lambda _: True, handler=default_request_handler, conn=None
+    ):
         with trio.CancelScope() as scope:
             self.scope = scope
             self.network = network.Network(conn)
             params = AddInterceptParameters(["beforeRequestSent"])
             callback = self._callback(request_filter, handler)
-            result = await self.network.add_intercept(event=BeforeRequestSent, params=params)
+            result = await self.network.add_intercept(
+                event=BeforeRequestSent, params=params
+            )
             intercept = result["intercept"]
             self.intercept = intercept
             await self.network.add_listener(event=BeforeRequestSent, callback=callback)
@@ -54,7 +59,10 @@ async def get(self, url, conn):
         await conn.execute(Navigate(params).cmd())
 
     async def remove_request_handler(self):
-        await self.network.remove_intercept(event=BeforeRequestSent, params=network.RemoveInterceptParameters(self.intercept))
+        await self.network.remove_intercept(
+            event=BeforeRequestSent,
+            params=network.RemoveInterceptParameters(self.intercept),
+        )
         self.scope.cancel()
 
     def _callback(self, request_filter, handler):

py/test/selenium/webdriver/common/bidi_network_tests.py

@@ -15,10 +15,11 @@
 # specific language governing permissions and limitations
 # under the License.
 import pytest
+import trio
 
-from selenium.webdriver.common.bidi.network import BeforeRequestSentParameters, ContinueRequestParameters
 from selenium.webdriver.common.bidi.cdp import open_cdp
-import trio
+from selenium.webdriver.common.bidi.network import BeforeRequestSentParameters
+from selenium.webdriver.common.bidi.network import ContinueRequestParameters
 
 
 @pytest.mark.no_driver_after_test
@@ -35,7 +36,12 @@ def request_handler(params: BeforeRequestSentParameters):
     ws_url = driver.caps.get("webSocketUrl")
     async with open_cdp(ws_url) as conn:
         async with trio.open_nursery() as nursery:
-            nursery.start_soon(driver.network.add_request_handler, request_filter, request_handler, conn)
+            nursery.start_soon(
+                driver.network.add_request_handler,
+                request_filter,
+                request_handler,
+                conn,
+            )
             await trio.sleep(1)
             await driver.network.get("https://www.example.com", conn)
             assert "Selenium" in driver.title