defi-state-querier 0.0.6__py3-none-any.whl → 0.0.9__py3-none-any.whl

defi_services/services/lending/lending_info/ethereum/morpho_aave_v3_eth.py

@@ -0,0 +1,51 @@
+MORPHO_AAVE_V3_ETH = {
+    "name": "Compound Lending Pool",
+    "comptrollerAddress": "0x33333aea097c193e66081e930c33020272b33333",
+    "poolToken": "0x9994e35db50125e0df82e4c2dde62496ce330999",
+    "type": "LENDING_POOL",
+    "forked": "morpho-aave-v3",
+    "reservesList": {
+        "0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2": {
+            "tToken": "0x4d5f47fa6a74757f35c14fd3a6ef8e3c9bc514e8",
+            "sdToken": "0x102633152313c81cd80419b6ecf66d14ad68949a",
+            "dToken": "0xea51d7853eefb32b6ee06b1c12e6dcca88be0ffe",
+            "liquidationThreshold": 0.83
+        },
+        "0x7f39c581f595b53c5cb19bd0b3f8da6c935e2ca0": {
+            "tToken": "0x0b925ed163218f6662a35e0f0371ac234f9e9371",
+            "sdToken": "0x39739943199c0fbfe9e5f1b5b160cd73a64cb85d",
+            "dToken": "0xc96113eed8cab59cd8a66813bcb0ceb29f06d2e4",
+            "liquidationThreshold": 0.8
+        },
+        "0x6b175474e89094c44da98b954eedeac495271d0f": {
+            "tToken": "0x018008bfb33d285247a21d44e50697654f754e63",
+            "sdToken": "0x413adac9e2ef8683adf5ddaece8f19613d60d1bb",
+            "dToken": "0xcf8d0c70c850859266f5c338b38f9d663181c314",
+            "liquidationThreshold": 0.8
+        },
+        "0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48": {
+            "tToken": "0x98c23e9d8f34fefb1b7bd6a91b7ff122f4e16f5c",
+            "sdToken": "0xb0fe3d292f4bd50de902ba5bdf120ad66e9d7a39",
+            "dToken": "0x72e95b8931767c79ba4eee721354d6e99a61d004",
+            "liquidationThreshold": 0.8
+        },
+        "0x2260fac5e5542a773aa44fbcfedf7c193bc2c599": {
+            "tToken": "0x5ee5bf7ae06d1be5997a1a72006fe6c607ec6de8",
+            "sdToken": "0xa1773f1ccf6db192ad8fe826d15fe1d328b03284",
+            "dToken": "0x40aabef1aa8f0eec637e0e7d92fbffb2f26a8b7b",
+            "liquidationThreshold": 0.78
+        },
+        "0xae78736cd615f374d3085123a210448e74fc6393": {
+            "tToken": "0xcc9ee9483f662091a1de4795249e24ac0ac2630f",
+            "sdToken": "0x1d1906f909cae494c7441604dafdddbd0485a925",
+            "dToken": "0xae8593dd575fe29a9745056aa91c4b746eee62c8",
+            "liquidationThreshold": 0.74
+        },
+        "0xbe9895146f7af43049ca1c1ae358b0541ea49704": {
+            "tToken": "0x977b6fc5de62598b08c85ac8cf2b745874e8b78c",
+            "sdToken": "0x82be6012cea6d147b968ebaea5ceecf6a5b4f493",
+            "dToken": "0x0c91bca95b5fe69164ce583a2ec9429a569798ed",
+            "liquidationThreshold": 0.74
+        },
+    }
+}

defi_services/services/lending/lending_info/ethereum/morpho_compound_eth.py

@@ -0,0 +1,44 @@
+MORPHO_COMPOUND_ETH = {
+    "name": "Compound Lending Pool",
+    "rewardToken": "0xc00e94cb662c3520282e6f5717214004a7f26888",
+    "comptrollerAddress": "0x8888882f8f843896699869179fb6e4f7e3b58888",
+    "lensAddress": "0x930f1b46e1d081ec1524efd95752be3ece51ef67",
+    "rewardManager": '0x78681e63b6f3ad81ecd64aecc404d765b529c80d',
+    "poolToken": "0x9994e35db50125e0df82e4c2dde62496ce330999",
+    "type": "LENDING_POOL",
+    "forked": "morpho",
+    "reservesList": {
+        "0x6b175474e89094c44da98b954eedeac495271d0f": {
+            "cToken": "0x5d3a536e4d6dbd6114cc1ead35777bab948e3643",
+            "liquidationThreshold": 0.835
+        },
+        "0xc00e94cb662c3520282e6f5717214004a7f26888": {
+            "cToken": "0x70e36f6bf80a52b3b46b3af8e106cc0ed743e8e4",
+            "liquidationThreshold": 0.6
+        },
+        "0x0000000000000000000000000000000000000000": {
+            "cToken": "0x4ddc2d193948926d02f9b1fe9e1daa0718270ed5",
+            "liquidationThreshold": 0.825
+        },
+        "0x956f47f50a910163d8bf957cf5846d573e7f87ca": {
+            "cToken": "0x7713dd9ca933848f6819f38b8352d9a15ea73f67",
+            "liquidationThreshold": 0.0
+        },
+        "0x1f9840a85d5af5bf1d1762f925bdaddc4201f984": {
+            "cToken": "0x35a18000230da775cac24873d00ff85bccded550",
+            "liquidationThreshold": 0.75
+        },
+        "0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48": {
+            "cToken": "0x39aa39c021dfbae8fac545936693ac917d5e7563",
+            "liquidationThreshold": 0.855
+        },
+        "0xdac17f958d2ee523a2206206994597c13d831ec7": {
+            "cToken": "0xf650c3d88d12db855b8bf7d11be6c55a4e07dcc9",
+            "liquidationThreshold": 0.0
+        },
+        "0x2260fac5e5542a773aa44fbcfedf7c193bc2c599": {
+            "cToken": "0xccf4429db6322d5c611ee964527d42e5d685dd6a",
+            "liquidationThreshold": 0.7
+        },
+    }
+}

defi_services/services/lending/lending_info/ethereum/wepiggy_eth.py

@@ -0,0 +1,41 @@
+WEPIGGY_ETH = {
+    "name": "Wepiggy Lending Pool",
+    "rewardToken": "0x6f620ec89b8479e97a6985792d0c64f237566746",
+    "comptrollerAddress": "0x0c8c1ab017c3c0c8a48dd9f1db2f59022d190f0b",
+    "lensAddress": "0x6a05bd123d780055c38526cc05d3c9b90d0e471c",
+    "distributionAddress": "0x3e5496e50793e72e6143a15bed1c2535f0b0b9b0",
+    "poolToken": "0x6f620ec89b8479e97a6985792d0c64f237566746",
+    "type": "LENDING_POOL",
+    "forked": "compound",
+    "reservesList": {
+        "0x1f9840a85d5af5bf1d1762f925bdaddc4201f984": {
+            "cToken": "0x82413f75f0da101e0fe7f6ff6cba3461f7e04f29",
+            "liquidationThreshold": 0.6
+        },
+        "0x6b175474e89094c44da98b954eedeac495271d0f": {
+            "cToken": "0x85166b72c87697a6acff24101b43fd54fe28a179",
+            "liquidationThreshold": 0.75
+        },
+        "0xdac17f958d2ee523a2206206994597c13d831ec7": {
+            "cToken": "0x5cfad792c4df1323188180778aec58e00eace32a",
+            "liquidationThreshold": 0.8
+        },
+        "0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48": {
+            "cToken": "0xf8e5b9738bf63adfff36a849f9b9c9617c8d8c1f",
+            "liquidationThreshold": 0.8
+        },
+        "0x2260fac5e5542a773aa44fbcfedf7c193bc2c599": {
+            "cToken": "0xc12b9d620bfcb48be3e0ccbf0ea80c717333b46f",
+            "liquidationThreshold": 0.7
+        },
+        "0x0000000000000000000000000000000000000000": {
+            "cToken": "0x27a94869341838d5783368a8503fda5fbcd7987c",
+            "liquidationThreshold": 0.8
+        },
+        "0xbbbbca6a901c926f240b89eacb641d8aec7aeafd": {
+            "cToken": "0x690aa2591e57180cba5a6123e9d462907a5e1c95",
+            "liquidationThreshold": 0.5
+        },
+
+    }
+}

defi_services/services/lending/liqee_service.py

@@ -3,18 +3,16 @@ import time
 
 from web3 import Web3
 
-from defi_services.abis.lending.cream.cream_comptroller_abi import CREAM_COMPTROLLER_ABI
-from defi_services.abis.lending.cream.cream_lens_abi import CREAM_LENS_ABI
 from defi_services.abis.lending.liqee.liqee_comptroller_abi import LIQEE_CONTROLLER_ABI
 from defi_services.abis.lending.liqee.liqee_lending_data_abi import LIQEE_LENDING_DATA_ABI
 from defi_services.abis.lending.liqee.liqee_token_abi import LIQEE_TOKEN_ABI
 from defi_services.abis.token.ctoken_abi import CTOKEN_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.query_constant import Query
-from defi_services.constants.token_constant import ContractAddresses, Token
-from defi_services.jobs.state_querier import StateQuerier
+from defi_services.constants.token_constant import Token
+from defi_services.jobs.queriers.state_querier import StateQuerier
 from defi_services.services.lending.lending_info.ethereum.liqee_eth import LIQEE_ETH
 from defi_services.services.protocol_services import ProtocolServices
 
@@ -29,7 +27,7 @@ class LiqeeInfo:
 
 class LiqeeStateService(ProtocolServices):
     def __init__(self, state_service: StateQuerier, chain_id: str = "0x1"):
-        self.name = f"{chain_id}_liqee"
+        self.name = f"{chain_id}_{Lending.liqee}"
         self.chain_id = chain_id
         self.pool_info = LiqeeInfo.mapping.get(chain_id)
         self.state_service = state_service
@@ -40,7 +38,7 @@ class LiqeeStateService(ProtocolServices):
     # BASIC FUNCTIONS
     def get_service_info(self):
         info = {
-            "liqee": {
+            Lending.liqee: {
                 "chain_id": self.chain_id,
                 "type": "lending",
                 "protocol_info": self.pool_info
@@ -300,25 +298,6 @@ class LiqeeStateService(ProtocolServices):
             ctoken, CTOKEN_ABI, fn_name, fn_paras, block_number
         )
 
-    def get_ctoken_metadata_all(
-            self,
-            reserves_info: dict = None,
-            block_number: int = "latest"
-    ):
-        tokens = [Web3.toChecksumAddress(value['cToken']) for key, value in reserves_info.items()]
-        key = f"cTokenMetadataAll_{self.pool_info.get('lensAddress')}_{block_number}".lower()
-        return {
-            key: self.get_lens_function_info("cTokenMetadataAll", tokens, block_number)
-        }
-
-    def ctoken_underlying_price_all(
-            self, reserves_info, block_number: int = 'latest'):
-        tokens = [Web3.toChecksumAddress(value['cToken']) for key, value in reserves_info.items()]
-        key = f"cTokenUnderlyingPriceAll_{self.pool_info.get('lensAddress')}_{block_number}".lower()
-        return {
-            key: self.get_lens_function_info("cTokenUnderlyingPriceAll", tokens, block_number)
-        }
-
     def get_all_markets(
             self, block_number: int = 'latest'):
         key = f"getAllMarkets_{self.pool_info.get('comptrollerAddress')}_{block_number}".lower()

defi_services/services/lending/morpho_aave_v2_services.py

@@ -0,0 +1,89 @@
+import logging
+import time
+
+from web3 import Web3
+
+from defi_services.abis.lending.aave_v2_and_forlks.aave_v2_incentives_abi import AAVE_V2_INCENTIVES_ABI
+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.morpho.morpho_aave_v2_comptroller_abi import MORPHO_AAVE_V2_COMPTROLLER
+from defi_services.abis.lending.morpho.morpho_aave_v2_lens_abi import MORPHO_AAVE_V2_LENS_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.aave_v2_services import AaveInfo
+from defi_services.services.lending.lending_info.ethereum.morpho_aave_v2_eth import MORPHO_AAVE_V2_ETH
+from defi_services.services.lending.morpho_compound_services import MorphoCompoundServices
+
+logger = logging.getLogger("Compound Lending Pool State Service")
+
+
+class MorphoAaveV2Info:
+    mapping = {
+        Chain.ethereum: MORPHO_AAVE_V2_ETH
+    }
+
+
+class MorphoAaveV2Services(MorphoCompoundServices):
+    def __init__(self, state_service: StateQuerier, chain_id: str = "0x1"):
+        super().__init__(state_service, chain_id)
+        self.name = f"{chain_id}_{Lending.morpho_aave_v2}"
+        self.chain_id = chain_id
+        self.compound_info = AaveInfo.mapping.get(chain_id)
+        self.pool_info = MorphoAaveV2Info.mapping.get(chain_id)
+        self.state_service = state_service
+        self.lending_abi = LENDING_POOL_ABI
+        self.incentive_abi = AAVE_V2_INCENTIVES_ABI
+        self.oracle_abi = ORACLE_ABI
+        self.lens_abi = MORPHO_AAVE_V2_LENS_ABI
+        self.comptroller_abi = MORPHO_AAVE_V2_COMPTROLLER
+        self.market_key = 'tToken'
+
+    # BASIC FUNCTIONS
+    def get_service_info(self):
+        info = {
+            Lending.morpho_aave_v2: {
+                "chain_id": self.chain_id,
+                "type": "lending",
+                "protocol_info": self.pool_info
+            }
+        }
+        return info
+
+    def get_dapp_asset_info(
+            self,
+            block_number: int = "latest"):
+        begin = time.time()
+        _w3 = self.state_service.get_w3()
+        pool_address = Web3.toChecksumAddress(self.pool_info['address'])
+        contract = _w3.eth.contract(address=pool_address, abi=self.lending_abi)
+        comptroller_contract = _w3.eth.contract(
+            address=_w3.toChecksumAddress(self.pool_info.get("comptrollerAddress")), abi=self.comptroller_abi)
+        markets = comptroller_contract.functions.getAllMarkets().call(block_identifier=block_number)
+        reserves_list = contract.functions.getReservesList().call(block_identifier=block_number)
+        reserves_info = {}
+        for token in reserves_list:
+            value = contract.functions.getReserveData(token).call(block_identifier=block_number)
+            key = token.lower()
+            ttoken = value[7]
+            if ttoken in markets:
+                reserves_info[key] = {}
+                reserves_info[key]["tToken"] = ttoken.lower()
+                reserves_info[key]["dToken"] = value[9].lower()
+                reserves_info[key]["sdToken"] = value[8].lower()
+                risk_param = bin(value[0][0])[2:]
+                reserves_info[key]["liquidationThreshold"] = int(risk_param[-31:-16], 2) / 10 ** 4
+        logger.info(f"Get reserves information in {time.time() - begin}s")
+        return reserves_info
+
+    # REWARDS BALANCE
+    def get_claimable_rewards_balance_function_info(
+            self,
+            wallet_address: str,
+            block_number: int = "latest",
+    ):
+        return {}
+
+    def calculate_claimable_rewards_balance(self, wallet_address: str, decoded_data: dict,
+                                            block_number: int = "latest"):
+        return {}

defi_services/services/lending/morpho_aave_v3_services.py

@@ -0,0 +1,159 @@
+import logging
+import time
+
+from web3 import Web3
+
+from defi_services.abis.lending.aave_v2_and_forlks.aave_v2_incentives_abi import AAVE_V2_INCENTIVES_ABI
+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.morpho.morpho_aave_v3_comptroller_abi import MORPHO_AAVE_V3_COMPTROLLER_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.aave_v3_services import AaveV3Info
+from defi_services.services.lending.lending_info.ethereum.morpho_aave_v3_eth import MORPHO_AAVE_V3_ETH
+from defi_services.services.lending.morpho_compound_services import MorphoCompoundServices
+
+logger = logging.getLogger("Compound Lending Pool State Service")
+
+
+class MorphoAaveV3Info:
+    mapping = {
+        Chain.ethereum: MORPHO_AAVE_V3_ETH
+    }
+
+
+class MorphoAaveV3Services(MorphoCompoundServices):
+    def __init__(self, state_service: StateQuerier, chain_id: str = "0x1"):
+        super().__init__(state_service, chain_id)
+        self.name = f"{chain_id}_{Lending.morpho_aave_v3}"
+        self.chain_id = chain_id
+        self.aave_info = AaveV3Info.mapping.get(chain_id)
+        self.pool_info = MorphoAaveV3Info.mapping.get(chain_id)
+        self.state_service = state_service
+        self.lending_abi = LENDING_POOL_ABI
+        self.incentive_abi = AAVE_V2_INCENTIVES_ABI
+        self.oracle_abi = ORACLE_ABI
+        self.comptroller_abi = MORPHO_AAVE_V3_COMPTROLLER_ABI
+        self.market_key = 'tToken'
+
+    # BASIC FUNCTIONS
+    def get_service_info(self):
+        info = {
+            Lending.morpho_aave_v3: {
+                "chain_id": self.chain_id,
+                "type": "lending",
+                "protocol_info": self.pool_info
+            }
+        }
+        return info
+
+    def get_dapp_asset_info(
+            self,
+            block_number: int = "latest"):
+        begin = time.time()
+        _w3 = self.state_service.get_w3()
+        pool_address = Web3.toChecksumAddress(self.aave_info['address'])
+        contract = _w3.eth.contract(address=pool_address, abi=self.lending_abi)
+        comptroller_contract = _w3.eth.contract(
+            address=_w3.toChecksumAddress(self.pool_info.get("comptrollerAddress")), abi=self.comptroller_abi)
+        markets = comptroller_contract.functions.getAllMarkets().call(block_identifier=block_number)
+        markets = [i.lower() for i in markets]
+        reserves_list = contract.functions.getReservesList().call(block_identifier=block_number)
+        reserves_info = {}
+        for token in reserves_list:
+            value = contract.functions.getReserveData(token).call(block_identifier=block_number)
+            key = token.lower()
+            if token in markets:
+                reserves_info[key] = {}
+                reserves_info[key]["tToken"] = value[8].lower()
+                reserves_info[key]["dToken"] = value[9].lower()
+                reserves_info[key]["sdToken"] = value[10].lower()
+                risk_param = bin(value[0][0])[2:]
+                reserves_info[key]["liquidationThreshold"] = int(risk_param[-31:-16], 2) / 10 ** 4
+        logger.info(f"Get reserves information in {time.time() - begin}s")
+        return reserves_info
+
+    def get_token_list(self):
+        begin = time.time()
+        tokens = [self.aave_info.get("poolToken")]
+        for token in self.aave_info.get("reservesList"):
+            if token == Token.native_token:
+                tokens.append(Token.wrapped_token.get(self.chain_id))
+                continue
+            tokens.append(token)
+        logger.info(f"Get token list related in {time.time()-begin}s")
+        return tokens
+
+    # REWARDS BALANCE
+    def get_claimable_rewards_balance_function_info(
+            self,
+            wallet_address: str,
+            block_number: int = "latest",
+    ):
+        return {}
+
+    def calculate_claimable_rewards_balance(self, wallet_address: str, decoded_data: dict,
+                                            block_number: int = "latest"):
+        return {}
+
+    # WALLET DEPOSIT BORROW BALANCE
+    def get_wallet_deposit_borrow_balance_function_info(
+            self,
+            wallet_address: str,
+            reserves_info: dict,
+            block_number: int = "latest"
+    ):
+
+        rpc_calls = {}
+        for token, value in reserves_info.items():
+            underlying = token
+            ctoken = value.get(self.market_key)
+            if token == Token.native_token:
+                underlying = Token.wrapped_token.get(self.chain_id)
+            underlying_borrow_key = f"borrowBalance_{self.name}_{ctoken}_{wallet_address}_{block_number}".lower()
+            underlying_balance_key = f"collateralBalance_{self.name}_{ctoken}_{wallet_address}_{block_number}".lower()
+            underlying_decimals_key = f"decimals_{underlying}_{block_number}".lower()
+            rpc_calls[underlying_borrow_key] = self.get_comptroller_function_info(
+                "borrowBalance", [token, wallet_address], block_number)
+            rpc_calls[underlying_balance_key] = self.get_comptroller_function_info(
+                "collateralBalance", [token, wallet_address], block_number)
+            rpc_calls[underlying_decimals_key] = self.state_service.get_function_info(
+                underlying, ERC20_ABI, "decimals", [], block_number
+            )
+
+        return rpc_calls
+
+    def calculate_wallet_deposit_borrow_balance(
+            self, wallet_address: str, reserves_info: dict, decoded_data: dict, token_prices: dict = None,
+            block_number: int = "latest"):
+        if token_prices is None:
+            token_prices = {}
+        result = {}
+        for token, value in reserves_info.items():
+            underlying = token
+            ctoken = value.get(self.market_key)
+            if token == Token.native_token:
+                underlying = Token.wrapped_token.get(self.chain_id)
+            get_total_deposit_id = f"collateralBalance_{self.name}_{ctoken}_{wallet_address}_{block_number}".lower()
+            get_total_borrow_id = f"borrowBalance_{self.name}_{ctoken}_{wallet_address}_{block_number}".lower()
+            get_decimals_id = f"decimals_{underlying}_{block_number}".lower()
+            decimals = decoded_data[get_decimals_id]
+            deposit_amount = decoded_data[get_total_deposit_id] / 10 ** decimals
+            borrow_amount = decoded_data[get_total_borrow_id] / 10 ** decimals
+            result[token] = {
+                "borrow_amount": borrow_amount,
+                "deposit_amount": deposit_amount,
+            }
+            if token_prices:
+                token_price = token_prices.get(underlying)
+            else:
+                token_price = None
+            if token_price is not None:
+                deposit_amount_in_usd = deposit_amount * token_price
+                borrow_amount_in_usd = borrow_amount * token_price
+                result[token]['borrow_amount_in_usd'] += borrow_amount_in_usd
+                result[token]['deposit_amount_in_usd'] += deposit_amount_in_usd
+        return result