defi-state-querier 0.5.5__py3-none-any.whl → 0.5.7__py3-none-any.whl

defi_services/__init__.py

@@ -1 +1 @@
-__version__ = "0.5.5"
+__version__ = "0.5.7"

defi_services/constants/chain_constant.py

@@ -28,6 +28,8 @@ class Chain:
     okc = '0x42'
     aurora = '0x4e454152'
     moonriver = '0x505'
+    blast = '0xee'
+    oasis_sapphire = '0x5afe'
 
     native_decimals = {
         tron: 6,

defi_services/constants/cosmos_decimals_constant.py

@@ -292,3 +292,13 @@ class Denoms:
             "decimal": 6
         }
     }
+    all = {**orai, **cosmos}
+
+
+class MulticallContract:
+    mapping = {
+        "orai": {
+            "multicall_balance": "orai1dyljypavg7qpt5d72a48a4pyg38d580aat55qql6tdcwfgydy6jsznk0h5",
+            "multicall": "orai1q7x644gmf7h8u8y6y8t9z9nnwl8djkmspypr6mxavsk9ual7dj0sxpmgwd"
+        }
+    }

defi_services/constants/token_constant.py

@@ -43,7 +43,9 @@ class Token:
         Chain.linea: '0xe5d7c2a44ffddf6b295a15c148167daaaf5cf34f',
         Chain.okc: '0x8f8526dbfd6e38e3d8307702ca8469bae6c56c15',
         Chain.aurora: '0xc9bdeed33cd01541e1eed10f90519d2c06fe3feb',
-        Chain.moonriver: '0x98878b06940ae243284ca214f92bb71a2b032b8a'
+        Chain.moonriver: '0x98878b06940ae243284ca214f92bb71a2b032b8a',
+        Chain.blast: '0x4300000000000000000000000000000000000004',
+        Chain.oasis_sapphire: '0x8bc2b030b299964eefb5e1e0b36991352e56d2d3'
     }
 
 

defi_services/jobs/processors/cosmos_state_processor.py

@@ -1,15 +1,26 @@
-from defi_services.constants.cosmos_decimals_constant import Denoms
+from defi_services.constants.cosmos_decimals_constant import Denoms, MulticallContract
+from defi_services.constants.network_constants import Chains
 from defi_services.services.cosmos_token_services import CosmosTokenServices
 
 
 class CosmosStateProcessor:
-    def __init__(self, lcd: str, rest_uri: str):
+    def __init__(self, lcd: str, rest_uri: str = None, chain_id: str = Chains.oraichain):
+        self.chain_id = chain_id
         self.lcd = lcd
         self.rest_uri = rest_uri
 
     def get_token_balance(self, address, tokens):
-        cosmos = CosmosTokenServices(self.lcd, self.rest_uri)
-        data = cosmos.query_balances(address, tokens)
+        cosmos = CosmosTokenServices(self.lcd, self.rest_uri, self.chain_id)
+        if self.chain_id in MulticallContract.mapping:
+            data = cosmos.query_balances(address, tokens)
+        else:
+            data = cosmos.query_coin_balances(address=address)
+            result = {}
+            for item in data:
+                denom = item.get('denom', "").lower()
+                amount = int(item.get('amount', None))
+                decimal = Denoms.all.get(denom, {}).get("decimal", 0)
+                result[denom] = amount / 10 ** decimal
         return data
 
     def run(self, address: str, queries: list):

defi_services/services/cosmos_token_services.py

@@ -7,11 +7,12 @@ from cosmpy.cosmwasm.rest_client import RestClient
 from cosmpy.cosmwasm.rest_client import CosmWasmRestClient
 from cosmpy.protos.cosmwasm.wasm.v1.query_pb2 import QuerySmartContractStateRequest
 
-from defi_services.constants.cosmos_decimals_constant import Denoms
+from defi_services.constants.cosmos_decimals_constant import Denoms, MulticallContract
 
 
 class CosmosTokenServices:
-    def __init__(self, lcd: str, rest_uri: str):
+    def __init__(self, lcd: str, rest_uri: str, chain_id: str):
+        self.chain_id = chain_id
         self.lcd = lcd
         self.rest_uri = rest_uri
         self.client = CosmWasmRestClient(RestClient(rest_address=rest_uri))
@@ -60,7 +61,7 @@ class CosmosTokenServices:
                 "data": self.encode_data({"token_info": {}})
             })
         queries.append({
-            "address": "orai1dyljypavg7qpt5d72a48a4pyg38d580aat55qql6tdcwfgydy6jsznk0h5",
+            "address": MulticallContract.mapping.get(self.chain_id).get("multicall_balance"),
             "data": self.encode_data(balance_query)
         })
         query = {
@@ -70,7 +71,7 @@ class CosmosTokenServices:
         }
         query = json.dumps(query).encode('utf-8')
         request_ = QuerySmartContractStateRequest(
-            address="orai1q7x644gmf7h8u8y6y8t9z9nnwl8djkmspypr6mxavsk9ual7dj0sxpmgwd", query_data=query)
+            address=MulticallContract.mapping.get(self.chain_id).get("multicall"), query_data=query)
         response_ = self.client.SmartContractState(request_)
         decoded_data = self.decode_response_data(response_, cw20_tokens, tokens)
         return decoded_data

defi_services/services/token_services.py

@@ -61,5 +61,10 @@ class TokenServices:
         decimals_token = token
         if token == Token.native_token:
             decimals_token = Token.wrapped_token.get(self.chain_id)
+
+            # Ignore decimals of missing native's wrapped token
+            if not decimals_token:
+                return {}
+
         key = f"decimals_{token}_{block_number}".lower()
         return {key: self.state_service.get_function_info(decimals_token, ERC20_ABI, "decimals", [], block_number)}

{defi_state_querier-0.5.5.dist-info → defi_state_querier-0.5.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: defi-state-querier
-Version: 0.5.5
+Version: 0.5.7
 Summary: Calculate apy, apr, and wallet information,... in decentralized applications.
 Home-page: https://github.com/Centic-io/defi-state-querier
 Author: Viet-Bang Pham

{defi_state_querier-0.5.5.dist-info → defi_state_querier-0.5.7.dist-info}/RECORD

@@ -1,4 +1,4 @@
-defi_services/__init__.py,sha256=78mfpLewKVki6c9UONSUdlVme_JsN9ZwIfp4Hf4jmG0,22
+defi_services/__init__.py,sha256=KiyyYbyEe0O858kmiWcg1RdmqGUYtk_JqRmc3_Ev2Q8,22
 defi_services/abis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 defi_services/abis/multicall_v3_abi.py,sha256=0aPxjrJJFU17fODjvYFRDn5Y5J1yi_AJKc8v1uohNGY,12352
 defi_services/abis/dex/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -294,14 +294,14 @@ defi_services/abis/vault/lido/lido_withdrawl_queue.py,sha256=qKdsXQLKlNU3a_Nk68q
 defi_services/abis/vault/lido/lido_wsteth.py,sha256=DJhMToJS43vV2r5WP2kZSiagt7R8MtKFT8u1xJBjZKU,8780
 defi_services/abis/vault/lido/lidor_accounting_oracle_hash_consensus.py,sha256=TWCLaccDgvWJZZKLAeoznYgGi3g3M_hFqoo3KE_L7K8,21039
 defi_services/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-defi_services/constants/chain_constant.py,sha256=aU7xLaQYLro-7sI5Jpyi0tEHhEQcchyxlxEEJaotFMM,1018
-defi_services/constants/cosmos_decimals_constant.py,sha256=dWhP6QRbvatuV5j2qEZMq2b4WXSFnU6DTWN4khQ2ac8,13338
+defi_services/constants/chain_constant.py,sha256=llTCq697jtt64a_6QRxZK71Q3SMeoIyOY8RJXw5i5s8,1067
+defi_services/constants/cosmos_decimals_constant.py,sha256=SjFkFbkvxVAk0jtok_1EaBh69ZE8S1fFAbVn6bdJO4s,13635
 defi_services/constants/db_constant.py,sha256=gDMFFjhPOgQHuS9sPOTFbsfSdIfdWQEE4tGlxHQhNsE,2566
 defi_services/constants/mongo_constant.py,sha256=qsDIq2Gg5MkBGaEOh4Fd9y6fOOidF4gGPnQwtaPon3E,398
 defi_services/constants/network_constants.py,sha256=aCUhPFOA9oAMXrIrw67wfEvc4B3HcTlCCz9U1Ku4o50,13574
 defi_services/constants/query_constant.py,sha256=6HSK5IToUQpKLqmTeCgKf0npwmog7GWFUmQpz_mbkgY,1088
 defi_services/constants/time_constant.py,sha256=bC-0pYnfIFFR_0iVvkxhfyHfOclWNw9O3DbdO_nh258,181
-defi_services/constants/token_constant.py,sha256=4iDWis589XF4_OIcBOqA7_qIfCde-nV6fDaV__5c2D4,4822
+defi_services/constants/token_constant.py,sha256=dGO_i-L_LLx4mSA0wIG4FoaL-OWOAd5bp6k7VucUjUQ,4965
 defi_services/constants/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 defi_services/constants/entities/dex_constant.py,sha256=oCt8RS-0KjFW5zR5IU1Ys49nYCCAraY8pTivOpXoyvs,343
 defi_services/constants/entities/dex_info_constant.py,sha256=6sLHxCmRz_6HW8WRe27FkOS-FGR-zC354iCTcEIjp78,1604
@@ -313,7 +313,7 @@ defi_services/constants/entities/vault_services.py,sha256=IMmQc15wS1KA7EuQqM_Ara
 defi_services/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 defi_services/jobs/processors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 defi_services/jobs/processors/call_state_processor.py,sha256=WseX0OaMA5D_ANYofrz7W2BooSMdHlGRay4NuiABmM4,436
-defi_services/jobs/processors/cosmos_state_processor.py,sha256=yGC0RRcSKslzUbAxfZzvQm0mRfBO4YWa7a8V1kKQ_RI,1220
+defi_services/jobs/processors/cosmos_state_processor.py,sha256=eZa2fO5Vh4n_VTJjKSHTud325KjHvFA_e8NcywDEun4,1813
 defi_services/jobs/processors/multi_call_state_processor.py,sha256=AksswGMSyStbnvkQcBVb0Y2tdpkUgc1CWoeSiMpGy8U,462
 defi_services/jobs/processors/multi_state_processor.py,sha256=Od7Y8zPz7xj0L-_0YuK4wg_94fX8Q1PquuKUUYteIJU,5731
 defi_services/jobs/processors/solana_state_processor.py,sha256=szsBYrkEV8kBzuA_iaSVescnwKJ2MHoTbut5kORbWTs,4065
@@ -325,13 +325,13 @@ defi_services/jobs/queriers/solana_state_querier.py,sha256=TQELYo6GUoF8s-LfetqYb
 defi_services/jobs/queriers/state_querier.py,sha256=q7cAW395urZjcWTpqw8m_nDunLyyTuAXJ3DRx_1HOxo,7384
 defi_services/jobs/queriers/substrate_state_querier.py,sha256=_T0Dk06sP_uBKaxgj2J_EBtJDoq-hi8jU7Np4iuO0LA,3858
 defi_services/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-defi_services/services/cosmos_token_services.py,sha256=8LfaKUxcaj9IMQm31a_UY5Zqayo0AmkvgIWXsgTcCjY,5181
+defi_services/services/cosmos_token_services.py,sha256=-QGmhS-WtwJbOnOFQ7CuM10glPgzZvYPtPlCE6IZIKM,5248
 defi_services/services/dex_protocol_services.py,sha256=J93P7kwy1v98aeA95ErT7jDCCOAelFuCNiTXl3PVFFo,6656
 defi_services/services/nft_services.py,sha256=vKva0OJQTTxgP1lrZ5ij2sw2N44JxN5dfLLtvNJB6aA,2188
 defi_services/services/protocol_services.py,sha256=3Yca433xGXowcV4EjFoQ90AJHg-SfuOBsDF7aMCJ8Zk,7626
 defi_services/services/solana_token_services.py,sha256=NQKUHGMBe9eQciOaY3N20rPbIuNz108akrysX-mXfJU,1865
 defi_services/services/substrate_token_services.py,sha256=HsjceBHo0fxQIcXgDaV0OBANLuggRwXAEPt7ASB0BKk,2904
-defi_services/services/token_services.py,sha256=t6eJYemNyiG_4TObjKnQOz8MmSqXBKjsC0A3fPBeDuM,2800
+defi_services/services/token_services.py,sha256=vISaOgwHsokpwWENGshRssvq2xuCVeL5gtr20SSYX78,2927
 defi_services/services/dex/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 defi_services/services/dex/pancakeswap_service.py,sha256=cN_L7897l8dh7DnNuHo6fcnYa-JLHI7yLLzBW1C9RlI,4637
 defi_services/services/dex/pancakeswap_v2_service.py,sha256=ppRruxXHKeY3BQY5DxBQoNDjNIc6kqo3Wjgd6mO9CcA,13098
@@ -464,8 +464,8 @@ defi_services/utils/market_service.py,sha256=imPtPHBkpEx5JnhqeIWYqbCjsIEm8IKBYHN
 defi_services/utils/memory_storage.py,sha256=BOT8laB0iVSCGE-oDlpWJQLbSC6X2blKX4zuQbs4inc,851
 defi_services/utils/sqrt_price_math.py,sha256=9lgUeWFT4wjl3Vq3b7-jZ2bGvvZx7dDBSfVnM3lsZ8o,5575
 defi_services/utils/thread_proxy.py,sha256=5Z8biAyEReUkh3vfJSvEv7GwMe3CsE5M8CbghkQtePw,2951
-defi_state_querier-0.5.5.dist-info/LICENSE,sha256=6jmfxa8nUIwfKnzZUxAHJSJ_IS7h7mpbJq26cWjoo-o,1063
-defi_state_querier-0.5.5.dist-info/METADATA,sha256=djRcAZmtLY0cG2hmAVNZoR7fnOYwhpMj43OuFucjMnI,4417
-defi_state_querier-0.5.5.dist-info/WHEEL,sha256=mguMlWGMX-VHnMpKOjjQidIo1ssRlCFu4a4mBpz1s2M,91
-defi_state_querier-0.5.5.dist-info/top_level.txt,sha256=C-OTxHK6MknKK-nAbEzCPDUl1M6pktRhgJrmsozdf6g,14
-defi_state_querier-0.5.5.dist-info/RECORD,,
+defi_state_querier-0.5.7.dist-info/LICENSE,sha256=6jmfxa8nUIwfKnzZUxAHJSJ_IS7h7mpbJq26cWjoo-o,1063
+defi_state_querier-0.5.7.dist-info/METADATA,sha256=2jNWm8RrkBwLUcc5rH05G4bzi4I98wrRKdKw6dyAttw,4417
+defi_state_querier-0.5.7.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+defi_state_querier-0.5.7.dist-info/top_level.txt,sha256=C-OTxHK6MknKK-nAbEzCPDUl1M6pktRhgJrmsozdf6g,14
+defi_state_querier-0.5.7.dist-info/RECORD,,

{defi_state_querier-0.5.5.dist-info → defi_state_querier-0.5.7.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (70.1.1)
+Generator: setuptools (70.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any