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

defi_services/services/lending/seamless_services.py

@@ -0,0 +1,313 @@
+import logging
+
+from web3 import Web3
+
+from defi_services.abis.lending.seamless.seamless_incentive import SEAMLESS_INCENTIVE_ABI
+from defi_services.abis.lending.seamless.seamless_lending_pool import SEAMLESS_LENDING_POOL_ABI
+from defi_services.abis.lending.seamless.seamless_oracle import SEAMLESS_ORACLE_ABI
+from defi_services.abis.token.erc20_abi import ERC20_ABI
+from defi_services.constants.chain_constant import Chain
+from defi_services.constants.db_constant import DBConst
+from defi_services.constants.entities.lending_constant import Lending
+from defi_services.constants.network_constants import NATIVE_TOKEN
+from defi_services.constants.time_constant import TimeConstants
+from defi_services.jobs.queriers.state_querier import StateQuerier
+from defi_services.services.lending.aave_v2_services import AaveV2StateService
+from defi_services.services.lending.lending_info.base.seamless_base import SEAMLESS_BASE
+
+logger = logging.getLogger("Seamless Lending Pool State Service")
+
+
+class SeamlessInfo:
+    mapping = {
+        Chain.base: SEAMLESS_BASE
+    }
+
+
+class SeamlessStateService(AaveV2StateService):
+    def __init__(self, state_service: StateQuerier, chain_id: str = Chain.base):
+        super().__init__(state_service, chain_id)
+        self.name = f"{chain_id}_{Lending.seamless}"
+        self.chain_id = chain_id
+        self.pool_info = SeamlessInfo.mapping.get(chain_id)
+        self.lending_abi = SEAMLESS_LENDING_POOL_ABI
+        self.incentive_abi = SEAMLESS_INCENTIVE_ABI
+        self.oracle_abi = SEAMLESS_ORACLE_ABI
+        self.state_service = state_service
+
+    def get_service_info(self):
+        info = {
+            Lending.seamless: {
+                "chain_id": self.chain_id,
+                "type": "lending",
+                "protocol_info": self.pool_info
+            }
+        }
+        return info
+
+    def get_dapp_asset_info(self, block_number: int = 'latest'):
+        _w3 = self.state_service.get_w3()
+        pool_address = self.pool_info.get("address")
+        pool_contract = _w3.eth.contract(address=_w3.to_checksum_address(pool_address), abi=self.lending_abi)
+        reserve_list = pool_contract.functions.getReservesList().call(block_identifier=block_number)
+        reserves_info = {}
+        for token in reserve_list:
+            token = token.lower()
+            reserve_data = pool_contract.functions.getReserveData(
+                _w3.to_checksum_address(token)).call(block_identifier=block_number)
+            reserves_info[token] = {}
+            reserves_info[token]["tToken"] = reserve_data[8].lower()
+            reserves_info[token]["sdToken"] = reserve_data[9].lower()
+            reserves_info[token]["dToken"] = reserve_data[10].lower()
+
+            risk_param = bin(reserve_data[0][0])[2:]
+            reserves_info[token]["loanToValue"] = int(risk_param[-15:], 2) / 10 ** 4
+            reserves_info[token]["liquidationThreshold"] = int(risk_param[-31:-16], 2) / 10 ** 4
+
+        return reserves_info
+
+    # CALCULATE APY LENDING POOL
+    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():
+            get_reserve_data_call_id = f'getReserveData_{self.name}_{token_address}_{block_number}'.lower()
+            reserve_data = decoded_data.get(get_reserve_data_call_id)
+            atoken = reserve_data[8].lower()
+            debt_token = reserve_data[10].lower()
+            decimals_call_id = f"decimals_{token_address}_{block_number}".lower()
+            atoken_total_supply_key = f'totalSupply_{atoken}_{block_number}'.lower()
+            debt_token_total_supply_key = f'totalSupply_{debt_token}_{block_number}'.lower()
+            sdebt_token = reserve_data[9].lower()
+
+
+            data = {
+                'underlying': token_address,
+                'underlying_decimals': decoded_data.get(decimals_call_id),
+                'a_token_supply': decoded_data.get(atoken_total_supply_key),
+                'd_token_supply': decoded_data.get(debt_token_total_supply_key),
+
+                'supply_apy': reserve_data[2],
+                'borrow_apy': reserve_data[4],
+
+            }
+            if sdebt_token != NATIVE_TOKEN:
+                sdebt_token_total_supply_key = f'totalSupply_{sdebt_token}_{block_number}'.lower()
+                data['sd_token_supply'] = decoded_data.get(sdebt_token_total_supply_key)
+                data['stable_borrow_apy'] = reserve_data[5]
+            else:
+                data['sd_token_supply'] = 0
+                data['stable_borrow_apy'] = 0
+
+            reserve_tokens_info.append(data)
+
+        return reserve_tokens_info
+
+    def get_apy_lending_pool_function_info_deprecated(
+            self,
+            reserves_info: dict,
+            block_number: int = "latest",
+    ):
+        rpc_calls = {}
+        reward_tokens = self.pool_info.get("rewardTokensList")
+        for token_address, value in reserves_info.items():
+            reserve_key = f"getReserveData_{self.name}_{token_address}_{block_number}".lower()
+            atoken_total_supply_key = f'totalSupply_{value["tToken"]}_{block_number}'.lower()
+            debt_token_total_supply_key = f'totalSupply_{value["dToken"]}_{block_number}'.lower()
+            decimals_key = f"decimals_{token_address}_{block_number}".lower()
+            for reward_token in reward_tokens:
+                atoken_assets_key = f"getRewardsData_{value['tToken']}_{reward_token}_{block_number}".lower()
+                debt_token_assets_key = f"getRewardsData_{value['dToken']}_{reward_token}_{block_number}".lower()
+                rpc_calls[atoken_assets_key] = self.get_function_incentive_info(
+                    "getRewardsData", [value['tToken'], reward_token], block_number)
+                rpc_calls[debt_token_assets_key] = self.get_function_incentive_info(
+                    "getRewardsData", [value['dToken'], reward_token], block_number)
+                if value['sdToken'] != NATIVE_TOKEN:
+                    sdebt_token_assets_key = f"getRewardsData_{value['sdToken']}_{reward_token}_{block_number}".lower()
+                    rpc_calls[sdebt_token_assets_key] = self.get_function_incentive_info(
+                    "getRewardsData", [value['sdToken'], reward_token], block_number)
+
+            rpc_calls[reserve_key] = self.get_function_lending_pool_info("getReserveData", [token_address])
+            rpc_calls[atoken_total_supply_key] = self.state_service.get_function_info(
+                value["tToken"], ERC20_ABI, "totalSupply", block_number=block_number)
+            rpc_calls[debt_token_total_supply_key] = self.state_service.get_function_info(
+                value["dToken"], ERC20_ABI, "totalSupply", block_number=block_number)
+            rpc_calls[decimals_key] = self.state_service.get_function_info(
+                token_address, ERC20_ABI, "decimals", block_number=block_number)
+
+            if value['sdToken'] != NATIVE_TOKEN:
+                sdebt_token_total_supply_key = f'totalSupply_{value["sdToken"]}_{block_number}'.lower()
+                rpc_calls[sdebt_token_total_supply_key] = self.state_service.get_function_info(
+                    value["sdToken"], ERC20_ABI, "totalSupply", block_number=block_number)
+
+        return rpc_calls
+
+    def get_apy_lending_pool_deprecated(
+            self,
+            atokens: dict,
+            debt_tokens: dict,
+            decimals: dict,
+            reserves_info: dict,
+            asset_data_tokens: dict,
+            total_supply_tokens: dict,
+            interest_rate: dict,
+            token_prices: dict,
+            pool_token_price: float,
+            pool_decimals: int = 18,
+    ):
+        reward_tokens = self.pool_info.get("rewardTokensList")
+        for token_address in reserves_info:
+            atoken = atokens.get(token_address)
+            debt_token = debt_tokens.get(token_address)
+            decimal = decimals.get(token_address)
+            total_supply_t = total_supply_tokens.get(atoken, 0)
+            total_supply_d = total_supply_tokens.get(debt_token, 0)
+            # update deposit, borrow apy
+            total_supply_t = total_supply_t / 10 ** decimal
+            total_supply_d = total_supply_d / 10 ** decimal
+            token_price = token_prices.get(token_address)
+            interest_rate[token_address].update({
+                "utilization": total_supply_d / total_supply_t,
+            })
+            total_supply_t_in_usd = total_supply_t * token_price
+            total_supply_d_in_usd = total_supply_d * token_price
+            if reward_tokens:
+                interest_rate[token_address][DBConst.reward_deposit_apy] = {}
+                interest_rate[token_address][DBConst.reward_borrow_apy] = {}
+                asset_data_t = asset_data_tokens.get(atoken)
+                asset_data_d = asset_data_tokens.get(debt_token)
+                # update deposit, borrow apy
+                for reward_address in reward_tokens:
+                    eps_t = asset_data_t[reward_address][1] / 10 ** 18
+                    eps_d = asset_data_d[reward_address][1] / 10 ** 18
+                    if total_supply_t:
+                        deposit_apr = eps_t * TimeConstants.A_YEAR * pool_token_price / (
+                                total_supply_t * token_price)
+                    else:
+                        deposit_apr = 0
+                    if total_supply_d:
+                        borrow_apr = eps_d * TimeConstants.A_YEAR * pool_token_price / (
+                                total_supply_d * token_price)
+                    else:
+                        borrow_apr = 0
+                    interest_rate[token_address][DBConst.reward_deposit_apy].update({
+                        DBConst.reward_borrow_apy: deposit_apr}
+                    )
+                    interest_rate[token_address][DBConst.reward_borrow_apy].update({
+                        reward_address: borrow_apr}
+                    )
+            # update liquidity
+            liquidity_log = {
+                'total_borrow': {
+                    DBConst.amount: total_supply_d,
+                    DBConst.value_in_usd: total_supply_d_in_usd},
+                'total_deposit': {
+                    DBConst.amount: total_supply_t,
+                    DBConst.value_in_usd: total_supply_t_in_usd}
+            }
+            interest_rate[token_address].update({DBConst.liquidity_change_logs: liquidity_log})
+
+        return interest_rate
+
+    def calculate_apy_lending_pool_function_call_deprecated(
+            self,
+            reserves_info: dict,
+            decoded_data: dict,
+            token_prices: dict,
+            pool_token_price: float,
+            pool_decimals: int = 18,
+            block_number: int = 'latest',
+    ):
+        reserves_data = {}
+        for token in reserves_info:
+            get_reserve_data_call_id = f'getReserveData_{self.name}_{token}_{block_number}'.lower()
+            reserves_data[token.lower()] = decoded_data.get(get_reserve_data_call_id)
+        reward_tokens = self.pool_info.get("rewardTokensList")
+        interest_rate, atokens, debt_tokens, sdebt_tokens, decimals, asset_data_tokens = {}, {}, {}, {}, {}, {}
+        total_supply_tokens = {}
+        for token_address in reserves_info:
+            lower_address = token_address.lower()
+            reserve_data = reserves_data[lower_address]
+            interest_rate[lower_address] = {
+                'deposit_apy': float(reserve_data[3]) / 10 ** 27,
+                'borrow_apy': float(reserve_data[4]) / 10 ** 27,
+                'stable_borrow_apy': float(reserve_data[5]) / 10 ** 27}
+            atoken = reserve_data[7].lower()
+            sdebt_token = reserve_data[8].lower()
+            debt_token = reserve_data[9].lower()
+            decimals_call_id = f"decimals_{token_address}_{block_number}".lower()
+            atoken_total_supply_key = f'totalSupply_{atoken}_{block_number}'.lower()
+            debt_token_total_supply_key = f'totalSupply_{debt_token}_{block_number}'.lower()
+            asset_data_tokens[atoken] = {}
+            asset_data_tokens[debt_token] = {}
+            if sdebt_token != NATIVE_TOKEN:
+                asset_data_tokens[sdebt_token] = {}
+
+            total_supply_tokens[atoken] = {}
+            for reward_token in reward_tokens:
+                atoken_assets_key = f"getRewardsData_{atoken}_{reward_token}_{block_number}".lower()
+                debt_token_assets_key = f"getRewardsData_{debt_token}_{reward_token}_{block_number}".lower()
+                asset_data_tokens[atoken][reward_token] = decoded_data.get(atoken_assets_key)
+                asset_data_tokens[debt_token][reward_token] = decoded_data.get(debt_token_assets_key)
+                total_supply_tokens[atoken][reward_token] = decoded_data.get(atoken_total_supply_key)
+                if sdebt_token != NATIVE_TOKEN:
+                    sdebt_token_assets_key = f"getRewardsData_{sdebt_tokens}_{reward_token}_{block_number}".lower()
+                    asset_data_tokens[sdebt_token][reward_token] = decoded_data.get(sdebt_token_assets_key)
+
+            atokens[lower_address] = atoken
+            debt_tokens[lower_address] = debt_token
+            sdebt_tokens[lower_address] = sdebt_token
+            decimals[lower_address] = decoded_data.get(decimals_call_id)
+            total_supply_tokens[debt_token] = decoded_data.get(debt_token_total_supply_key)
+            if sdebt_token != NATIVE_TOKEN:
+                sdebt_token_total_supply_key = f'totalSupply_{sdebt_token}_{block_number}'.lower()
+                total_supply_tokens[sdebt_token] = decoded_data.get(sdebt_token_total_supply_key)
+
+        asset_price_key = f"getAssetsPrices_{self.name}_{block_number}".lower()
+        if not token_prices and asset_price_key in decoded_data:
+            token_prices = {}
+            prices = decoded_data.get(asset_price_key)
+            for pos in range(len(reserves_info.keys())):
+                token_prices[reserves_info[pos].lower()] = prices[pos] / 10 ** pool_decimals
+
+        data = self.get_apy_lending_pool_deprecated(
+            atokens, debt_tokens, decimals, reserves_info, asset_data_tokens, total_supply_tokens, interest_rate,
+            token_prices, pool_token_price, pool_decimals
+        )
+
+        return data
+
+    # REWARDS BALANCE
+    def get_rewards_balance_function_info(
+            self,
+            wallet,
+            reserves_info: dict = None,
+            block_number: int = "latest"
+    ):
+        rpc_calls = {}
+        reward_tokens = self.pool_info.get("rewardTokensList")
+        for reward_token in reward_tokens:
+            decimals_call_id = f"decimals_{reward_token}_{block_number}".lower()
+            rpc_calls[decimals_call_id] = self.state_service.get_function_info(
+                reward_token, ERC20_ABI, "decimals", block_number=block_number)
+        tokens = []
+        for key, value in reserves_info.items():
+            tokens += [Web3.to_checksum_address(value["tToken"]), Web3.to_checksum_address(value["dToken"])]
+        key = f"getAllUserRewards_{self.name}_{wallet}_{block_number}".lower()
+        rpc_calls[key] = self.get_function_incentive_info("getAllUserRewards", [tokens, Web3.to_checksum_address(wallet)], block_number)
+        return rpc_calls
+
+    def calculate_rewards_balance(
+            self, wallet: str, reserves_info: dict, decoded_data: dict, block_number: int = "latest"):
+        key = f"getAllUserRewards_{self.name}_{wallet}_{block_number}".lower()
+        rewards = decoded_data.get(key)
+        result = dict(zip(*rewards))
+        for key, value in result.items():
+            decimals_call_id = f"decimals_{key}_{block_number}".lower()
+            value /= 10 ** decoded_data.get(decimals_call_id, 18)
+            result[key] = {"amount": value}
+        return result

defi_services/services/lending/sonne_service.py

@@ -0,0 +1,64 @@
+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/trava_services.py

@@ -3,8 +3,8 @@ import time
 
 from web3 import Web3
 
-from defi_services.abis.lending.aave_v2_and_forlks.oracle_abi import ORACLE_ABI
-from defi_services.abis.lending.aave_v2_and_forlks.staked_incentives_abi import STAKED_INCENTIVES_ABI
+from defi_services.abis.lending.aave_v2_and_forks.oracle_abi import ORACLE_ABI
+from defi_services.abis.lending.aave_v2_and_forks.staked_incentives_abi import STAKED_INCENTIVES_ABI
 from defi_services.abis.lending.trava.trava_lending_pool_abi import TRAVA_LENDING_POOL_ABI
 from defi_services.abis.token.erc20_abi import ERC20_ABI
 from defi_services.constants.chain_constant import Chain

defi_services/services/lending/uwu_services.py

@@ -3,9 +3,9 @@ import time
 
 from web3 import Web3
 
-from defi_services.abis.lending.aave_v2_and_forlks.lending_pool_abi import LENDING_POOL_ABI
-from defi_services.abis.lending.aave_v2_and_forlks.oracle_abi import ORACLE_ABI
-from defi_services.abis.lending.aave_v2_and_forlks.uwu_incentives_abi import UWU_INCENTIVES_ABI
+from defi_services.abis.lending.aave_v2_and_forks.lending_pool_abi import LENDING_POOL_ABI
+from defi_services.abis.lending.aave_v2_and_forks.oracle_abi import ORACLE_ABI
+from defi_services.abis.lending.aave_v2_and_forks.uwu_incentives_abi import UWU_INCENTIVES_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

defi_services/services/lending/valas_services.py

@@ -3,8 +3,8 @@ import time
 
 from web3 import Web3
 
-from defi_services.abis.lending.aave_v2_and_forlks.lending_pool_abi import LENDING_POOL_ABI
-from defi_services.abis.lending.aave_v2_and_forlks.oracle_abi import ORACLE_ABI
+from defi_services.abis.lending.aave_v2_and_forks.lending_pool_abi import LENDING_POOL_ABI
+from defi_services.abis.lending.aave_v2_and_forks.oracle_abi import ORACLE_ABI
 from defi_services.abis.lending.valas.chef_incentives_controller import CHEF_INCENTIVES_CONTROLLER
 from defi_services.abis.lending.valas.valas_multi_fee_distribution import VALAS_MULTI_FEE_DISTRIBUTION
 from defi_services.abis.token.erc20_abi import ERC20_ABI

defi_services/services/lending/venus_services.py

@@ -151,16 +151,22 @@ class VenusStateService(CompoundStateService):
 
     def calculate_rewards_balance(
             self, wallet: str, reserves_info: dict, decoded_data: dict, block_number: int = "latest"):
+        if not self.pool_info.get('lensAddress'):
+            return {}
+
         w3 = self.state_service.get_w3()
-        contract = w3.eth.contract(address=w3.to_checksum_address(self.pool_info.get("lensAddress")), abi = self.lens_abi)
+        contract = w3.eth.contract(address=w3.to_checksum_address(self.pool_info.get("lensAddress")), abi=self.lens_abi)
         # get_reward_id = f"pendingRewards_{self.name}_{wallet}_{block_number}".lower()
-        return_data = contract.functions.pendingRewards(w3.to_checksum_address(wallet), w3.to_checksum_address(self.pool_info.get("comptrollerAddress"))).call(block_identifier=block_number)
+        return_data = contract.functions.pendingRewards(
+            w3.to_checksum_address(wallet),
+            w3.to_checksum_address(self.pool_info.get("comptrollerAddress"))).call(block_identifier=block_number)
+
         rewards = return_data[2]
         for item in return_data[-1]:
             rewards += item[-1]
         reward_token = self.pool_info.get("rewardToken")
         result = {
-            reward_token: {"amount": rewards/10**18}
+            reward_token: {"amount": rewards / 10 ** 18}
         }
         return result