from decimal import Decimal
from typing import Optional
from a_sync import a_sync
from multicall.call import Call
from y import ENVIRONMENT_VARIABLES as ENVS
from y.classes.common import ERC20, WeiBalance
from y.contracts import Contract, has_methods
from y.datatypes import Address, Block
from y.utils.cache import optional_async_diskcache
METHODS = "main()(address)", "issuanceAvailable()(uint)", "redemptionAvailable()(uint)"
[docs]
@a_sync(default="sync")
@optional_async_diskcache
async def is_rtoken(token_address: Address) -> bool:
"""
Check if the given token is a Reserve Protocol R-token.
Args:
token_address: The address of the token to check.
Returns:
True if the token is a Reserve Protocol R-token, False otherwise.
Example:
>>> address = "0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48" # USDC (not an R-token)
>>> is_r = is_rtoken(address)
>>> print(is_r)
False
"""
return await has_methods(token_address, METHODS, sync=False)
[docs]
@a_sync(default="sync")
async def get_price(
token_address: Address,
block: Optional[Block] = None,
skip_cache: bool = ENVS.SKIP_CACHE,
) -> Decimal:
"""
Get the price of a Reserve Protocol R-token in USD.
Args:
token_address: The address of the R-token.
block (optional): The block number to query. Defaults to None (latest).
skip_cache (optional): Whether to skip cache. Defaults to :obj:`ENVS.SKIP_CACHE`.
Returns:
The price of the R-token in USD.
Raises:
TypeError: If the token is not a valid R-token.
Exception: If unable to calculate the price.
Example:
>>> address = "0xaCeeD87BD5754c3d714F3Bd43a9B7B0C9250ab0D" # RSV token
>>> price = await get_price(address, sync=False)
>>> print(price)
1.00
"""
main = await Call(token_address, "main()(address)", block_id=block)
if main is None:
raise TypeError(main, token_address, await is_rtoken(token_address))
basket_handler = await Contract.coroutine(
await Call(main, "basketHandler()(address)", block_id=block)
)
low, high = await basket_handler.price.coroutine(block_identifier=block)
return Decimal(low + high) // 2 / 10**18
tokens, *_ = await basket_handler.getPrimeBasket.coroutine(block_identifier=block)
tokens = [ERC20(token, asynchronous=True) for token in tokens]
balances = [
WeiBalance(
balance, token, block=block, skip_cache=skip_cache, asynchronous=True
)
for token, balance in zip(
tokens, await basket_handler.quantity.map(tokens, block_identifier=block)
)
]
print(balances)
values = WeiBalance.value_usd.map(balances).values()
print(values)
value = sum(values)
print(value)
supply = await ERC20(token_address, asynchronous=True).total_supply_readable(block)
print(f"ts: {supply}")
raise Exception(value / supply)