yt_dlp/networking/_websockets.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import contextlib
 import io
 import logging
 import ssl
@@ -39,40 +38,27 @@
 import websockets.sync.client
 from websockets.uri import parse_uri
 
-# In websockets Connection, recv_exc and recv_events_exc are defined
-# after the recv events handler thread is started [1].
-# On our CI using PyPy, in some cases a race condition may occur
-# where the recv events handler thread tries to use these attributes before they are defined [2].
-# 1: https://github.com/python-websockets/websockets/blame/de768cf65e7e2b1a3b67854fb9e08816a5ff7050/src/websockets/sync/connection.py#L93
-# 2: "AttributeError: 'ClientConnection' object has no attribute 'recv_events_exc'. Did you mean: 'recv_events'?"
-import websockets.sync.connection  # isort: split
-with contextlib.suppress(Exception):
-    # > 12.0
-    websockets.sync.connection.Connection.recv_exc = None
-    # 12.0
-    websockets.sync.connection.Connection.recv_events_exc = None
-
 
 class WebsocketsResponseAdapter(WebSocketResponse):
 
-    def __init__(self, ws: websockets.sync.client.ClientConnection, url):
+    def __init__(self, wsw: websockets.sync.client.ClientConnection, url):
         super().__init__(
-            fp=io.BytesIO(ws.response.body or b''),
+            fp=io.BytesIO(wsw.response.body or b''),
             url=url,
-            headers=ws.response.headers,
-            status=ws.response.status_code,
-            reason=ws.response.reason_phrase,
+            headers=wsw.response.headers,
+            status=wsw.response.status_code,
+            reason=wsw.response.reason_phrase,
         )
-        self._ws = ws
+        self.wsw = wsw
 
     def close(self):
-        self._ws.close()
+        self.wsw.close()
         super().close()
 
     def send(self, message):
         # https://websockets.readthedocs.io/en/stable/reference/sync/client.html#websockets.sync.client.ClientConnection.send
         try:
-            return self._ws.send(message)
+            return self.wsw.send(message)
         except (websockets.exceptions.WebSocketException, RuntimeError, TimeoutError) as e:
             raise TransportError(cause=e) from e
         except SocksProxyError as e:
@@ -83,7 +69,7 @@ def send(self, message):
     def recv(self):
         # https://websockets.readthedocs.io/en/stable/reference/sync/client.html#websockets.sync.client.ClientConnection.recv
         try:
-            return self._ws.recv()
+            return self.wsw.recv()
         except SocksProxyError as e:
             raise ProxyError(cause=e) from e
         except (websockets.exceptions.WebSocketException, RuntimeError, TimeoutError) as e: