implement new block mode

README.md

@@ -124,13 +124,10 @@ async def main():
     )
 
     # Sign and broadcast a transaction
-    tx = await c.send_tx_sync_mode(wallet.sign_and_build(txn))
-    print(tx.to_json(indent=4))
+    tx_block = await c.send_tx_block_mode(wallet.sign_and_build(txn))
 
-    # Wait for transaction is included
-    time.sleep(4)
-    tx_response = await c.get_tx_response(tx.txhash)
-    print(tx_response.to_json(indent=4))
+    # Converting to JSON for readability
+    print(tx_block.to_json(indent=4))
 
 
 if __name__ == "__main__":

examples/ledger_example.py

@@ -45,13 +45,10 @@ async def main():
     )
 
     # Sign and broadcast a transaction
-    tx = await c.send_tx_sync_mode(wallet.sign_and_build(txn))
-    print(tx.to_json(indent=4))
+    tx_block = await c.send_tx_block_mode(wallet.sign_and_build(txn))
 
-    # Wait for transaction is included
-    time.sleep(4)
-    tx_response = await c.get_tx_response(tx.txhash)
-    print(tx_response.to_json(indent=4))
+    # Converting to JSON for readability
+    print(tx_block.to_json(indent=4))
 
 
 if __name__ == "__main__":

examples/request_data_example.py

@@ -48,13 +48,10 @@ async def main():
     )
 
     # Sign and broadcast a transaction
-    tx = await c.send_tx_sync_mode(wallet.sign_and_build(txn))
-    print(tx.to_json(indent=4))
+    tx_block = await c.send_tx_block_mode(wallet.sign_and_build(txn))
 
-    # Wait for transaction is included
-    time.sleep(4)
-    tx_response = await c.get_tx_response(tx.txhash)
-    print(tx_response.to_json(indent=4))
+    # Converting to JSON for readability
+    print(tx_block.to_json(indent=4))
 
 
 if __name__ == "__main__":

examples/send_band_example.py

@@ -42,13 +42,10 @@ async def main():
     )
 
     # Sign and broadcast a transaction
-    tx = await c.send_tx_sync_mode(wallet.sign_and_build(txn))
-    print(tx.to_json(indent=4))
+    tx_block = await c.send_tx_block_mode(wallet.sign_and_build(txn))
 
-    # Wait for transaction is included
-    time.sleep(4)
-    tx_response = await c.get_tx_response(tx.txhash)
-    print(tx_response.to_json(indent=4))
+    # Converting to JSON for readability
+    print(tx_block.to_json(indent=4))
 
 
 if __name__ == "__main__":

pyband/client.py

@@ -45,7 +45,6 @@ def __del__(self) -> None:
 
     def close(self) -> None:
         """Closes the connection."""
-
         self.__channel.close()
 
     def get_channel(self) -> Channel:
@@ -207,6 +206,35 @@ async def send_tx_async_mode(self, tx_bytes: bytes) -> TxResponse:
         )
         return resp.tx_response
 
+    async def send_tx_block_mode(self, tx_bytes: bytes, timeout: int = 5, pull_interval: int = 1) -> TxResponse:
+        """Sends a transaction in block mode.
+        Sends a transaction and waits until the transaction has been committed to a block before returning the response.
+        Args:
+            tx_bytes: A signed transaction in raw bytes.
+        Returns:
+            The transaction response.
+        """
+
+        resp = await self.stub_tx.broadcast_tx(
+            BroadcastTxRequest(tx_bytes=tx_bytes, mode=BroadcastMode.BROADCAST_MODE_SYNC)
+        )
+        if resp.tx_response.code != 0:
+            return resp.tx_response
+
+        tx_hash = resp.tx_response.txhash
+        while True:
+            timeout -= pull_interval
+            time.sleep(pull_interval)
+
+            try:
+                tx_response = await self.get_tx_response(tx_hash)
+                break
+            except Exception as e:
+                if timeout < 0:
+                    raise e
+
+        return tx_response
+
     async def get_chain_id(self) -> str:
         """Gets the chain ID.