defi-state-querier 0.4.2__py3-none-any.whl → 0.4.3__py3-none-any.whl

defi_services/__init__.py

@@ -1 +1 @@
-__version__ = "0.4.2"
+__version__ = "0.4.3"

defi_services/constants/entities/vault_constant.py

@@ -0,0 +1,5 @@
+class Vault:
+    # service
+    trava_vault = "trava-vault"
+
+    all = [trava_vault]

defi_services/constants/entities/vault_services.py

@@ -0,0 +1,24 @@
+from defi_services.constants.chain_constant import Chain
+from defi_services.constants.entities.vault_constant import Vault
+from defi_services.services.vault.trava_vault_services import TravaVaultStateService
+
+
+class VaultServices:
+    # chain
+    bsc = {
+        Vault.trava_vault: TravaVaultStateService
+    }
+
+    ethereum = {
+        Vault.trava_vault: TravaVaultStateService
+    }
+
+    fantom = {
+        Vault.trava_vault: TravaVaultStateService
+    }
+
+    mapping = {
+        Chain.bsc: bsc,
+        Chain.ethereum: ethereum,
+        Chain.fantom: fantom
+    }

defi_services/constants/query_constant.py

@@ -10,6 +10,7 @@ class Query:
     token_balance = 'token_balance'
     nft_balance = 'nft_balance'
     deposit_borrow = 'deposit_borrow'
+    staking_reward = 'staking_reward'
     protocol_reward = 'protocol_reward'
     protocol_apy = 'protocol_apy'
     health_factor = 'health_factor'

defi_services/jobs/processors/state_processor.py

@@ -4,12 +4,13 @@ from web3 import Web3
 
 from defi_services.constants.chain_constant import Chain
 from defi_services.constants.entities.lending_services import LendingServices
+from defi_services.constants.entities.vault_services import VaultServices
 from defi_services.constants.query_constant import Query
 from defi_services.jobs.queriers.state_querier import StateQuerier
 from defi_services.services.nft_services import NFTServices
 from defi_services.services.protocol_services import ProtocolServices
 from defi_services.services.token_services import TokenServices
-from defi_services.utils.convert_address import base58_to_hex, convert_address_dict
+from defi_services.utils.convert_address import base58_to_hex
 from defi_services.utils.init_services import init_services
 
 logger = logging.getLogger("StateProcessor")
@@ -23,6 +24,7 @@ class StateProcessor:
         self.token_service = TokenServices(self.state_querier, chain_id)
         self.nft_service = NFTServices(self.state_querier, chain_id)
         self.lending_services = LendingServices.mapping.get(chain_id)
+        self.vault_services = VaultServices.mapping.get(chain_id, {})
 
     def get_service_info(self):
         info = self.nft_service.get_service_info()
@@ -104,7 +106,7 @@ class StateProcessor:
                 if Query.nft_balance == query_type:
                     data = self.nft_service.get_data(wallet, entity, entity_value, block_number)
 
-            elif entity in self.lending_services:
+            elif (entity in self.lending_services) or (entity in self.vault_services):
                 entity_service: ProtocolServices = self.services.get(entity)
                 data = entity_service.get_data(
                     [query_type], wallet, entity_value, block_number, **kwargs)

defi_services/services/lending/aave_v2_services.py

@@ -438,7 +438,7 @@ class AaveV2StateService(ProtocolServices):
         return rpc_calls
 
     def calculate_rewards_balance(
-            self, decoded_data: dict, wallet: str, block_number: int = "latest"):
+            self, wallet: str, reserves_info: dict, decoded_data: dict, block_number: int = "latest"):
         reward_token = self.pool_info['rewardToken']
         key = f"getRewardsBalance_{self.name}_{wallet}_{block_number}".lower()
         rewards = decoded_data.get(key) / 10 ** 18

defi_services/services/lending/aave_v3_services.py

@@ -299,7 +299,7 @@ class AaveV3StateService(AaveV2StateService):
         return rpc_calls
 
     def calculate_rewards_balance(
-            self, decoded_data: dict, wallet: str, block_number: int = "latest"):
+            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))

defi_services/services/lending/apeswap_services.py

@@ -116,7 +116,7 @@ class ApeSwapStateService(CompoundStateService):
         return {}
 
     def calculate_rewards_balance(
-            self, decoded_data: dict, wallet: str, block_number: int = "latest"):
+            self, wallet: str, reserves_info: dict, decoded_data: dict, block_number: int = "latest"):
         return {}
 
     def get_ctoken_function_info(self, ctoken: str, fn_name: str, fn_paras: list = None, block_number: int = "latest"):

defi_services/services/lending/compound_service.py

@@ -282,7 +282,7 @@ class CompoundStateService(ProtocolServices):
         get_reward_id = f"getCompBalanceMetadataExt_{self.name}_{wallet}_{block_number}".lower()
         return {get_reward_id: rpc_call}
 
-    def calculate_rewards_balance(self, decoded_data: dict, wallet: str, block_number: int = "latest"):
+    def calculate_rewards_balance(self, wallet: str, reserves_info: dict, decoded_data: dict, block_number: int = "latest"):
         get_reward_id = f"getCompBalanceMetadataExt_{self.name}_{wallet}_{block_number}".lower()
         rewards = decoded_data.get(get_reward_id)[-1] / 10 ** 18
         reward_token = self.pool_info.get("rewardToken")

defi_services/services/lending/compound_v3_services.py

@@ -258,10 +258,7 @@ class CompoundV3StateService(CompoundStateService):
         return result
 
     def calculate_rewards_balance(
-            self,
-            decoded_data: dict,
-            wallet: str,
-            block_number: int = "latest"):
+            self, wallet: str, reserves_info: dict, decoded_data: dict, block_number: int = "latest"):
         reward_amount = 0
         reward_address = self.pool_info.get("rewardAddress")
         for token, value in self.pool_info.get("reservesList").items():

defi_services/services/lending/flux_services.py

@@ -89,10 +89,7 @@ class FluxStateService(CompoundStateService):
         return {get_reward_id: rpc_call}
 
     def calculate_rewards_balance(
-            self,
-            decoded_data: dict,
-            wallet: str,
-            block_number: int = "latest"):
+            self, wallet: str, reserves_info: dict, decoded_data: dict, block_number: int = "latest"):
         get_reward_id = f"compAccrued_{self.name}_{wallet}_{block_number}".lower()
         rewards = decoded_data.get(get_reward_id) / 10 ** 18
         reward_token = self.pool_info.get("rewardToken")

defi_services/services/lending/granary_services.py

@@ -331,10 +331,7 @@ class GranaryStateService(ProtocolServices):
         return rpc_calls
 
     def calculate_rewards_balance(
-            self,
-            decoded_data: dict,
-            wallet: str,
-            block_number: int = "latest"):
+            self, wallet: str, reserves_info: dict, decoded_data: dict, block_number: int = "latest"):
         reward_tokens = self.pool_info.get("rewardToken")
         if not reward_tokens:
             return {}

defi_services/services/lending/iron_bank_service.py

@@ -153,10 +153,7 @@ class IronBankStateService(CompoundStateService):
         return {get_reward_id: rpc_call}
 
     def calculate_rewards_balance(
-            self,
-            decoded_data: dict,
-            wallet: str,
-            block_number: int = "latest"):
+            self, wallet: str, reserves_info: dict, decoded_data: dict, block_number: int = "latest"):
         if self.chain_id in [Chain.optimism, Chain.avalanche]:
             return {}
         get_reward_id = f"compAccrued_{self.name}_{wallet}_{block_number}".lower()

defi_services/services/lending/justlend_service.py

@@ -92,17 +92,17 @@ class JustLendStateService(CompoundStateService):
     # REWARDS BALANCE
     def get_rewards_balance_function_info(
             self,
-            wallets: str,
+            wallet: str,
             reserves_info: dict = None,
             block_number: int = "latest",
     ):
-        rpc_call = self.get_comptroller_function_info("compAccrued", [wallets])
-        get_reward_id = f"compAccrued_{self.name}_{wallets}_{block_number}".lower()
+        rpc_call = self.get_comptroller_function_info("compAccrued", [wallet])
+        get_reward_id = f"compAccrued_{self.name}_{wallet}_{block_number}".lower()
         return {get_reward_id: rpc_call}
 
-    def calculate_rewards_balance(self, decoded_data: dict, wallets: str,
-                                  block_number: int = "latest"):
-        get_reward_id = f"compAccrued_{self.name}_{wallets}_{block_number}".lower()
+    def calculate_rewards_balance(
+            self, wallet: str, reserves_info: dict, decoded_data: dict, block_number: int = "latest"):
+        get_reward_id = f"compAccrued_{self.name}_{wallet}_{block_number}".lower()
         rewards = decoded_data.get(get_reward_id) / 10 ** 18
         reward_token = self.pool_info.get("rewardToken")
         result = {

defi_services/services/lending/lending_info/bsc/liqee_bsc.py

@@ -1,109 +1,157 @@
 LIQEE_BSC = {
-    "name": "Liqee Lending Pool",
-    "rewardToken": "0x4a9a2b2b04549c3927dd2c9668a5ef3fca473623",
-    "comptrollerAddress": "",
-    "lensAddress": "",
-    "controllerAddress": "0x0b53e608bd058bb54748c35148484fd627e6dc0a",
-    "lendingDataAddress": "0xeb92b72272e66be56b74392519cf81e150bfae0e",
-    "poolToken": "0x4a9a2b2b04549c3927dd2c9668a5ef3fca473623",
-    "type": "LENDING_POOL",
-    "forked": "compound",
-    "reservesList": {
-        "0xe9e7cea3dedca5984780bafc599bd69add087d56": {
-            "cToken": "0x5511b64ae77452c7130670c79298dec978204a47",
-            "liquidationThreshold": 0.75
-        },
-        "0x8ac76a51cc950d9822d68b83fe1ad97b32cd580d": {
-            "cToken": "0xaf9c10b341f55465e8785f0f81dbb52a9bfe005d",
-            "liquidationThreshold": 0.85
-        },
-        "0x55d398326f99059ff775485246999027b3197955": {
-            "cToken": "0x0bf8c72d618b5d46b055165e21d661400008fa0f",
-            "liquidationThreshold": 0.85
-        },
-        "0x1af3f329e8be154074d8769d1ffa4ee058b1dbc3": {
-            "cToken": "0xad5ec11426970c32da48f58c92b1039bc50e5492",
-            "liquidationThreshold": 0.85
-        },
-        "0x4a9a2b2b04549c3927dd2c9668a5ef3fca473623": {
-            "cToken": "0xec3fd540a2dee6f479be539d64da593a59e12d08",
-            "liquidationThreshold": 0.6
-        },
-        "0xa05478b34fad86cc86f07eb6698c69ea60b266e3": {
-            "cToken": "0xc35acaeedb814f42b2214378d8950f8555b2d670",
-            "liquidationThreshold": 0.8
-        },
-        "0x7130d2a12b9bcbfae4f2634d864a1ee1ce3ead9c": {
-            "cToken": "0x0b66a250dadf3237ddb38d485082a7bfe400356e",
-            "liquidationThreshold": 0.8
-        },
-        "0x2170ed0880ac9a755fd29b2688956bd959f933f8": {
-            "cToken": "0x390bf37355e9df6ea2e16eed5686886da6f47669",
-            "liquidationThreshold": 0.8
-        },
-        "0xbf5140a22578168fd562dccf235e5d43a02ce9b1": {
-            "cToken": "0xee9099c1318cf960651b3196747640eb84b8806b",
-            "liquidationThreshold": 0.7
-        },
-        "0x0eb3a705fc54725037cc9e008bdede697f62f335": {
-            "cToken": "0x55012ad2f0a50195aef44f403536df2465009ef7",
-            "liquidationThreshold": 0.7
-        },
-        "0x7083609fce4d1d8dc0c979aab8c869ea2c873402": {
-            "cToken": "0x9ab060ba568b86848bf19577226184db6192725b",
-            "liquidationThreshold": 0.7
-        },
-        "0x0d8ce2a99bb6e3b7db580ed848240e4a0f9ae153": {
-            "cToken": "0xd739a569ec254d6a20ecf029f024816be58fb810",
-            "liquidationThreshold": 0.7
-        },
-        "0x3ee2200efb3400fabb9aacf31297cbdd1d435d47": {
-            "cToken": "0xfc5bb1e8c29b100ef8f12773f972477bcab68862",
-            "liquidationThreshold": 0.7
-        },
-        "0x0000000000000000000000000000000000000000": {
-            "cToken": "0xd57e1425837567f74a35d07669b23bfb67aa4a93",
-            "liquidationThreshold": 0.8
-        },
-        "0xb5102cee1528ce2c760893034a4603663495fd72": {
-            "cToken": "0x36f4c36d1f6e8418ecb2402f896b2a8fedde0991",
-            "liquidationThreshold": 0.7
-        },
-        "0x367c17d19fcd0f7746764455497d63c8e8b2bba3": {
-            "cToken": "0xb22ef996c0a2d262a19db2a66a256067f51511eb",
-            "liquidationThreshold": 0.7
-        },
-        "0x20ecc92f0a33e16e8cf0417dfc3f586cf597f3a9": {
-            "cToken": "0x6e42423e1bcb6a093a58e203b5eb6e8a8023b4e5",
-            "liquidationThreshold": 0.7
-        },
-        "0x463e3d1e01d048fdf872710f7f3745b5cdf50d0e": {
-            "cToken": "0x6ac0a0b3959c1e5fcbd09b59b09abf7c53c72346",
-            "liquidationThreshold": 0.7
-        },
-        "0x1d2f0da169ceb9fc7b3144628db156f3f6c60dbe": {
-            "cToken": "0x6d64effe3af8697336fc57efd5a7517ad526dd6d",
-            "liquidationThreshold": 0.7
-        },
-        "0x4338665cbb7b2485a8855a139b75d5e34ab0db94": {
-            "cToken": "0xd957bea67aadb8a72061ce94d033c631d1c1e6ac",
-            "liquidationThreshold": 0.7
-        },
-        "0xf8a0bf9cf54bb92f17374d9e9a321e6a111a51bd": {
-            "cToken": "0x50e894894809f642de1e11b4076451734c963087",
-            "liquidationThreshold": 0.7
-        },
-        "0x0e09fabb73bd3ade0a17ecc321fd13a19e81ce82": {
-            "cToken": "0xefae8f7af4bada590d4e707d900258fc72194d73",
-            "liquidationThreshold": 0.7
-        },
-        "0x8ff795a6f4d97e7887c79bea79aba5cc76444adf": {
-            "cToken": "0x9747e26c5ad01d3594ea49ccf00790f564193c15",
-            "liquidationThreshold": 0.7
-        },
-        "0x16939ef78684453bfdfb47825f8a5f714f12623a": {
-            "cToken": "0x8be8cd81737b282c909f1911f3f0ade630c335aa",
-            "liquidationThreshold": 0.7
-        }
+  "comptrollerAddress": "",
+  "controllerAddress": "0x0b53e608bd058bb54748c35148484fd627e6dc0a",
+  "forked": "compound",
+  "lendingDataAddress": "0xeb92b72272e66be56b74392519cf81e150bfae0e",
+  "lensAddress": "",
+  "name": "Liqee Lending Pool",
+  "poolToken": "0x4a9a2b2b04549c3927dd2c9668a5ef3fca473623",
+  "reservesList": {
+    "0xe9e7cea3dedca5984780bafc599bd69add087d56": {
+      "cToken": "0x5511b64ae77452c7130670c79298dec978204a47",
+      "exchangeRate": 1.0727907490506678,
+      "liquidationThreshold": 0.75,
+      "loanToValue": 0.75
+    },
+    "0x8ac76a51cc950d9822d68b83fe1ad97b32cd580d": {
+      "cToken": "0xaf9c10b341f55465e8785f0f81dbb52a9bfe005d",
+      "exchangeRate": 1.0915276103263118,
+      "liquidationThreshold": 0.85,
+      "loanToValue": 0.85
+    },
+    "0x55d398326f99059ff775485246999027b3197955": {
+      "cToken": "0x0bf8c72d618b5d46b055165e21d661400008fa0f",
+      "exchangeRate": 1.107541263208309,
+      "liquidationThreshold": 0.85,
+      "loanToValue": 0.85
+    },
+    "0x1af3f329e8be154074d8769d1ffa4ee058b1dbc3": {
+      "cToken": "0xad5ec11426970c32da48f58c92b1039bc50e5492",
+      "exchangeRate": 1.090369296984632,
+      "liquidationThreshold": 0.85,
+      "loanToValue": 0.85
+    },
+    "0x4a9a2b2b04549c3927dd2c9668a5ef3fca473623": {
+      "cToken": "0xec3fd540a2dee6f479be539d64da593a59e12d08",
+      "exchangeRate": 1.023706366745555,
+      "liquidationThreshold": 0.6,
+      "loanToValue": 0.6
+    },
+    "0xa05478b34fad86cc86f07eb6698c69ea60b266e3": {
+      "cToken": "0xc35acaeedb814f42b2214378d8950f8555b2d670",
+      "exchangeRate": 1,
+      "liquidationThreshold": 0.8,
+      "loanToValue": 0.8
+    },
+    "0x7130d2a12b9bcbfae4f2634d864a1ee1ce3ead9c": {
+      "cToken": "0x0b66a250dadf3237ddb38d485082a7bfe400356e",
+      "exchangeRate": 1.0008955018668733,
+      "liquidationThreshold": 0.8,
+      "loanToValue": 0.8
+    },
+    "0x2170ed0880ac9a755fd29b2688956bd959f933f8": {
+      "cToken": "0x390bf37355e9df6ea2e16eed5686886da6f47669",
+      "exchangeRate": 1.0180063558704127,
+      "liquidationThreshold": 0.8,
+      "loanToValue": 0.8
+    },
+    "0xbf5140a22578168fd562dccf235e5d43a02ce9b1": {
+      "cToken": "0xee9099c1318cf960651b3196747640eb84b8806b",
+      "exchangeRate": 1.0014743731623101,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0x0eb3a705fc54725037cc9e008bdede697f62f335": {
+      "cToken": "0x55012ad2f0a50195aef44f403536df2465009ef7",
+      "exchangeRate": 1.0253052659477702,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0x7083609fce4d1d8dc0c979aab8c869ea2c873402": {
+      "cToken": "0x9ab060ba568b86848bf19577226184db6192725b",
+      "exchangeRate": 1.0615419244367703,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0x0d8ce2a99bb6e3b7db580ed848240e4a0f9ae153": {
+      "cToken": "0xd739a569ec254d6a20ecf029f024816be58fb810",
+      "exchangeRate": 1.0066415558494175,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0x3ee2200efb3400fabb9aacf31297cbdd1d435d47": {
+      "cToken": "0xfc5bb1e8c29b100ef8f12773f972477bcab68862",
+      "exchangeRate": 1.0138756837775817,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0x0000000000000000000000000000000000000000": {
+      "cToken": "0xd57e1425837567f74a35d07669b23bfb67aa4a93",
+      "exchangeRate": 1.1026774278138651,
+      "liquidationThreshold": 0.8,
+      "loanToValue": 0.8
+    },
+    "0xb5102cee1528ce2c760893034a4603663495fd72": {
+      "cToken": "0x36f4c36d1f6e8418ecb2402f896b2a8fedde0991",
+      "exchangeRate": 1,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0x367c17d19fcd0f7746764455497d63c8e8b2bba3": {
+      "cToken": "0xb22ef996c0a2d262a19db2a66a256067f51511eb",
+      "exchangeRate": 1,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0x20ecc92f0a33e16e8cf0417dfc3f586cf597f3a9": {
+      "cToken": "0x6e42423e1bcb6a093a58e203b5eb6e8a8023b4e5",
+      "exchangeRate": 1,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0x463e3d1e01d048fdf872710f7f3745b5cdf50d0e": {
+      "cToken": "0x6ac0a0b3959c1e5fcbd09b59b09abf7c53c72346",
+      "exchangeRate": 1,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0x1d2f0da169ceb9fc7b3144628db156f3f6c60dbe": {
+      "cToken": "0x6d64effe3af8697336fc57efd5a7517ad526dd6d",
+      "exchangeRate": 1.0095313806387172,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0x4338665cbb7b2485a8855a139b75d5e34ab0db94": {
+      "cToken": "0xd957bea67aadb8a72061ce94d033c631d1c1e6ac",
+      "exchangeRate": 1.009766921268889,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0xf8a0bf9cf54bb92f17374d9e9a321e6a111a51bd": {
+      "cToken": "0x50e894894809f642de1e11b4076451734c963087",
+      "exchangeRate": 1.0113349577900335,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0x0e09fabb73bd3ade0a17ecc321fd13a19e81ce82": {
+      "cToken": "0xefae8f7af4bada590d4e707d900258fc72194d73",
+      "exchangeRate": 1.3218997732331732,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0x8ff795a6f4d97e7887c79bea79aba5cc76444adf": {
+      "cToken": "0x9747e26c5ad01d3594ea49ccf00790f564193c15",
+      "exchangeRate": 1.086362444236688,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
+    },
+    "0x16939ef78684453bfdfb47825f8a5f714f12623a": {
+      "cToken": "0x8be8cd81737b282c909f1911f3f0ade630c335aa",
+      "exchangeRate": 1.0046816669180731,
+      "liquidationThreshold": 0.7,
+      "loanToValue": 0.7
     }
+  },
+  "rewardToken": "0x4a9a2b2b04549c3927dd2c9668a5ef3fca473623",
+  "type": "LENDING_POOL"
 }