|
29 | 29 | query_pb2 as authz_query, |
30 | 30 | authz_pb2 as authz_type, |
31 | 31 | ) |
| 32 | +from .proto.cosmos.bank.v1beta1 import ( |
| 33 | + query_pb2_grpc as bank_query_grpc, |
| 34 | + query_pb2 as bank_query, |
| 35 | +) |
32 | 36 | from .proto.cosmos.tx.v1beta1 import ( |
33 | 37 | service_pb2_grpc as tx_service_grpc, |
34 | 38 | service_pb2 as tx_service, |
@@ -92,6 +96,7 @@ def __init__( |
92 | 96 | self.stubCosmosTendermint = tendermint_query_grpc.ServiceStub(self.chain_channel) |
93 | 97 | self.stubAuth = auth_query_grpc.QueryStub(self.chain_channel) |
94 | 98 | self.stubAuthz = authz_query_grpc.QueryStub(self.chain_channel) |
| 99 | + self.stubBank = bank_query_grpc.QueryStub(self.chain_channel) |
95 | 100 | self.stubTx = tx_service_grpc.ServiceStub(self.chain_channel) |
96 | 101 |
|
97 | 102 | # attempt to load from disk |
@@ -304,6 +309,21 @@ async def get_grants(self, granter: str, grantee: str, **kwargs): |
304 | 309 | ) |
305 | 310 | ) |
306 | 311 |
|
| 312 | + async def get_bank_balances(self, address: str): |
| 313 | + return await self.stubBank.AllBalances( |
| 314 | + bank_query.QueryAllBalancesRequest( |
| 315 | + address=address |
| 316 | + ) |
| 317 | + ) |
| 318 | + |
| 319 | + async def get_bank_balance(self, address: str, denom: str): |
| 320 | + return await self.stubBank.Balance( |
| 321 | + bank_query.QueryBalanceRequest( |
| 322 | + address=address, |
| 323 | + denom=denom |
| 324 | + ) |
| 325 | + ) |
| 326 | + |
307 | 327 | # Injective Exchange client methods |
308 | 328 |
|
309 | 329 | # Auction RPC |
@@ -387,6 +407,37 @@ async def stream_blocks(self): |
387 | 407 | req = explorer_rpc_pb.StreamBlocksRequest() |
388 | 408 | return self.stubExplorer.StreamBlocks(req) |
389 | 409 |
|
| 410 | + async def get_peggy_deposits(self, **kwargs): |
| 411 | + req = explorer_rpc_pb.GetPeggyDepositTxsRequest( |
| 412 | + sender=kwargs.get("sender"), |
| 413 | + receiver=kwargs.get("receiver"), |
| 414 | + limit=kwargs.get("limit"), |
| 415 | + skip=kwargs.get("skip") |
| 416 | + ) |
| 417 | + return await self.stubExplorer.GetPeggyDepositTxs(req) |
| 418 | + |
| 419 | + async def get_peggy_withdrawals(self, **kwargs): |
| 420 | + req = explorer_rpc_pb.GetPeggyWithdrawalTxsRequest( |
| 421 | + sender=kwargs.get("sender"), |
| 422 | + receiver=kwargs.get("receiver"), |
| 423 | + limit=kwargs.get("limit"), |
| 424 | + skip=kwargs.get("skip") |
| 425 | + ) |
| 426 | + return await self.stubExplorer.GetPeggyWithdrawalTxs(req) |
| 427 | + |
| 428 | + async def get_ibc_transfers(self, **kwargs): |
| 429 | + req = explorer_rpc_pb.GetIBCTransferTxsRequest( |
| 430 | + sender=kwargs.get("sender"), |
| 431 | + receiver=kwargs.get("receiver"), |
| 432 | + src_channel=kwargs.get("src_channel"), |
| 433 | + src_port=kwargs.get("src_port"), |
| 434 | + dest_channel=kwargs.get("dest_channel"), |
| 435 | + dest_port=kwargs.get("dest_port"), |
| 436 | + limit=kwargs.get("limit"), |
| 437 | + skip=kwargs.get("skip") |
| 438 | + ) |
| 439 | + return await self.stubExplorer.GetIBCTransferTxs(req) |
| 440 | + |
390 | 441 | # AccountsRPC |
391 | 442 |
|
392 | 443 | async def stream_subaccount_balance(self, subaccount_id: str): |
|
0 commit comments