Fix add_delegate method in Tx Service API

safe-global · Apr 5, 2024 · fc1e148 · fc1e148
1 parent 4f21f8a
commit fc1e148
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/gnosis/safe/api/transaction_service_api/transaction_service_api.py b/gnosis/safe/api/transaction_service_api/transaction_service_api.py
@@ -215,6 +215,7 @@ def add_delegate(
         self,
         safe_address: ChecksumAddress,
         delegate_address: ChecksumAddress,
+        delegator_address: ChecksumAddress,
         label: str,
         signer_account: LocalAccount,
     ) -> bool:
@@ -223,12 +224,11 @@ def add_delegate(
         add_payload = {
             "safe": safe_address,
             "delegate": delegate_address,
+            "delegator": delegator_address,
             "signature": signature.signature.hex(),
             "label": label,
         }
-        response = self._post_request(
-            f"/api/v1/safes/{safe_address}/delegates/", add_payload
-        )
+        response = self._post_request("/api/v1/delegates/", add_payload)
         if not response.ok:
             raise SafeAPIException(f"Cannot add delegate: {response.content}")
         return True

diff --git a/gnosis/safe/api/transaction_service_api/transaction_service_messages.py b/gnosis/safe/api/transaction_service_api/transaction_service_messages.py
@@ -12,11 +12,10 @@ def get_totp() -> int:
     return int(time.time()) // 3600
 
 
-def get_delegate_message(cls, delegate_address: ChecksumAddress) -> str:
+def get_delegate_message(delegate_address: ChecksumAddress) -> str:
     """
     Retrieves the required message for creating or removing a delegate on Safe Transaction Service.
 
-    :param cls:
     :param delegate_address:
     :return: generated str message
     """