apr_base.py

from gcc_utils import gccPrint
from utils.node import (
    w3,
    init_chef,
    init_chefv2,
)
from utils.prices import (
    getDexTokenUSDRatio,
)
from utils.reserves import (
    getDataV1Pools,
    getDataV2Pools,
)
from utils.constants import (
    V1_POOLS,
    V2_POOLS,
    V2_STABLEPOOL_METADATA,
    WETH_ADDRESS,
    WNEAR_ADDRESS,
    WNEAR_USDC,
    WETH_USDC,
    TRI_ADDRESS,
    WNEAR_TRI,
)
from utils.rewarder_configs import formatRewarderConfigItem, getRewarderConfigs


def apr_base():
    gccPrint("Starting APR BASE")
    data = []

    ## chef data
    tri_decimals = 18
    chef = init_chef()
    totalAllocPoint = chef.functions.totalAllocPoint().call()
    triPerBlock = chef.functions.triPerBlock().call()

    ## chefv2 calls
    dummyLPPoolId = 7
    dummyLPToken = "0x9990a658F71248cc507Ea62946f0EB7728491B70"
    dummyLpPoolInfo = chef.functions.poolInfo(dummyLPPoolId).call()
    assert dummyLpPoolInfo[0].lower() == dummyLPToken.lower()
    dummyLpAllocPoint = dummyLpPoolInfo[1]
    dummyLpTotalSecondRewardRate = (
        triPerBlock * dummyLpAllocPoint / (totalAllocPoint * 10**tri_decimals)
    )

    # Chef V2 calls
    chefv2 = init_chefv2()
    totalAllocPointV2 = chefv2.functions.totalAllocPoint().call()

    ### Getting initial prices
    wnearUsdRatio = getDexTokenUSDRatio(w3, WNEAR_USDC, WNEAR_ADDRESS)
    wethUsdRatio = getDexTokenUSDRatio(w3, WETH_USDC, WETH_ADDRESS)
    triUsdRatio = getDexTokenUSDRatio(w3, WNEAR_TRI, TRI_ADDRESS, wnearUsdRatio)

    for id, address in V1_POOLS.items():
        gccPrint(f"V1 Reached here {address}")
        # Chef V1
        data.append(
            getDataV1Pools(
                w3,
                id,
                address,
                chef,
                triPerBlock,
                totalAllocPoint,
                tri_decimals,
                triUsdRatio,
                wnearUsdRatio,
                wethUsdRatio,
            )
        )

    for id, pool in V2_POOLS.items():
        gccPrint(f"V2 Reached here {id}: {pool['LP']}")
        data.append(
            getDataV2Pools(
                w3,
                id,
                pool,
                chefv2,
                dummyLpTotalSecondRewardRate,
                totalAllocPointV2,
                triUsdRatio,
                wnearUsdRatio,
                wethUsdRatio,
            )
        )

    fetched_rewarder_configs = getRewarderConfigs()
    for rewarder_config in fetched_rewarder_configs:
        id, pool = formatRewarderConfigItem(rewarder_config)
        gccPrint(f"Fetched V2 Reached here {id}: {pool['LP']}")

        if pool["LPType"] == "StableAMM" and (id in V2_STABLEPOOL_METADATA) is False:
            gccPrint(f"Skipping pool ID {id}: ID Not Found in V2_STABLEPOOL_METADATA")
            continue

        data.append(
            getDataV2Pools(
                w3,
                id,
                pool,
                chefv2,
                dummyLpTotalSecondRewardRate,
                totalAllocPointV2,
                triUsdRatio,
                wnearUsdRatio,
                wethUsdRatio,
            )
        )

    # mock_data = [
    #     {
    #         "LPToken": "0x22E38A04c14624e6deb66762Dc98E793db2C80Cb",
    #         "isComplexNRewarder": "true",
    #         "RewardTokens": [
    #             "0x8BEc47865aDe3B172A928df8f990Bc7f2A3b9f79",
    #             "0xc21Ff01229e982d7c8b8691163B0A3Cb8F357453",
    #         ],
    #         "Rewarder": "0xB247fA13a9BE90123C4B94982A2b30c60E6182ab",
    #         "RewarderPriceLPs": [AURORA_WNEAR, META_WNEAR],
    #         "PoolId": 43,
    #     }
    # ]

    # for rewarder_config in mock_data:
    #     id, pool = formatRewarderConfigItem(rewarder_config)
    #     gccPrint(f"Mock V2 Reached here {id}: {pool['LP']}")

    #     data.append(
    #         getDataV2Pools(
    #             w3,
    #             id,
    #             pool,
    #             chefv2,
    #             dummyLpTotalSecondRewardRate,
    #             totalAllocPointV2,
    #             triUsdRatio,
    #             wnearUsdRatio,
    #             wethUsdRatio,
    #         )
    #     )

    # print(data)

    gccPrint("Ending APR BASE")

    return data


if __name__ == "__main__":
    apr_base()