def get_balance(configuration, token_symbol=None):
"""
Get balance from account address.
:param configuration: loaded configuration file instance
:param token_symbol: None for ETH, ERC20 symbol for other tokens
:return:
"""
wallet_address = Wallet(configuration).get_address()
if token_symbol is None:
balance = Wallet(configuration).get_balance(wallet_address)
else:
contract_address = configuration.contracts[token_symbol]
contract = Contract(configuration, contract_address)
balance = contract.get_balance(wallet_address)
return balance, wallet_address
def get_balance(configuration, token_symbol=None):
"""
Get balance from account address.
:param configuration: loaded configuration file instance
:param token_symbol: None for ETH, ERC20 symbol for other tokens
:return:
"""
wallet_address = Wallet(configuration).get_address()
if token_symbol is None:
balance = Wallet(configuration).get_balance(wallet_address)
else:
try: # check if token is added to the wallet
contract_address = configuration.contracts[token_symbol]
except KeyError:
raise ERC20NotExistsException()
contract = Contract(configuration, contract_address)
balance = contract.get_balance(wallet_address)
return balance, wallet_address