feat: add tm endpoint to network obj, add stream event example

Author: nhannamsiu

examples/chain_client/38_StreamEvent.py

@@ -0,0 +1,50 @@
+# Copyright 2021 Injective Labs
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Injective Exchange API client for Python. Example only."""
+
+import sys
+sys.path.insert(0, '/Users/nam/desktop/injective/sdk-python/')
+
+import asyncio
+import logging
+import json
+import websockets
+
+from pyinjective.async_client import AsyncClient
+from pyinjective.constant import Network
+
+async def main() -> None:
+    network = Network.mainnet()
+    client = AsyncClient(network, insecure=False)
+
+    event_filter = "tm.event='Tx' AND message.sender='inj1rwv4zn3jptsqs7l8lpa3uvzhs57y8duemete9e' AND message.action='/injective.exchange.v1beta1.MsgBatchUpdateOrders'"
+    query = json.dumps({
+        "jsonrpc": "2.0",
+        "method": "subscribe",
+        "id": "0",
+        "params": {
+            "query": event_filter
+        },
+    })
+
+    async with websockets.connect(network.tm_websocket_endpoint) as ws:
+        await ws.send(query)
+        while True:
+            events = await ws.recv()
+            print(events)
+            await asyncio.sleep(1)
+
+if __name__ == '__main__':
+    logging.basicConfig(level=logging.INFO)
+    asyncio.get_event_loop().run_until_complete(main())

pyinjective/constant.py

@@ -63,13 +63,15 @@ class Network:
     def __init__(
         self,
         lcd_endpoint: str ,
+        tm_websocket_endpoint: str,
         grpc_endpoint: str ,
         grpc_exchange_endpoint: str ,
         chain_id: str ,
         fee_denom: str ,
         env: str
     ):
         self.lcd_endpoint = lcd_endpoint
+        self.tm_websocket_endpoint = tm_websocket_endpoint
         self.grpc_endpoint = grpc_endpoint
         self.grpc_exchange_endpoint = grpc_exchange_endpoint
         self.chain_id = chain_id
@@ -80,6 +82,7 @@ def __init__(
     def devnet(cls):
         return cls(
             lcd_endpoint='https://devnet.lcd.injective.dev',
+            tm_websocket_endpoint='wss://devnet.tm.injective.dev/websocket',
             grpc_endpoint='devnet.injective.dev:9900',
             grpc_exchange_endpoint='devnet.injective.dev:9910',
             chain_id='injective-777',
@@ -91,6 +94,7 @@ def devnet(cls):
     def testnet(cls):
         return cls(
             lcd_endpoint='https://k8s.testnet.lcd.injective.network',
+            tm_websocket_endpoint='wss://k8s.testnet.tm.injective.network/websocket',
             grpc_endpoint='k8s.testnet.chain.grpc.injective.network:443',
             grpc_exchange_endpoint='k8s.testnet.exchange.grpc.injective.network:443',
             chain_id='injective-888',
@@ -102,7 +106,6 @@ def testnet(cls):
     def mainnet(cls, node='k8s'):
         nodes = [
             'k8s',
-            'lb',
             'sentry0',  # us, prod
             'sentry1',  # us, prod
             'sentry2',  # us, staging
@@ -111,17 +114,20 @@ def mainnet(cls, node='k8s'):
         if node not in nodes:
             raise ValueError('Must be one of {}'.format(nodes))
 
-        if node == 'lb' or node == 'k8s':
+        if node == 'k8s':
             lcd_endpoint='https://k8s.mainnet.lcd.injective.network'
+            tm_websocket_endpoint='wss://k8s.mainnet.tm.injective.network/websocket'
             grpc_endpoint='k8s.mainnet.chain.grpc.injective.network:443'
             grpc_exchange_endpoint='k8s.mainnet.exchange.grpc.injective.network:443'
         else:
             lcd_endpoint='https://lcd.injective.network'
+            tm_websocket_endpoint=f'ws://{node}.injective.network:26657/websocket'
             grpc_endpoint=f'{node}.injective.network:9900'
             grpc_exchange_endpoint=f'{node}.injective.network:9910'
 
         return cls(
             lcd_endpoint=lcd_endpoint,
+            tm_websocket_endpoint=tm_websocket_endpoint,
             grpc_endpoint=grpc_endpoint,
             grpc_exchange_endpoint=grpc_exchange_endpoint,
             chain_id='injective-1',
@@ -133,6 +139,7 @@ def mainnet(cls, node='k8s'):
     def local(cls):
         return cls(
             lcd_endpoint='http://localhost:10337',
+            tm_websocket_endpoint='ws://localost:26657/websocket',
             grpc_endpoint='localhost:9900',
             grpc_exchange_endpoint='localhost:9910',
             chain_id='injective-1',