feat: added multi nft token status update endpoint

fireblocks · Oct 10, 2023 · b97b02a · b97b02a
1 parent 8a23faa
commit b97b02a
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/fireblocks_sdk/api_types.py b/fireblocks_sdk/api_types.py
@@ -301,6 +301,10 @@ class NFTOwnershipStatusValues(str, Enum):
     ARCHIVED = "ARCHIVED"
 
 
+class NFTOwnershipStatusUpdatedPayload:
+    assetId: str
+    status: NFTOwnershipStatusValues
+
 class GetOwnedCollectionsSortValue(str, Enum):
     COLLECTION_NAME = "name"
 

diff --git a/fireblocks_sdk/sdk.py b/fireblocks_sdk/sdk.py
@@ -11,7 +11,7 @@
     DestinationTransferPeerPath, TransferTicketTerm, TRANSACTION_TRANSFER, SIGNING_ALGORITHM, UnsignedMessage, \
     FEE_LEVEL, PagedVaultAccountsRequestFilters, TransactionDestination, NFTOwnershipStatusValues, IssueTokenRequest, \
     GetAssetWalletsFilters, TimePeriod, GetOwnedCollectionsSortValue, GetOwnedNftsSortValues, GetNftsSortValues, OrderValues, \
-    GetOwnedAssetsSortValues, PolicyRule
+    GetOwnedAssetsSortValues, PolicyRule, NFTOwnershipStatusUpdatedPayload
 from .sdk_token_provider import SdkTokenProvider
 
 
@@ -248,6 +248,16 @@ def update_nft_ownership_status(self, id: str, status: NFTOwnershipStatusValues)
 
         return self._put_request(url, {"status": status.value})
 
+    def update_nft_ownerships_status(self, payload: List[NFTOwnershipStatusUpdatedPayload]):
+        """Updates tokens status for a tenant, in all tenant vaults.
+
+        Args:
+            payload (string): List of assets with status for update
+        """
+        url = "/v1/nfts/ownership/tokens/status"
+
+        return self._put_request(url, payload)
+
     def get_supported_assets(self):
         """Gets all assets that are currently supported by Fireblocks"""