Merge pull request pyppeteer#252 from mborsetti/websockets

Mattwmaster58 · web-flow · commit 0fd15a7c4b2d · 2021-06-12T12:15:24.000-04:00
support websockets 9.1
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyppeteer/connection.py b/pyppeteer/connection.py
@@ -10,6 +10,7 @@
 
 from pyee import EventEmitter
 import websockets
+from websockets.legacy.client import connect as ws_connect
 
 from pyppeteer.errors import NetworkError
 
@@ -40,8 +41,7 @@ def __init__(self, url: str, loop: asyncio.AbstractEventLoop,
         self._sessions: Dict[str, CDPSession] = dict()
         self.connection: CDPSession
         self._connected = False
-        self._ws = websockets.client.connect(
-            self._url, max_size=None, loop=self._loop, ping_interval=None, ping_timeout=None)
+        self._ws = ws_connect(self._url, max_size=None, loop=self._loop, ping_interval=None, ping_timeout=None)
         self._recv_fut = self._loop.create_task(self._recv_loop())
         self._closeCallback: Optional[Callable[[], None]] = None
 
diff --git a/pyproject.toml b/pyproject.toml
@@ -51,7 +51,7 @@ importlib-metadata = { version = "^2.1.1", python = "<3.8" }
 pyee = "^8.1.0"
 tqdm = "^4.42.1"
 urllib3 = "^1.25.8"
-websockets = "^8.1"
+websockets = "^9.1"
 
 [tool.poetry.dev-dependencies]
 tox = "^3.20.1"
