Add withdraw_stake_balance. Fixes #90

andelf · Mar 15, 2024 · d02ab54 · d02ab54
1 parent f8b38df
commit d02ab54
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/tronpy/async_tron.py b/tronpy/async_tron.py
@@ -431,6 +431,16 @@ def freeze_balance(self, owner: TAddress, amount: int, resource: str = "ENERGY")
         }
         return self._build_transaction("FreezeBalanceV2Contract", payload)
 
+    def withdraw_stake_balance(self, owner: TAddress) -> "AsyncTransactionBuilder":
+        """Withdraw all stake v2 balance after waiting for 14 days since unfreeze_balance call.
+
+        :param owner:
+        """
+        payload = {
+            "owner_address": keys.to_hex_address(owner),
+        }
+        return self._build_transaction("WithdrawExpireUnfreezeContract", payload)
+
     def unfreeze_balance(
         self, owner: TAddress, resource: str = "ENERGY", *, unfreeze_balance: int
     ) -> "AsyncTransactionBuilder":

diff --git a/tronpy/tron.py b/tronpy/tron.py
@@ -420,6 +420,16 @@ def freeze_balance(self, owner: TAddress, amount: int, resource: str = "ENERGY")
         }
         return self._build_transaction("FreezeBalanceV2Contract", payload)
 
+    def withdraw_stake_balance(self, owner: TAddress) -> "TransactionBuilder":
+        """Withdraw all stake v2 balance after waiting for 14 days since unfreeze_balance call.
+
+        :param owner:
+        """
+        payload = {
+            "owner_address": keys.to_hex_address(owner),
+        }
+        return self._build_transaction("WithdrawExpireUnfreezeContract", payload)
+
     def unfreeze_balance(self, owner: TAddress, resource: str = "ENERGY", *, unfreeze_balance: int) -> "TransactionBuilder":
         """Unfreeze balance to get TRX back.