ruff format

Author: lawrence-forooghian

.github/workflows/check.yml

@@ -21,7 +21,7 @@ jobs:
           python-version: "3.x"
 
       - run: npm ci
-      - run: pip install mitmproxy ruff mypy
+      - run: pip install mitmproxy ruff mypy # TODO I’m not sure what is the right way of doing this
 
       # Run JS checks
       - run: npm run format:check

src/python/mitmproxy_addon.py

@@ -4,14 +4,16 @@
 import websockets
 import json
 
+
 async def send_ready_notification() -> None:
     uri = "ws://localhost:8001"
-    logging.info(f'sending mitmproxyReady JSON-RPC notification to {uri}')
+    logging.info(f"sending mitmproxyReady JSON-RPC notification to {uri}")
     async with websockets.connect(uri) as websocket:
-        notification_dto = { "jsonrpc": "2.0", "method": "mitmproxyReady" }
+        notification_dto = {"jsonrpc": "2.0", "method": "mitmproxyReady"}
         data = json.dumps(notification_dto)
         await websocket.send(data)
 
+
 class MitmproxyAddon:
     def running(self) -> None:
         # tell the control API that we’re ready to receive traffic
@@ -21,12 +23,14 @@ def running(self) -> None:
     def request(self, flow: http.HTTPFlow) -> None:
         # To make sure that when running in local redirect mode (and hence intercepting all traffic from the test process) we don’t mess with traffic from the test process to the control API
         # TODO see extended comments re this in test-node.yml and why it hasn’t yet been an issue in practice on macOS
-        if not flow.request.port in [80, 443]:
+        if flow.request.port not in [80, 443]:
             return
 
         # (b'Connection', b'Upgrade'), (b'Upgrade', b'websocket')
         intercept = MitmproxyAddon.is_websocket_upgrade_request(flow.request)
-        logging.info(f'MitmproxyAddon {"intercepting" if intercept else "not intercepting"} `request` {flow.request.url}, headers {flow.request.headers}')
+        logging.info(
+            f'MitmproxyAddon {"intercepting" if intercept else "not intercepting"} `request` {flow.request.url}, headers {flow.request.headers}'
+        )
         # pretty_host takes the "Host" header of the request into account,
         # which is useful in transparent mode where we usually only have the IP
         # otherwise.
@@ -38,18 +42,26 @@ def request(self, flow: http.HTTPFlow) -> None:
 
             flow.request.host = "localhost"
             flow.request.port = 8002
-            flow.request.scheme = 'http'
+            flow.request.scheme = "http"
             # TODO understand how port fits into this
-            flow.request.headers['Ably-Test-Host'] = original_host
+            flow.request.headers["Ably-Test-Host"] = original_host
             match original_scheme:
-                case 'http':
-                    flow.request.headers['Ably-Test-Proto'] = 'ws'
-                case 'https':
-                    flow.request.headers['Ably-Test-Proto'] = 'wss'
+                case "http":
+                    flow.request.headers["Ably-Test-Proto"] = "ws"
+                case "https":
+                    flow.request.headers["Ably-Test-Proto"] = "wss"
 
     @staticmethod
     def is_websocket_upgrade_request(request: http.Request) -> bool:
         # TODO this request handling is a bit fragile, the special case for `split` is just to handle the fact that Firefox sends 'Connection: keep-alive, Upgrade'
-        return True if 'Connection' in request.headers and ('Upgrade' in request.headers['Connection'].split(", ")) and 'Upgrade' in request.headers and request.headers['Upgrade'] == 'websocket' else False
+        return (
+            True
+            if "Connection" in request.headers
+            and ("Upgrade" in request.headers["Connection"].split(", "))
+            and "Upgrade" in request.headers
+            and request.headers["Upgrade"] == "websocket"
+            else False
+        )
+
 
 addons = [MitmproxyAddon()]