Add lock_period to delegate_resource. Fixes andelf#112

mbalashovv · Mar 15, 2024 · f8b38df · f8b38df
1 parent 4893fa3
commit f8b38df
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/tronpy/async_tron.py b/tronpy/async_tron.py
@@ -461,7 +461,13 @@ def unfreeze_balance_legacy(
         return self._build_transaction("UnfreezeBalanceContract", payload)
 
     def delegate_resource(
-        self, owner: TAddress, receiver: TAddress, balance: int, resource: str = "BANDWIDTH", lock: bool = False
+        self,
+        owner: TAddress,
+        receiver: TAddress,
+        balance: int,
+        resource: str = "BANDWIDTH",
+        lock: bool = False,
+        lock_period: int = None,
     ) -> "AsyncTransactionBuilder":
         """Delegate bandwidth or energy resources to other accounts in Stake2.0.
 
@@ -470,6 +476,7 @@ def delegate_resource(
         :param balance:
         :param resource: Resource type, can be ``"ENERGY"`` or ``"BANDWIDTH"``
         :param lock: Optionally lock delegated resources for 3 days.
+        :param lock_period: Optionally lock delegated resources for a specific period. Default: 3 days.
         """
 
         payload = {
@@ -479,6 +486,8 @@ def delegate_resource(
             "resource": resource,
             "lock": lock,
         }
+        if lock_period is not None:
+            payload["lock_period"] = lock_period
 
         return self._build_transaction("DelegateResourceContract", payload)
 

diff --git a/tronpy/tron.py b/tronpy/tron.py
@@ -452,7 +452,13 @@ def unfreeze_balance_legacy(
         return self._build_transaction("UnfreezeBalanceContract", payload)
 
     def delegate_resource(
-        self, owner: TAddress, receiver: TAddress, balance: int, resource: str = "BANDWIDTH", lock: bool = False
+        self,
+        owner: TAddress,
+        receiver: TAddress,
+        balance: int,
+        resource: str = "BANDWIDTH",
+        lock: bool = False,
+        lock_period: int = None,
     ) -> "TransactionBuilder":
         """Delegate bandwidth or energy resources to other accounts in Stake2.0.
 
@@ -461,6 +467,7 @@ def delegate_resource(
         :param balance:
         :param resource: Resource type, can be ``"ENERGY"`` or ``"BANDWIDTH"``
         :param lock: Optionally lock delegated resources for 3 days.
+        :param lock_period: Optionally lock delegated resources for a specific period. Default: 3 days.
         """
 
         payload = {
@@ -470,6 +477,8 @@ def delegate_resource(
             "resource": resource,
             "lock": lock,
         }
+        if lock_period is not None:
+            payload["lock_period"] = lock_period
 
         return self._build_transaction("DelegateResourceContract", payload)