diff --git a/core.py b/core.py index 296e974..1ee33d8 100755 --- a/core.py +++ b/core.py @@ -416,6 +416,16 @@ def get_token_balance(token_address, wallet_address, decimals=False): return round(token_balance, 15) else: 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 round(token_supply, 15) + else: + return float(round(from_token_decimals(token_supply, token_info['decimals']), 15)) def get_token_info(token_address, attempts=18):