Add new withdraw_rewards method

andelf · Feb 2, 2023 · 7e765a5 · 7e765a5
1 parent 9bc7442
commit 7e765a5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/tronpy/async_tron.py b/tronpy/async_tron.py
@@ -457,6 +457,11 @@ def vote_witness(self, owner: TAddress, *votes: Tuple[TAddress, int]) -> "AsyncT
         payload = {"owner_address": keys.to_hex_address(owner), "votes": votes}
         return self._build_transaction("VoteWitnessContract", payload)
 
+    def withdraw_rewards(self, owner: TAddress) -> "AsyncTransactionBuilder":
+        """Withdraw voting rewards."""
+        payload = {"owner_address": keys.to_hex_address(owner)}
+        return self._build_transaction("WithdrawBalanceContract", payload)
+
     # Contract
 
     def deploy_contract(self, owner: TAddress, contract: AsyncContract) -> "AsyncTransactionBuilder":

diff --git a/tronpy/tron.py b/tronpy/tron.py
@@ -442,6 +442,11 @@ def vote_witness(self, owner: TAddress, *votes: Tuple[TAddress, int]) -> "Transa
         payload = {"owner_address": keys.to_hex_address(owner), "votes": votes}
         return self._build_transaction("VoteWitnessContract", payload)
 
+    def withdraw_rewards(self, owner: TAddress) -> "TransactionBuilder":
+        """Withdraw voting rewards."""
+        payload = {"owner_address": keys.to_hex_address(owner)}
+        return self._build_transaction("WithdrawBalanceContract", payload)
+
     # Contract
 
     def deploy_contract(self, owner: TAddress, contract: Contract) -> "TransactionBuilder":