update wait_for_tansaction api to support long poll

Author: fishronsage

aptos_sdk/async_client.py

@@ -568,25 +568,43 @@ async def transaction_pending(self, txn_hash: str) -> bool:
             raise ApiError(response.text, response.status_code)
         return response.json()["type"] == "pending_transaction"
 
-    async def wait_for_transaction(self, txn_hash: str) -> None:
+    async def _wait_for_txn(self, txn_hash: str) -> None:
         """
         Waits up to the duration specified in client_config for a transaction to move past pending
         state.
         """
-
-        count = 0
-        while await self.transaction_pending(txn_hash):
-            assert (
-                count < self.client_config.transaction_wait_in_seconds
-            ), f"transaction {txn_hash} timed out"
+        response = await self._get(endpoint=f"transactions/by_hash/{txn_hash}")
+        if response.status_code != 404 and response.status_code >= 400:
+            raise ApiError(response.text, response.status_code)
+        is_pending = response.json()["type"] == "pending_transaction"
+
+        # If the transaction is pending, we do a long wait once to avoid polling
+        if is_pending:
+            response = await self._get(endpoint=f"transactions/wait_by_hash/{txn_hash}")
+            if response.status_code != 404 and response.status_code >= 400:
+                raise ApiError(response.text, response.status_code)
+            is_pending = response.json()["type"] == "pending_transaction"
+
+        # Now we do polling to see if the transaction is still pending
+        while is_pending:
+            response = await self._get(endpoint=f"transactions/by_hash/{txn_hash}")
+            if response.status_code != 404 and response.status_code >= 400:
+                raise ApiError(response.text, response.status_code)
+            is_pending = response.json()["type"] == "pending_transaction"
+            if not is_pending:
+                break
             await asyncio.sleep(1)
-            count += 1
 
-        response = await self._get(endpoint=f"transactions/by_hash/{txn_hash}")
         assert (
             "success" in response.json() and response.json()["success"]
         ), f"{response.text} - {txn_hash}"
 
+    async def wait_for_transaction(self, txn_hash: str) -> None:
+        await asyncio.wait_for(
+            self._wait_for_txn(txn_hash),
+            timeout=self.client_config.transaction_wait_in_seconds,
+        )
+
     async def account_transaction_sequence_number_status(
         self, address: AccountAddress, sequence_number: int
     ) -> bool: