defi-state-querier 0.5.26__py3-none-any.whl → 0.5.28__py3-none-any.whl

defi_services/services/lending/moonwell_service.py

@@ -1,120 +0,0 @@
-import logging
-
-from defi_services.abis.lending.moonwell.moonwell_comptroller_abi import MOONWELL_COMPTROLLER_ABI
-from defi_services.abis.lending.moonwell.moonwell_ctoken_abi import MOONWELL_CTOKEN_ABI
-from defi_services.abis.token.erc20_abi import ERC20_ABI
-from defi_services.constants.chain_constant import Chain
-from defi_services.constants.entities.lending_constant import Lending
-from defi_services.constants.token_constant import Token
-from defi_services.jobs.queriers.state_querier import StateQuerier
-from defi_services.services.lending.lending_info.base.moonwell_base import MOONWELL_BASE
-from defi_services.services.lending.venus_services import VenusStateService
-
-logger = logging.getLogger("MoonWell Lending Pool State Service")
-
-
-class MoonWellInfo:
-    mapping = {
-        Chain.base: MOONWELL_BASE
-    }
-
-
-class MoonWellStateService(VenusStateService):
-    def __init__(self, state_service: StateQuerier, chain_id: str = "0x38"):
-        super().__init__(state_service, chain_id)
-        self.name = f"{chain_id}_{Lending.moonwell}"
-        self.chain_id = chain_id
-        self.pool_info = MoonWellInfo.mapping.get(chain_id)
-        self.state_service = state_service
-        self.comptroller_abi = MOONWELL_COMPTROLLER_ABI
-        self.vtoken_abi = MOONWELL_CTOKEN_ABI
-
-    def get_service_info(self):
-        info = {
-            Lending.moonwell: {
-                "chain_id": self.chain_id,
-                "type": "lending",
-                "protocol_info": self.pool_info
-            }
-        }
-        return info
-
-    def get_apy_lending_pool_function_info(
-            self,
-            reserves_info: dict,
-            block_number: int = "latest"
-    ):
-        rpc_calls = {}
-        for token_address, reserve_info in reserves_info.items():
-            if token_address != Token.native_token:
-                query_id = f"decimals_{token_address}_{block_number}".lower()
-                rpc_calls[query_id] = self.state_service.get_function_info(token_address, ERC20_ABI, "decimals", [], block_number)
-
-            ctoken = reserve_info.get("cToken")
-            for fn_name in ['decimals', 'totalSupply', 'totalBorrows', 'supplyRatePerTimestamp', 'borrowRatePerTimestamp', 'exchangeRateStored']:
-                query_id = f"{fn_name}_{ctoken}_{block_number}".lower()
-                rpc_calls[query_id] = self.get_ctoken_function_info(
-                    ctoken=ctoken,
-                    fn_name=fn_name,
-                    block_number=block_number
-                )
-
-        return rpc_calls
-
-    def get_reserve_tokens_metadata(
-            self,
-            decoded_data: dict,
-            reserves_info: dict,
-            block_number: int = "latest"
-    ):
-        reserve_tokens_info = []
-        for token_address, reserve_info in reserves_info.items():
-            if token_address != Token.native_token:
-                underlying_decimals_query_id = f"decimals_{token_address}_{block_number}".lower()
-                underlying_decimals = decoded_data.get(underlying_decimals_query_id)
-            else:
-                underlying_decimals = Chain.native_decimals.get(self.chain_id, 18)
-
-            ctoken = reserve_info.get("cToken")
-            ctoken_decimals_query_id = f"decimals_{ctoken}_{block_number}".lower()
-            total_supply_query_id = f"totalSupply_{ctoken}_{block_number}".lower()
-            total_borrow_query_id = f"totalBorrows_{ctoken}_{block_number}".lower()
-            supply_rate_query_id = f"supplyRatePerTimestamp_{ctoken}_{block_number}".lower()
-            borrow_rate_query_id = f"borrowRatePerTimestamp_{ctoken}_{block_number}".lower()
-            exchange_rate_query_id = f"exchangeRateStored_{ctoken}_{block_number}".lower()
-
-            reserve_tokens_info.append({
-                "token": ctoken,
-                "token_decimals": decoded_data.get(ctoken_decimals_query_id),
-                "borrow_rate": decoded_data.get(borrow_rate_query_id),
-                "supply_rate": decoded_data.get(supply_rate_query_id),
-                "supply": decoded_data.get(total_supply_query_id),
-                "borrow": decoded_data.get(total_borrow_query_id),
-                "exchange_rate": decoded_data.get(exchange_rate_query_id),
-                "underlying_decimals": underlying_decimals,
-                "underlying": token_address
-            })
-        return reserve_tokens_info
-
-    @classmethod
-    def _calculate_interest_rates(
-            cls, token_info: dict, pool_decimals: int, apx_block_speed_in_seconds: float):
-        exchange_rate = float(token_info["exchange_rate"]) / 10 ** (18 - 8 + token_info["underlying_decimals"])
-
-        total_borrow = float(token_info["borrow"]) / 10 ** int(token_info["underlying_decimals"])
-        total_supply = float(token_info["supply"]) * exchange_rate / 10 ** int(token_info["token_decimals"])
-
-        supply_apy = ((token_info["supply_rate"] / 10 ** pool_decimals) * 86400 + 1) ** 365 - 1
-        borrow_apy = ((token_info["borrow_rate"] / 10 ** pool_decimals) * 86400 + 1) ** 365 - 1
-
-        return {
-            'deposit_apy': supply_apy,
-            'borrow_apy': borrow_apy,
-            'total_deposit': total_supply,
-            'total_borrow': total_borrow
-        }
-
-    def get_ctoken_function_info(self, ctoken: str, fn_name: str, fn_paras: list = None, block_number: int = "latest"):
-        return self.state_service.get_function_info(
-            ctoken, self.vtoken_abi, fn_name, fn_paras, block_number
-        )

defi_services/services/lending/sonne_service.py

@@ -1,64 +0,0 @@
-import logging
-
-from defi_services.abis.lending.venus.venus_comptroller_abi import VENUS_COMPTROLLER_ABI
-from defi_services.abis.lending.venus.vtoken_abi import VTOKEN_ABI
-from defi_services.constants.chain_constant import Chain
-from defi_services.constants.entities.lending_constant import Lending
-from defi_services.jobs.queriers.state_querier import StateQuerier
-from defi_services.services.lending.lending_info.base.sonne_base import SONNE_BASE
-from defi_services.services.lending.venus_services import VenusStateService
-
-logger = logging.getLogger("Ionic Lending Pool State Service")
-
-
-class SonneInfo:
-    mapping = {
-        Chain.base: SONNE_BASE
-    }
-
-
-class SonneStateService(VenusStateService):
-    def __init__(self, state_service: StateQuerier, chain_id: str = "0x38"):
-        super().__init__(state_service, chain_id)
-        self.name = f"{chain_id}_{Lending.sonne}"
-        self.chain_id = chain_id
-        self.pool_info = SonneInfo.mapping.get(chain_id)
-        self.state_service = state_service
-        self.comptroller_abi = VENUS_COMPTROLLER_ABI
-        self.vtoken_abi = VTOKEN_ABI
-
-    def get_service_info(self):
-        info = {
-            Lending.sonne: {
-                "chain_id": self.chain_id,
-                "type": "lending",
-                "protocol_info": self.pool_info
-            }
-        }
-        return info
-
-    def calculate_apy_lending_pool_function_call(
-            self,
-            reserves_info: dict,
-            decoded_data: dict,
-            token_prices: dict,
-            pool_token_price: float,
-            pool_decimals: int = 18,
-            block_number: int = "latest",
-    ):
-        reserve_tokens_info = self.get_reserve_tokens_metadata(decoded_data, reserves_info, block_number)
-
-        data = {}
-        for token_info in reserve_tokens_info:
-            underlying_token = token_info['underlying']
-            c_token = token_info['token']
-
-            assets = {
-                underlying_token: self._calculate_interest_rates(
-                    token_info, pool_decimals=pool_decimals,
-                    apx_block_speed_in_seconds=1
-                )
-            }
-            data[c_token] = assets
-
-        return data

defi_services/services/lending/zerolend_services.py

@@ -1,36 +0,0 @@
-import logging
-
-from defi_services.constants.chain_constant import Chain
-from defi_services.constants.entities.lending_constant import Lending
-from defi_services.jobs.queriers.state_querier import StateQuerier
-from defi_services.services.lending.aave_v3_services import AaveV3StateService
-from defi_services.services.lending.lending_info.base.zerolend_base import ZEROLEND_BASE
-from defi_services.services.lending.lending_info.ethereum.zerolend_eth import ZEROLEND_ETH
-from defi_services.services.lending.lending_info.zksync.zerolend_zksync import ZEROLEND_ZKSYNC
-
-logger = logging.getLogger("ZeroLend Lending Pool State Service")
-
-
-class ZeroLendInfo:
-    mapping = {
-        Chain.zksync: ZEROLEND_ZKSYNC,
-        Chain.ethereum: ZEROLEND_ETH,
-        Chain.base: ZEROLEND_BASE
-    }
-
-
-class ZeroLendStateService(AaveV3StateService):
-    def __init__(self, state_service: StateQuerier, chain_id: str = Chain.zksync):
-        super().__init__(state_service, chain_id)
-        self.name = f"{chain_id}_{Lending.zerolend}"
-        self.pool_info = ZeroLendInfo.mapping.get(chain_id)
-
-    def get_service_info(self):
-        info = {
-            Lending.zerolend: {
-                "chain_id": self.chain_id,
-                "type": "lending",
-                "protocol_info": self.pool_info
-            }
-        }
-        return info

defi_services/services/vault/tcv_vault_services.py

@@ -1,108 +0,0 @@
-import logging
-
-from defi_services.abis.vault.tcv_abi import TCV_ABI
-from defi_services.constants.chain_constant import Chain
-from defi_services.constants.entities.vault_constant import Vault
-from defi_services.constants.token_constant import Token
-from defi_services.jobs.queriers.state_querier import StateQuerier
-from defi_services.services.protocol_services import ProtocolServices
-from defi_services.services.vault.vault_info.arbitrum.tcv_arb import TCV_VAULT_ARBITRUM
-
-logger = logging.getLogger("Trava Vault State Service")
-
-
-class TCVVaultInfo:
-    mapping = {
-        Chain.arbitrum: TCV_VAULT_ARBITRUM
-    }
-
-
-class TCVVaultStateService(ProtocolServices):
-    def __init__(self, state_service: StateQuerier, chain_id: str = "0xa4b1"):
-        super().__init__()
-        self.name = f"{chain_id}_{Vault.tcv_vault}"
-        self.chain_id = chain_id
-        self.pool_info = TCVVaultInfo.mapping.get(chain_id)
-        self.vault_abi = TCV_ABI
-        self.state_service = state_service
-
-    # BASIC FUNCTION
-    def get_service_info(self):
-        info = {
-            Vault.tcv_vault: {
-                "chain_id": self.chain_id,
-                "type": "vault",
-                "protocol_info": self.pool_info
-            }
-        }
-        return info
-
-    def get_token_list(self):
-        reward_token = self.pool_info.get('rewardToken')
-
-        tokens = []
-        if isinstance(reward_token, list):
-            tokens += reward_token
-        elif isinstance(reward_token, str):
-            tokens.append(reward_token)
-
-        for token, info in self.pool_info.get("reservesList", {}).items():
-            asset_address = info['tokenIn']
-            if asset_address == Token.native_token:
-                tokens.append(Token.wrapped_token.get(self.chain_id))
-            else:
-                tokens.append(asset_address)
-
-        tokens = list(set(tokens))
-        return tokens
-
-    # WALLET STAKING BALANCE
-    def get_wallet_staking_balance_function_info(
-            self,
-            wallet: str,
-            reserves_info: dict,
-            block_number: int = "latest",
-            return_reward: bool = False
-    ):
-        rpc_calls = {}
-        for token in reserves_info:
-            rpc_calls[f'currentNow_{token}_{wallet}_{block_number}'.lower()] = self.state_service.get_function_info(
-                token, TCV_ABI, "currentNow", [wallet], block_number=block_number)
-            rpc_calls[f'totalLiquidityNFT_{token}_{wallet}_{block_number}'.lower()] = self.state_service.get_function_info(
-                token, TCV_ABI, "totalLiquidityNFT", block_number=block_number)
-
-        return rpc_calls
-
-    def calculate_wallet_staking_balance(
-            self,
-            wallet: str,
-            reserves_info: dict,
-            decoded_data: dict,
-            token_prices: dict,
-            block_number: int = 'latest',
-            return_reward: bool = False
-    ):
-        result = {}
-        for token, information in reserves_info.items():
-            liquidity_user_key = f'currentNow_{token}_{wallet}_{block_number}'.lower()
-            liquidity_of_vault_key = f'totalLiquidityNFT_{token}_{wallet}_{block_number}'.lower()
-            liquidity_user = decoded_data[liquidity_user_key]
-            liquidity_of_vault = decoded_data[liquidity_of_vault_key]
-            result[token] = {
-                "liquidity_user": liquidity_user,
-                "liquidity_of_vault": liquidity_of_vault
-            }
-        return result
-
-    # REWARDS BALANCE
-    def get_rewards_balance_function_info(
-            self,
-            wallet,
-            reserves_info: dict = None,
-            block_number: int = "latest"
-    ):
-        pass
-
-    def calculate_rewards_balance(
-            self, wallet: str, reserves_info: dict, decoded_data: dict, block_number: int = "latest"):
-        pass

defi_services/services/vault/vault_info/arbitrum/tcv_arb.py

@@ -1,58 +0,0 @@
-TCV_VAULT_ARBITRUM = {
-    "name": "TCV Vault",
-    "rewardToken": "0x3cd16d9372fc938ac84444d1f47701bb4faa97ec",
-    "airdropAddress": "0x9094264783f48880c12e703cdd1f11c4dff7f1ac",
-    "reservesList": {
-        "0xe467db55710cf35e2dc0402104e23221f0e12e66": {
-            "vaultName": "eth/usdc",
-            "tokenIn": "0xe467db55710cf35e2dc0402104e23221f0e12e66",
-            "pool": '0xc6962004f452be9203591991d15f6b388e09e8d0',
-            "poolInfo": {
-                "token0": "0x82af49447d8a07e3bd95bd0d56f35241523fbab1",
-                "token1": "0xaf88d065e77c8cc2239327c5edb3a432268e5831",
-                "fee": 500
-            }
-        },
-        "0x9403be93fddedf88a3ed7d11bfb643b13e5cbc27": {
-            "vaultName": "eth/usdt",
-            "tokenIn": "0x9403be93fddedf88a3ed7d11bfb643b13e5cbc27",
-            "pool": '0x641c00a822e8b671738d32a431a4fb6074e5c79d',
-            "poolInfo": {
-                "token0": "0x82af49447d8a07e3bd95bd0d56f35241523fbab1",
-                "token1": "0xfd086bc7cd5c481dcc9c85ebe478a1c0b69fcbb9",
-                "fee": 500
-            }
-
-        },
-        "0xae83866e6b48e29b225a10bc236f2665cc4f081e": {
-            "vaultName": "eth/arb",
-            "tokenIn": "0xae83866e6b48e29b225a10bc236f2665cc4f081e",
-            "pool": '0xc6f780497a95e246eb9449f5e4770916dcd6396a',
-            "poolInfo": {
-                "token0": "0x82af49447d8a07e3bd95bd0d56f35241523fbab1",
-                "token1": "0x912ce59144191c1204e64559fe8253a0e49e6548",
-                "fee": 500
-            }
-        },
-        "0xaae3866f0233aec20580c21a3c10791afd38c8c1": {
-            "vaultName": "eth/usdc(bridgeusdc)",
-            "tokenIn": "0xaae3866f0233aec20580c21a3c10791afd38c8c1",
-            "pool": '0xc31e54c7a869b9fcbecc14363cf510d1c41fa443',
-            "poolInfo": {
-                "token0": "0x82af49447d8a07e3bd95bd0d56f35241523fbab1",
-                "token1": "0xff970a61a04b1ca14834a43f5de4533ebddb5cc8",
-                "fee": 500
-            }
-        },
-        "0xa02d1ecbe8eefaf7d64871ee0ae6404318df9702": {
-            "vaultName": "eth/usdc",
-            "tokenIn": "0xa02d1ecbe8eefaf7d64871ee0ae6404318df9702",
-            "pool": '0xc473e2aee3441bf9240be85eb122abb059a3b57c',
-            "poolInfo": {
-                "token0": "0x82af49447d8a07e3bd95bd0d56f35241523fbab1",
-                "token1": "0xaf88d065e77c8cc2239327c5edb3a432268e5831",
-                "fee": 3000
-            }
-        }
-    }
-}