diff --git a/core.py b/core.py index 6dd8bf4..1b14c7b 100755 --- a/core.py +++ b/core.py @@ -418,16 +418,6 @@ def get_token_balance(token_address, wallet_address, decimals=False): return float(round(from_token_decimals(token_balance, token_info['decimals']), 15)) -def get_token_supply(token_address, decimals=False): - token_contract = load_contract(token_address) - token_info = get_token_info(token_address) - token_supply = token_contract.functions.totalSupply().call() - if decimals: - return token_supply - else: - return float(round(from_token_decimals(token_supply, token_info['decimals']), 15)) - - def get_token_info(token_address, attempts=18): os.makedirs(token_folder := "./data/tokens".format(token_address), exist_ok=True) token_info_file = "{}/{}.json".format(token_folder, token_address) @@ -469,6 +459,16 @@ def get_token_info(token_address, attempts=18): return token_info +def get_token_supply(token_address, decimals=False): + token_contract = load_contract(token_address) + token_info = get_token_info(token_address) + token_supply = token_contract.functions.totalSupply().call() + if decimals: + return token_supply + else: + return float(round(from_token_decimals(token_supply, token_info['decimals']), 15)) + + def interpret_exception_message(e): logging.debug(e) if 'insufficient funds for gas * price + value' in str(e):