defi-state-querier 0.5.4__py3-none-any.whl → 0.5.6__py3-none-any.whl

defi_services/__init__.py

@@ -1 +1 @@
-__version__ = "0.5.4"
+__version__ = "0.5.6"

defi_services/constants/cosmos_decimals_constant.py

@@ -1,9 +1,5 @@
 class Denoms:
-    def __init__(self, denom: str, decimal: int):
-        self.denom = denom
-        self.decimal = decimal
-
-    data = {
+    cosmos = {
         "uatom": {
             "name": "ATOM",
             "denom": "uatom",
@@ -185,3 +181,124 @@ class Denoms:
             "decimal": 6
         }
     }
+    orai = {
+        "orai1gjdmce2cfu2ld7cw0p9p7mkmssnry4zhwz3tmsmah435pkdv9qkqg2n7qm": {
+            "name": "Aave USDT",
+            "denom": "orai1gjdmce2cfu2ld7cw0p9p7mkmssnry4zhwz3tmsmah435pkdv9qkqg2n7qm",
+            "decimal": 6
+        },
+        "orai10ldgzued6zjp0mkqwsv2mux3ml50l97c74x8sg": {
+            "name": "aiRight",
+            "denom": "orai10ldgzued6zjp0mkqwsv2mux3ml50l97c74x8sg",
+            "decimal": 6,
+            "base_denom": "AIRI"
+        },
+        "orai10g6frpysmdgw5tdqke47als6f97aqmr8s3cljsvjce4n5enjftcqtamzsd": {
+            "name": "Bitcoin",
+            "denom": "orai10g6frpysmdgw5tdqke47als6f97aqmr8s3cljsvjce4n5enjftcqtamzsd",
+            "decimal": 6
+        },
+        "ibc/a2e2eec9057a4a1c2c0a6a4c78b0239118df5f278830f50b4a6bdd7a66506b78": {
+            "name": "Cosmos Hub",
+            "denom": "ibc/a2e2eec9057a4a1c2c0a6a4c78b0239118df5f278830f50b4a6bdd7a66506b78",
+            "decimal": 6
+        },
+        "orai19rtmkk6sn4tppvjmp5d5zj6gfsdykrl5rw2euu5gwur3luheuuusesqn49": {
+            "name": "Injective",
+            "denom": "orai19rtmkk6sn4tppvjmp5d5zj6gfsdykrl5rw2euu5gwur3luheuuusesqn49",
+            "decimal": 6
+        },
+        "orai1nd4r053e3kgedgld2ymen8l9yrw8xpjyaal7j5": {
+            "name": "Kawaii Islands",
+            "denom": "orai1nd4r053e3kgedgld2ymen8l9yrw8xpjyaal7j5",
+            "decimal": 6
+        },
+        "orai1gzvndtzceqwfymu2kqhta2jn6gmzxvzqwdgvjw": {
+            "name": "Milky",
+            "denom": "orai1gzvndtzceqwfymu2kqhta2jn6gmzxvzqwdgvjw",
+            "decimal": 6,
+            "base_denom": "MILKY"
+        },
+        "ibc/576b1d63e401b6a9a071c78a1d1316d016ec9333d2feb14ad503fac4b8731cd1": {
+            "name": "Neutaro",
+            "denom": "ibc/576b1d63e401b6a9a071c78a1d1316d016ec9333d2feb14ad503fac4b8731cd1",
+            "decimal": 6
+        },
+        "orai1hn8w33cqvysun2aujk5sv33tku4pgcxhhnsxmvnkfvdxagcx0p8qa4l98q": {
+            "name": "OraiDEX",
+            "denom": "orai1hn8w33cqvysun2aujk5sv33tku4pgcxhhnsxmvnkfvdxagcx0p8qa4l98q",
+            "decimal": 6
+        },
+        "orai1lplapmgqnelqn253stz6kmvm3ulgdaytn89a8mz9y85xq8wd684s6xl3lt": {
+            "name": "Orchai",
+            "denom": "orai1lplapmgqnelqn253stz6kmvm3ulgdaytn89a8mz9y85xq8wd684s6xl3lt",
+            "decimal": 6
+        },
+        "orai": {
+            "name": "Oraichain",
+            "denom": "orai",
+            "decimal": 6,
+            "base_denom": "orai"
+        },
+        "orai1lus0f0rhx8s03gdllx2n6vhkmf0536dv57wfge": {
+            "name": "OraiDEX",
+            "denom": "orai1lus0f0rhx8s03gdllx2n6vhkmf0536dv57wfge",
+            "decimal": 6
+        },
+        "orai19q4qak2g3cj2xc2y3060t0quzn3gfhzx08rjlrdd3vqxhjtat0cq668phq": {
+            "name": "Orchai Protocol Staked Compound ATOM",
+            "denom": "orai19q4qak2g3cj2xc2y3060t0quzn3gfhzx08rjlrdd3vqxhjtat0cq668phq",
+            "decimal": 6
+        },
+        "ibc/9c4dcd21b48231d0bc2ac3d1b74a864746b37e4292694c93c617324250d002fc": {
+            "name": "Osmosis",
+            "denom": "ibc/9c4dcd21b48231d0bc2ac3d1b74a864746b37e4292694c93c617324250d002fc",
+            "decimal": 6
+        },
+        "orai1x6uwy6p5rkh4tv44pztafvkj7y9pgqpfyg82006ts4l97fy0kjzs5mzm8p": {
+            "name": "Stride Staked Atom",
+            "denom": "orai1x6uwy6p5rkh4tv44pztafvkj7y9pgqpfyg82006ts4l97fy0kjzs5mzm8p",
+            "decimal": 6
+        },
+        "orai1d2e02yn3h382ajsqjuz0xlc8qvv5d320m2walmcnfsc406azk9vqf639fa": {
+            "name": "Stride Staked Osmo",
+            "denom": "orai1d2e02yn3h382ajsqjuz0xlc8qvv5d320m2walmcnfsc406azk9vqf639fa",
+            "decimal": 6
+        },
+        "orai12hzjxfh77wl572gdzct2fxv2arxcwh6gykc7qh": {
+            "name": "Tether",
+            "denom": "orai12hzjxfh77wl572gdzct2fxv2arxcwh6gykc7qh",
+            "decimal": 6
+        },
+        "orai15un8msx3n5zf9ahlxmfeqd2kwa5wm0nrpxer304m9nd5q6qq0g6sku5pdd": {
+            "name": "USDC",
+            "denom": "orai15un8msx3n5zf9ahlxmfeqd2kwa5wm0nrpxer304m9nd5q6qq0g6sku5pdd",
+            "decimal": 6,
+            "base_denom": "USDC"
+        },
+        "orai1dqa52a7hxxuv8ghe7q5v0s36ra0cthea960q2cukznleqhk0wpnshfegez": {
+            "name": "WETH",
+            "denom": "orai1dqa52a7hxxuv8ghe7q5v0s36ra0cthea960q2cukznleqhk0wpnshfegez",
+            "decimal": 6
+        },
+        "orai1c7tpjenafvgjtgm9aqwm7afnke6c56hpdms8jc6md40xs3ugd0es5encn0": {
+            "name": "Wrapped Tron",
+            "denom": "orai1c7tpjenafvgjtgm9aqwm7afnke6c56hpdms8jc6md40xs3ugd0es5encn0",
+            "decimal": 6
+        },
+        "orai1065qe48g7aemju045aeyprflytemx7kecxkf5m7u5h5mphd0qlcs47pclp": {
+            "name": "Staking Compound ORAI",
+            "denom": "orai1065qe48g7aemju045aeyprflytemx7kecxkf5m7u5h5mphd0qlcs47pclp",
+            "decimal": 6
+        }
+    }
+    all = {**orai, **cosmos}
+
+
+class MulticallContract:
+    mapping = {
+        "orai": {
+            "multicall_balance": "orai1dyljypavg7qpt5d72a48a4pyg38d580aat55qql6tdcwfgydy6jsznk0h5",
+            "multicall": "orai1q7x644gmf7h8u8y6y8t9z9nnwl8djkmspypr6mxavsk9ual7dj0sxpmgwd"
+        }
+    }

defi_services/jobs/processors/cosmos_state_processor.py

@@ -1,26 +1,32 @@
-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, denom: str):
+    def __init__(self, lcd: str, rest_uri: str = None, chain_id: str = Chains.oraichain):
+        self.chain_id = chain_id
         self.lcd = lcd
-        self.denom = denom
+        self.rest_uri = rest_uri
 
-    def get_token_balance(self, address):
-        cosmos = CosmosTokenServices(self.lcd, self.denom)
-        data = cosmos.query_balances(address)
-        result = {}
-        for item in data:
-            denom = item.get('denom', "").lower()
-            amount = int(item.get('amount', None))
-            decimal = Denoms.data.get(denom, {}).get("decimal", 0)
-            result[denom] = amount / 10 ** decimal
-        return result
+    def get_token_balance(self, 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):
         result = []
-        token_balances = self.get_token_balance(address)
+        tokens = [query.get('entity_id').lower() for query in queries if query.get('query_type') == 'token_balance']
+        token_balances = self.get_token_balance(address, tokens)
         for query in queries:
             query_id = query.get("query_id")
             entity_id = query.get("entity_id")

defi_services/services/cosmos_token_services.py

@@ -1,15 +1,26 @@
 import json
 from urllib.parse import quote
 import requests
+import base64
+
+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, MulticallContract
 
 
 class CosmosTokenServices:
-    def __init__(self, lcd: str, denom: str):
+    def __init__(self, lcd: str, rest_uri: str, chain_id: str):
+        self.chain_id = chain_id
         self.lcd = lcd
-        self.denom = denom
+        self.rest_uri = rest_uri
+        self.client = CosmWasmRestClient(RestClient(rest_address=rest_uri))
+        self.decimals = Denoms.cosmos
+        self.decimals.update(Denoms.orai)
 
     # queries the balance of all coins for a single account.
-    def query_balances(self, address: str):
+    def query_coin_balances(self, address: str):
         responses = []
         endpoint = '/cosmos/bank/v1beta1/balances/'
         try:
@@ -40,3 +51,91 @@ class CosmosTokenServices:
             return json.loads(requests.get(self.lcd + endpoint, timeout=60).content)
         except Exception:
             raise Exception
+
+    def query_balances(self, address: str, tokens: list):
+        balance_query, cw20_tokens = self.get_balance_function_info(address, tokens)
+        queries = []
+        for token in cw20_tokens:
+            queries.append({
+                "address": token,
+                "data": self.encode_data({"token_info": {}})
+            })
+        queries.append({
+            "address": MulticallContract.mapping.get(self.chain_id).get("multicall_balance"),
+            "data": self.encode_data(balance_query)
+        })
+        query = {
+            "aggregate": {
+                "queries": queries
+            }
+        }
+        query = json.dumps(query).encode('utf-8')
+        request_ = QuerySmartContractStateRequest(
+            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
+
+    def decode_response_data(self, response_data, cw20_tokens, tokens):
+        response_data = json.loads(response_data.data)
+        return_data = response_data.get("return_data", [])
+        if not len(return_data):
+            return None
+        decimals = {}
+        for idx in range(0, len(cw20_tokens)):
+            encoded_data = return_data[idx]
+            decoded_data = self.decode_data(encoded_data['data'])
+            decimals[cw20_tokens[idx]] = decoded_data.get("decimals")
+
+        balance_data = self.decode_data(return_data[-1]['data'])
+        balances = {}
+        for idx in range(0, len(tokens)):
+            balances[tokens[idx]] = int(balance_data[idx])
+
+        for token in tokens:
+            if token in decimals:
+                balances[token] /= 10**decimals[token]
+            else:
+                balances[token] /= 10**self.decimals.get(token, {}).get("decimal", 0)
+
+        return balances
+
+    @staticmethod
+    def encode_data(params: dict):
+        data = json.dumps(params).encode('utf-8')
+        encode_data = base64.b64encode(data).decode()
+        return encode_data
+
+    @staticmethod
+    def decode_data(encode_data: str):
+        decoded_data = base64.b64decode(encode_data)
+        decoded_data = json.loads(decoded_data)
+        return decoded_data
+
+    @staticmethod
+    def get_balance_function_info(address: str, tokens: list):
+        data = {
+            "balance": {
+                "address": address,
+                "assets": []
+            }
+        }
+        cw20_tokens = []
+        for token in tokens:
+            head = token[:4]
+            if len(token) < 10 or head == 'ibc/':
+                data["balance"]["assets"].append({
+                    "native_token": {
+                        "denom": token
+                    }
+                })
+            else:
+                data["balance"]["assets"].append(
+                    {
+                        "token": {
+                            "contract_addr": token
+                        }
+                    }
+                )
+                cw20_tokens.append(token)
+        return data, cw20_tokens

{defi_state_querier-0.5.4.dist-info → defi_state_querier-0.5.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: defi-state-querier
-Version: 0.5.4
+Version: 0.5.6
 Summary: Calculate apy, apr, and wallet information,... in decentralized applications.
 Home-page: https://github.com/Centic-io/defi-state-querier
 Author: Viet-Bang Pham
@@ -13,9 +13,10 @@ Requires-Python: >=3.6
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: web3 >=6.0.0
-Requires-Dist: query-blockchain-state-lib ==1.0.1
+Requires-Dist: query-blockchain-state-lib >=1.0.1
 Requires-Dist: python-dotenv >=1.0.0
 Requires-Dist: pymongo >=4.3.3
+Requires-Dist: cosmpy >=0.9.2
 
 # defi-state-querier
 

{defi_state_querier-0.5.4.dist-info → defi_state_querier-0.5.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-defi_services/__init__.py,sha256=DITpct-LrdIsTgwx2NgH5Ghx5y8Xgz1YMimy1ZV5RTY,22
+defi_services/__init__.py,sha256=CMH34Gt1AqO7z_TqRj94XwohGoVCf8aes0djkqm45mk,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
@@ -295,7 +295,7 @@ defi_services/abis/vault/lido/lido_wsteth.py,sha256=DJhMToJS43vV2r5WP2kZSiagt7R8
 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=Qk51o5d3VNVYJxMqU23UXTOTWBIyKHlbQ-nuYS8An10,8630
+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
@@ -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=2S5fhqHQ4pUtRaLnbFJNUo4jKc5MHq7HA-0FHebOJyo,1334
+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,7 +325,7 @@ 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=6z_eE7lF5tY461VJ6uVoG18hOdXKeWR4NgDDP-MGI0k,1603
+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
@@ -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.4.dist-info/LICENSE,sha256=6jmfxa8nUIwfKnzZUxAHJSJ_IS7h7mpbJq26cWjoo-o,1063
-defi_state_querier-0.5.4.dist-info/METADATA,sha256=gTQEjp_THAVXFoD6zox3Quu6bFZqKb4hXlMPBCcgMzw,4387
-defi_state_querier-0.5.4.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-defi_state_querier-0.5.4.dist-info/top_level.txt,sha256=C-OTxHK6MknKK-nAbEzCPDUl1M6pktRhgJrmsozdf6g,14
-defi_state_querier-0.5.4.dist-info/RECORD,,
+defi_state_querier-0.5.6.dist-info/LICENSE,sha256=6jmfxa8nUIwfKnzZUxAHJSJ_IS7h7mpbJq26cWjoo-o,1063
+defi_state_querier-0.5.6.dist-info/METADATA,sha256=jksOD0A7CWoHH9ykENU3VhLBGav0P_DA5MWriPxQ74o,4417
+defi_state_querier-0.5.6.dist-info/WHEEL,sha256=mguMlWGMX-VHnMpKOjjQidIo1ssRlCFu4a4mBpz1s2M,91
+defi_state_querier-0.5.6.dist-info/top_level.txt,sha256=C-OTxHK6MknKK-nAbEzCPDUl1M6pktRhgJrmsozdf6g,14
+defi_state_querier-0.5.6.dist-info/RECORD,,

{defi_state_querier-0.5.4.dist-info → defi_state_querier-0.5.6.dist-info}/WHEEL

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