add get_token_supply

This commit is contained in:
Alexander 2024-07-11 11:15:50 -04:00
parent 0913ad0d15
commit 17fa53256b

10
core.py
View file

@ -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):