make websockets compatible with python3.10

Signed-off-by: Harmouch101 <[email protected]>

Author: wiseaidev

setup.py

@@ -87,7 +87,7 @@
         "requests>=2.16.0,<3.0.0",
         # remove typing_extensions after python_requires>=3.8, see web3._utils.compat
         "typing-extensions>=3.7.4.1,<4;python_version<'3.8'",
-        "websockets>=9.1,<10",
+        "websockets>=9.1,<=10.0",
     ],
     python_requires='>=3.6,<4',
     extras_require=extras_require,

web3/providers/websocket.py

@@ -2,6 +2,7 @@
 import json
 import logging
 import os
+import sys
 from threading import (
     Thread,
 )
@@ -10,6 +11,7 @@
 )
 from typing import (
     Any,
+    Dict,
     Optional,
     Type,
     Union,
@@ -40,6 +42,13 @@
 DEFAULT_WEBSOCKET_TIMEOUT = 10
 
 
+def _loop_if_py_lt_38(loop: asyncio.AbstractEventLoop) -> Dict[str, Any]:
+    """
+    Helper for the removal of the loop argument in Python 3.10.
+    """
+    return {"loop": loop} if sys.version_info[:2] < (3, 8) else {}
+
+
 def _start_event_loop(loop: asyncio.AbstractEventLoop) -> None:
     asyncio.set_event_loop(loop)
     loop.run_forever()
@@ -70,7 +79,7 @@ def __init__(
     async def __aenter__(self) -> WebSocketClientProtocol:
         if self.ws is None:
             self.ws = await connect(
-                uri=self.endpoint_uri, loop=self.loop, **self.websocket_kwargs
+                uri=self.endpoint_uri, **_loop_if_py_lt_38(self.loop), **self.websocket_kwargs
             )
         return self.ws