defi-state-querier 0.5.4__tar.gz → 0.5.6__tar.gz

{defi_state_querier-0.5.4 → defi_state_querier-0.5.6}/PKG-INFO

@@ -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 → defi_state_querier-0.5.6}/setup.py

@@ -29,8 +29,9 @@ setuptools.setup(
     python_requires='>=3.6',
     install_requires=[
         "web3>=6.0.0",
-        "query-blockchain-state-lib==1.0.1",
+        "query-blockchain-state-lib>=1.0.1",
         "python-dotenv>=1.0.0",
-        "pymongo>=4.3.3"
+        "pymongo>=4.3.3",
+        "cosmpy>=0.9.2"
     ],
 )

defi_state_querier-0.5.6/src/defi_services/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.5.6"

{defi_state_querier-0.5.4 → defi_state_querier-0.5.6}/src/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_state_querier-0.5.6/src/defi_services/jobs/processors/cosmos_state_processor.py

@@ -0,0 +1,43 @@
+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 = 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, 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 = []
+        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")
+            query_type = query.get("query_type")
+            if query_type != "token_balance":
+                continue
+
+            result.append(
+                {"query_id": query_id,
+                 "query_type": query_type,
+                 "entity_id": entity_id,
+                 "token_balance": token_balances.get(entity_id, 0)
+                 })
+        return result

defi_state_querier-0.5.6/src/defi_services/services/cosmos_token_services.py

@@ -0,0 +1,141 @@
+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, 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))
+        self.decimals = Denoms.cosmos
+        self.decimals.update(Denoms.orai)
+
+    # queries the balance of all coins for a single account.
+    def query_coin_balances(self, address: str):
+        responses = []
+        endpoint = '/cosmos/bank/v1beta1/balances/'
+        try:
+            results = json.loads(requests.get(self.lcd + endpoint + address, timeout=60).content)
+            responses += results['balances']
+            pagination = results['pagination']['next_key']
+            while pagination is not None:
+                results = json.loads(
+                    requests.get(self.lcd + endpoint + '?pagination.key=' + quote(str(pagination)), timeout=60).content)
+                responses += results['balances']
+                pagination = results['pagination']['next_key']
+            return responses
+        except Exception:
+            raise Exception
+
+    # queries the balance of a given denom for a single account.
+    def query_balances_by_denom(self, address, denom):
+        endpoint = '/cosmos/bank/v1beta1/balances/'
+        try:
+            return json.loads(
+                requests.get(self.lcd + endpoint + address + '/by_denom?denom=' + denom, timeout=60).content)
+        except Exception:
+            raise Exception
+
+    def query_token_decimal(self):
+        endpoint = '/cosmos/bank/v1beta1/denoms_metadata'
+        try:
+            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 → defi_state_querier-0.5.6}/src/defi_state_querier.egg-info/PKG-INFO

@@ -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 → defi_state_querier-0.5.6}/src/defi_state_querier.egg-info/requires.txt

@@ -1,4 +1,5 @@
 web3>=6.0.0
-query-blockchain-state-lib==1.0.1
+query-blockchain-state-lib>=1.0.1
 python-dotenv>=1.0.0
 pymongo>=4.3.3
+cosmpy>=0.9.2

defi_state_querier-0.5.4/src/defi_services/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.5.4"

defi_state_querier-0.5.4/src/defi_services/jobs/processors/cosmos_state_processor.py

@@ -1,37 +0,0 @@
-from defi_services.constants.cosmos_decimals_constant import Denoms
-from defi_services.services.cosmos_token_services import CosmosTokenServices
-
-
-class CosmosStateProcessor:
-    def __init__(self, lcd: str, denom: str):
-        self.lcd = lcd
-        self.denom = denom
-
-    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 run(self, address: str, queries: list):
-        result = []
-        token_balances = self.get_token_balance(address)
-        for query in queries:
-            query_id = query.get("query_id")
-            entity_id = query.get("entity_id")
-            query_type = query.get("query_type")
-            if query_type != "token_balance":
-                continue
-
-            result.append(
-                {"query_id": query_id,
-                 "query_type": query_type,
-                 "entity_id": entity_id,
-                 "token_balance": token_balances.get(entity_id, 0)
-                 })
-        return result

defi_state_querier-0.5.4/src/defi_services/services/cosmos_token_services.py

@@ -1,42 +0,0 @@
-import json
-from urllib.parse import quote
-import requests
-
-
-class CosmosTokenServices:
-    def __init__(self, lcd: str, denom: str):
-        self.lcd = lcd
-        self.denom = denom
-
-    # queries the balance of all coins for a single account.
-    def query_balances(self, address: str):
-        responses = []
-        endpoint = '/cosmos/bank/v1beta1/balances/'
-        try:
-            results = json.loads(requests.get(self.lcd + endpoint + address, timeout=60).content)
-            responses += results['balances']
-            pagination = results['pagination']['next_key']
-            while pagination is not None:
-                results = json.loads(
-                    requests.get(self.lcd + endpoint + '?pagination.key=' + quote(str(pagination)), timeout=60).content)
-                responses += results['balances']
-                pagination = results['pagination']['next_key']
-            return responses
-        except Exception:
-            raise Exception
-
-    # queries the balance of a given denom for a single account.
-    def query_balances_by_denom(self, address, denom):
-        endpoint = '/cosmos/bank/v1beta1/balances/'
-        try:
-            return json.loads(
-                requests.get(self.lcd + endpoint + address + '/by_denom?denom=' + denom, timeout=60).content)
-        except Exception:
-            raise Exception
-
-    def query_token_decimal(self):
-        endpoint = '/cosmos/bank/v1beta1/denoms_metadata'
-        try:
-            return json.loads(requests.get(self.lcd + endpoint, timeout=60).content)
-        except Exception:
-            raise Exception