Allowed passing websockets connection kwargs

Author: voith

web3/providers/websocket.py

@@ -32,14 +32,15 @@ def get_default_endpoint():
 
 class PersistentWebSocket:
 
-    def __init__(self, endpoint_uri, loop):
+    def __init__(self, endpoint_uri, loop, **kwargs):
         self.ws = None
         self.endpoint_uri = endpoint_uri
         self.loop = loop
+        self.kwargs = kwargs
 
     async def __aenter__(self):
         if self.ws is None:
-            self.ws = await websockets.connect(uri=self.endpoint_uri, loop=self.loop)
+            self.ws = await websockets.connect(uri=self.endpoint_uri, loop=self.loop, **self.kwargs)
         return self.ws
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):
@@ -55,13 +56,13 @@ class WebsocketProvider(JSONBaseProvider):
     logger = logging.getLogger("web3.providers.WebsocketProvider")
     _loop = None
 
-    def __init__(self, endpoint_uri=None):
+    def __init__(self, endpoint_uri=None, **kwargs):
         self.endpoint_uri = endpoint_uri
         if self.endpoint_uri is None:
             self.endpoint_uri = get_default_endpoint()
         if WebsocketProvider._loop is None:
             WebsocketProvider._loop = _get_threaded_loop()
-        self.conn = PersistentWebSocket(self.endpoint_uri, WebsocketProvider._loop)
+        self.conn = PersistentWebSocket(self.endpoint_uri, WebsocketProvider._loop, **kwargs)
         super().__init__()
 
     def __str__(self):