Skip to content

Commit

Permalink
Merge pull request #16 from fireblocks/vault_assets_balance
Browse files Browse the repository at this point in the history
 Add get_vault_balance_by_asset & Update query parameters
  • Loading branch information
tomervil authored Mar 1, 2021
2 parents e91c728 + bb2db30 commit 02bf76c
Showing 1 changed file with 20 additions and 6 deletions.
26 changes: 20 additions & 6 deletions fireblocks_sdk/sdk.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,11 +28,12 @@ def get_supported_assets(self):

def get_vault_accounts(self, name_prefix=None, name_suffix=None):
"""Gets all vault accounts for your tenant
Args:
name_prefix (string, optional): Vault name prefix
name_suffix (string, optional): Vault name suffix
Args:
name_prefix (string, optional): Vault account name prefix
name_suffix (string, optional): Vault account name suffix
"""

url = f"/v1/vault/accounts"

params = {}
Expand Down Expand Up @@ -859,8 +860,8 @@ def get_vault_assets_balance(self, account_name_prefix=None, account_name_suffix
"""Gets vault assets accumulated balance
Args:
name_prefix (string, optional): Vault name prefix
name_suffix (string, optional): Vault name suffix
account_name_prefix (string, optional): Vault account name prefix
account_name_suffix (string, optional): Vault account name suffix
"""
url = f"/v1/vault/assets"

Expand All @@ -876,6 +877,19 @@ def get_vault_assets_balance(self, account_name_prefix=None, account_name_suffix
url = url + "?" + urllib.parse.urlencode(params)

return self._get_request(url)

def get_vault_balance_by_asset(self, asset_id=None):
"""Gets vault accumulated balance by asset
Args:
asset_id (str, optional): The asset symbol (e.g BTC, ETH)
"""
url = f"/v1/vault/assets"

if asset_id:
url += f"/{asset_id}"

return self._get_request(url)

def create_raw_transaction(self, raw_message, source=None, asset_id=None, note=None):
"""Creates a new raw transaction with the specified parameters
Expand Down

0 comments on commit 02bf76c

Please sign in to comment.