helius-python 0.5.0__py3-none-any.whl → 0.5.2__py3-none-any.whl

helius/laserstream/websockets.py

@@ -132,6 +132,10 @@ class WebSocketClient:
         base_url="wss://mainnet.helius-rpc.com",
         api_key: str | None = None,
         proxy: str | None = None,
+        open_timeout: float | None = 10,
+        ping_interval: float | None = 20,
+        ping_timeout: float | None = 20,
+        close_timeout: float | None = 10,
     ):
 
         base_url = base_url
@@ -144,7 +148,14 @@ class WebSocketClient:
         if not api_key:
             raise ValueError("No API key provided.")
         uri = httpx.URL(base_url).copy_with(path="/", params={"api-key": api_key})
-        self._websocket = connect(str(uri), proxy=proxy)
+        self._websocket = connect(
+            str(uri),
+            proxy=proxy,
+            open_timeout=open_timeout,
+            ping_interval=ping_interval,
+            ping_timeout=ping_timeout,
+            close_timeout=close_timeout,
+        )
 
     def close(self):
         self._websocket.close()
@@ -163,6 +174,27 @@ class WebSocketClient:
         response = self._websocket.recv()
         return json.loads(response)
 
+    def receive(self) -> tuple[dict | None, Notification, int]:
+        response = json.loads(self._websocket.recv())
+        model = self.MODELS[response["method"]]
+        result = response["params"]["result"]
+        subscription = response["params"]["subscription"]
+        if isinstance(result, dict):
+            context = result.get("context")
+            value = result.get("value")
+        else:
+            context, value = None, None
+        if value is not None:
+            notification = model.model_validate(value)
+        else:
+            notification = model.model_validate(result)
+        return context, notification, subscription
+
+    def listen(self):
+        while True:
+            context, notification, subscription = self.receive()
+            yield context, notification, subscription
+
     def _unsubscribe(self, subscription_type, subscription) -> bool:
         request = (
             JsonRpcRequest(method=f"{subscription_type}Unsubscribe")
@@ -376,24 +408,3 @@ class WebSocketClient:
 
     def vote_unsubscribe(self, subscription) -> bool:
         return self._unsubscribe("vote", subscription)
-
-    def receive(self) -> tuple[dict | None, Notification, int]:
-        response = json.loads(self._websocket.recv())
-        model = self.MODELS[response["method"]]
-        result = response["params"]["result"]
-        subscription = response["params"]["subscription"]
-        if isinstance(result, dict):
-            context = result.get("context")
-            value = result.get("value")
-        else:
-            context, value = None, None
-        if value is not None:
-            notification = model.model_validate(value)
-        else:
-            notification = model.model_validate(result)
-        return context, notification, subscription
-
-    def listen(self):
-        while True:
-            context, notification, subscription = self.receive()
-            yield context, notification, subscription

{helius_python-0.5.0.dist-info → helius_python-0.5.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: helius-python
-Version: 0.5.0
+Version: 0.5.2
 Summary: Typed Python client for the Helius API
 Project-URL: Homepage, https://github.com/markosnarinian/helius-python
 Project-URL: Issues, https://github.com/markosnarinian/helius-python/issues

{helius_python-0.5.0.dist-info → helius_python-0.5.2.dist-info}/RECORD

@@ -1,7 +1,7 @@
 helius/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 helius/admin/__init__.py,sha256=HboRJXZNzuDAc6r3kj2QZtEcu2U3EWJHv81dYDJGjaA,262
 helius/admin/admin.py,sha256=dcmOgeu10Q43QZC8K-a25vp2JNAb4hs5gu3YWaovV4Y,2541
-helius/laserstream/websockets.py,sha256=cndqxXO9_I4aSjAeSkz3Q3u80jrFR0JMfXw09vzbgvQ,12448
+helius/laserstream/websockets.py,sha256=zk3vpSORateUpi0s6Jq-MSBA384G9tp8bvJ_f5liHts,12809
 helius/rpc/__init__.py,sha256=sW31r7_8PRn-bQePJSbIK8NvlkMor0b59E4rnvvUT9I,992
 helius/rpc/client.py,sha256=zgubEGlSHAs-bHRSL0AP-X5c4l4oKJzoeriBM4DJnwk,37498
 helius/rpc/models.py,sha256=_o9B9VBgNBkYGGeglEgWxkffwS1thym2KfAfPekm590,7682
@@ -9,7 +9,7 @@ helius/utils/__init__.py,sha256=szG5WTfKl1dfqsZb2efmeZr5zJBKlEBgNCWYGCo7VbU,87
 helius/utils/json_rpc_request.py,sha256=hm7WkPo0CnwMnuHE4qR-d93wGArZRPRm-B3uueykL3E,1479
 helius/webhooks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 helius/webhooks/webhooks.py,sha256=IXCF7sDS6QpcY5r03WTmXNpOqoltii3pSHPita0f83o,19803
-helius_python-0.5.0.dist-info/METADATA,sha256=k69sDfATDw_3K3qINHmRc3eEVCG7VoZX8aJlDPUdx7I,37939
-helius_python-0.5.0.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
-helius_python-0.5.0.dist-info/licenses/LICENSE,sha256=bZc2EDmq_GsWH77uK8LBn1kqqrOPcp2f9p-ys9bYa1E,1072
-helius_python-0.5.0.dist-info/RECORD,,
+helius_python-0.5.2.dist-info/METADATA,sha256=13MGhAUwyrVv9XYfaslwOYqQcWCYiH5H0yEBVfSxGPs,37939
+helius_python-0.5.2.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
+helius_python-0.5.2.dist-info/licenses/LICENSE,sha256=bZc2EDmq_GsWH77uK8LBn1kqqrOPcp2f9p-ys9bYa1E,1072
+helius_python-0.5.2.dist-info/RECORD,,