(feat) Regenerated proto files based on Injective Core with support for chain streams

Author: abel

pyinjective/async_client.py

@@ -57,10 +57,11 @@
     injective_portfolio_rpc_pb2 as portfolio_rpc_pb,
     injective_portfolio_rpc_pb2_grpc as portfolio_rpc_grpc,
 )
-
 from .proto.injective.types.v1beta1 import (
     account_pb2
 )
+from pyinjective.proto.injective.stream.v1beta1 import query_pb2 as chain_stream_query
+from pyinjective.proto.injective.stream.v1beta1 import query_pb2_grpc as stream_rpc_grpc
 
 from .core.network import Network
 from .utils.logger import LoggerProvider
@@ -145,6 +146,13 @@ def __init__(
             self.explorer_channel
         )
 
+        self.chain_stream_channel = (
+            grpc.aio.secure_channel(network.chain_stream_endpoint, credentials)
+            if (network.use_secure_connection and credentials is not None)
+            else grpc.aio.insecure_channel(network.chain_stream_endpoint)
+        )
+        self.chain_stream_stub = stream_rpc_grpc.StreamStub(channel=self.chain_stream_channel)
+
         # timeout height update routine
         self.cron = aiocron.crontab(
             "* * * * * */{}".format(DEFAULT_TIMEOUTHEIGHT_SYNC_INTERVAL),
@@ -928,6 +936,20 @@ async def stream_account_portfolio(self, account_address: str, **kwargs):
         )
         return self.stubPortfolio.StreamAccountPortfolio(request=req, metadata=metadata)
 
+    async def chain_stream(
+        self,
+        bank_balances_filter: Optional[chain_stream_query.BankBalancesFilter] = None,
+        subaccount_deposits_filter: Optional[chain_stream_query.SubaccountDepositsFilter] = None
+    ):
+
+        request = chain_stream_query.StreamRequest(
+            bank_balances_filter=bank_balances_filter,
+            subaccount_deposits_filter=subaccount_deposits_filter)
+        metadata = await self.network.chain_metadata(
+            metadata_query_provider=self._chain_cookie_metadata_requestor
+        )
+        return self.chain_stream_stub.Stream(request=request, metadata=metadata)
+
     async def composer(self):
         return Composer(
             network=self.network.string(),

pyinjective/composer.py

@@ -34,6 +34,7 @@
 from pyinjective.proto.cosmos.base.v1beta1 import coin_pb2 as cosmos_dot_base_dot_v1beta1_dot_coin__pb2
 
 from .proto.cosmwasm.wasm.v1 import tx_pb2 as wasm_tx_pb
+from pyinjective.proto.injective.stream.v1beta1 import query_pb2 as chain_stream_query
 
 from .constant import ADDITIONAL_CHAIN_FORMAT_DECIMALS, INJ_DENOM
 from typing import Dict, List, Optional
@@ -901,6 +902,14 @@ def MsgVote(
             proposal_id=proposal_id, voter=voter, option=option
         )
 
+    def chain_stream_bank_balances_filter(self, accounts: List[str]) -> chain_stream_query.BankBalancesFilter:
+        return chain_stream_query.BankBalancesFilter(accounts=accounts)
+
+    def chain_stream_subaccount_deposits_filter(
+            self, subaccount_ids: List[str]
+    ) -> chain_stream_query.SubaccountDepositsFilter:
+        return chain_stream_query.SubaccountDepositsFilter(subaccount_ids=subaccount_ids)
+
     # data field format: [request-msg-header][raw-byte-msg-response]
     # you need to figure out this magic prefix number to trim request-msg-header off the data
     # this method handles only exchange responses

pyinjective/core/network.py

@@ -172,6 +172,7 @@ def __init__(
         grpc_endpoint: str,
         grpc_exchange_endpoint: str,
         grpc_explorer_endpoint: str,
+        chain_stream_endpoint: str,
         chain_id: str,
         fee_denom: str,
         env: str,
@@ -183,6 +184,7 @@ def __init__(
         self.grpc_endpoint = grpc_endpoint
         self.grpc_exchange_endpoint = grpc_exchange_endpoint
         self.grpc_explorer_endpoint = grpc_explorer_endpoint
+        self.chain_stream_endpoint = chain_stream_endpoint
         self.chain_id = chain_id
         self.fee_denom = fee_denom
         self.env = env
@@ -197,6 +199,7 @@ def devnet(cls):
             grpc_endpoint="devnet.injective.dev:9900",
             grpc_exchange_endpoint="devnet.injective.dev:9910",
             grpc_explorer_endpoint="devnet.injective.dev:9911",
+            chain_stream_endpoint="devnet.injective.dev:9999",
             chain_id="injective-777",
             fee_denom="inj",
             env="devnet",
@@ -218,6 +221,7 @@ def testnet(cls, node="lb"):
             grpc_endpoint = "testnet.sentry.chain.grpc.injective.network:443"
             grpc_exchange_endpoint = "testnet.sentry.exchange.grpc.injective.network:443"
             grpc_explorer_endpoint = "testnet.sentry.explorer.grpc.injective.network:443"
+            chain_stream_endpoint = "testnet.sentry.chain.stream.injective.network:443"
             cookie_assistant = BareMetalLoadBalancedCookieAssistant()
             use_secure_connection = True
         else:
@@ -226,6 +230,7 @@ def testnet(cls, node="lb"):
             grpc_endpoint = "testnet.chain.grpc.injective.network"
             grpc_exchange_endpoint = "testnet.exchange.grpc.injective.network"
             grpc_explorer_endpoint = "testnet.explorer.grpc.injective.network"
+            chain_stream_endpoint = "testnet.chain.stream.injective.network"
             cookie_assistant = DisabledCookieAssistant()
             use_secure_connection = True
 
@@ -235,6 +240,7 @@ def testnet(cls, node="lb"):
             grpc_endpoint=grpc_endpoint,
             grpc_exchange_endpoint=grpc_exchange_endpoint,
             grpc_explorer_endpoint=grpc_explorer_endpoint,
+            chain_stream_endpoint=chain_stream_endpoint,
             chain_id="injective-888",
             fee_denom="inj",
             env="testnet",
@@ -260,6 +266,7 @@ def mainnet(cls, node="lb"):
             grpc_endpoint = "sentry.chain.grpc.injective.network:443"
             grpc_exchange_endpoint = "sentry.exchange.grpc.injective.network:443"
             grpc_explorer_endpoint = "sentry.explorer.grpc.injective.network:443"
+            chain_stream_endpoint = "sentry.chain.stream.injective.network:443"
             cookie_assistant = BareMetalLoadBalancedCookieAssistant()
             use_secure_connection = True
         elif node == "lb_k8s":
@@ -268,6 +275,7 @@ def mainnet(cls, node="lb"):
             grpc_endpoint = "k8s.global.mainnet.chain.grpc.injective.network:443"
             grpc_exchange_endpoint = "k8s.global.mainnet.exchange.grpc.injective.network:443"
             grpc_explorer_endpoint = "k8s.global.mainnet.explorer.grpc.injective.network:443"
+            chain_stream_endpoint = "k8s.global.mainnet.chain.stream.injective.network:443"
             cookie_assistant = KubernetesLoadBalancedCookieAssistant()
             use_secure_connection = True
         else:
@@ -276,6 +284,7 @@ def mainnet(cls, node="lb"):
             grpc_endpoint = f"{node}.injective.network:9900"
             grpc_exchange_endpoint = f"{node}.injective.network:9910"
             grpc_explorer_endpoint = f"{node}.injective.network:9911"
+            chain_stream_endpoint = f"{node}.injective.network:9999"
             cookie_assistant = DisabledCookieAssistant()
             use_secure_connection = False
 
@@ -285,6 +294,7 @@ def mainnet(cls, node="lb"):
             grpc_endpoint=grpc_endpoint,
             grpc_exchange_endpoint=grpc_exchange_endpoint,
             grpc_explorer_endpoint=grpc_explorer_endpoint,
+            chain_stream_endpoint=chain_stream_endpoint,
             chain_id="injective-1",
             fee_denom="inj",
             env="mainnet",
@@ -300,6 +310,7 @@ def local(cls):
             grpc_endpoint="localhost:9900",
             grpc_exchange_endpoint="localhost:9910",
             grpc_explorer_endpoint="localhost:9911",
+            chain_stream_endpoint="localhost:9999",
             chain_id="injective-1",
             fee_denom="inj",
             env="local",
@@ -315,6 +326,7 @@ def custom(
             grpc_endpoint,
             grpc_exchange_endpoint,
             grpc_explorer_endpoint,
+            chain_stream_endpoint,
             chain_id,
             env,
             cookie_assistant: Optional[CookieAssistant] = None,
@@ -327,6 +339,7 @@ def custom(
             grpc_endpoint=grpc_endpoint,
             grpc_exchange_endpoint=grpc_exchange_endpoint,
             grpc_explorer_endpoint=grpc_explorer_endpoint,
+            chain_stream_endpoint=chain_stream_endpoint,
             chain_id=chain_id,
             fee_denom="inj",
             env=env,