strideutils 0.8.7__tar.gz → 0.8.8__tar.gz

{strideutils-0.8.7 → strideutils-0.8.8}/.bumpversion.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.8.7
+current_version = 0.8.8
 commit = False
 tag = False
 

{strideutils-0.8.7 → strideutils-0.8.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: strideutils
-Version: 0.8.7
+Version: 0.8.8
 Summary: Utility functions for monitoring the Stride blockchain
 Author-email: Stride Labs <hello@stridelabs.co>
 License: MIT License
@@ -44,6 +44,7 @@ Requires-Dist: python-dotenv==1.0.1
 Requires-Dist: pyyaml>=6.0.1
 Requires-Dist: requests>=2.32.3
 Requires-Dist: pycryptodome>=3.20.0
+Requires-Dist: web3==7.6.1
 Provides-Extra: dev
 Requires-Dist: pytest>=8.1.1; extra == "dev"
 Requires-Dist: mypy>=1.10.0; extra == "dev"

{strideutils-0.8.7 → strideutils-0.8.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "strideutils"
-version = "0.8.7"
+version = "0.8.8"
 authors = [
     {name = "Stride Labs", email = "hello@stridelabs.co"},
 ]
@@ -32,6 +32,7 @@ dependencies = [
     "pyyaml>=6.0.1",
     "requests>=2.32.3",
     "pycryptodome>=3.20.0",
+    "web3==7.6.1"
 ]
 
 [project.optional-dependencies]

{strideutils-0.8.7 → strideutils-0.8.8}/strideutils/stride_requests.py

@@ -11,6 +11,7 @@ import pandas as pd
 import pytz
 import requests
 from requests.adapters import HTTPAdapter, Retry
+from web3 import Web3
 
 from strideutils import types
 from strideutils.stride_config import config
@@ -58,6 +59,16 @@ COSMWASM_CONTRACT_QUERY = "cosmwasm/wasm/v1/contract/{contract_address}/smart/{q
 CHANNELS_SUMMARY_QUERY_BY_HOST = "https://channels.main.stridenet.co/api/data/{chain_id}"
 CHANNELS_SUMMARY_QUERY = "https://channels.main.stridenet.co/api/data"
 
+ERC20_ABI = [
+    {
+        "constant": True,
+        "inputs": [{"name": "_owner", "type": "address"}],
+        "name": "balanceOf",
+        "outputs": [{"name": "balance", "type": "uint256"}],
+        "type": "function",
+    }
+]
+
 requests_session = requests.Session()
 
 retries = Retry(total=5, backoff_factor=0.5, status_forcelist=[400, 403, 404, 500, 502, 503, 504])
@@ -1351,12 +1362,22 @@ def query_evm_method(method: str, params: list[str], rpc_endpoint: str) -> str:
     return response["result"]
 
 
-def get_evm_balance(address: str, rpc_endpoint: str) -> int:
+def get_evm_balance(address: str, rpc_endpoint: str, token_address: str | None = None, decimals: int = 16) -> int:
     """
-    Queries the native token balance of an evm address
+    Queries the native or ERC20 token balance of an evm address
     Address should be a hex address with an 0x prefix
     """
-    params = [address, "latest"]
-    balance_raw = query_evm_method("eth_getBalance", params=params, rpc_endpoint=rpc_endpoint)
-    balance = int(balance_raw, 16)
-    return balance
+    w3 = Web3(Web3.HTTPProvider(rpc_endpoint))
+    assert w3.is_address(address), f"Invalid wallet address: {address}"
+    assert not token_address or w3.is_address(token_address), f"Invalid token address: {token_address}"
+
+    wallet_address = Web3.to_checksum_address(address)
+
+    if not token_address:
+        balance = w3.eth.get_balance(wallet_address)
+    else:
+        token_address = Web3.to_checksum_address(token_address)
+        contract = w3.eth.contract(address=token_address, abi=ERC20_ABI)
+        balance = contract.functions.balanceOf(wallet_address).call()
+
+    return balance / (10**decimals)

{strideutils-0.8.7 → strideutils-0.8.8}/strideutils/stride_requests.pyi

@@ -39,6 +39,7 @@ MULTISIG_REDEMPTION_RECORDS_QUERY: str
 COSMWASM_CONTRACT_QUERY: str
 CHANNELS_SUMMARY_QUERY_BY_HOST: str
 CHANNELS_SUMMARY_QUERY: str
+ERC20_ABI: Incomplete
 requests_session: Incomplete
 retries: Incomplete
 
@@ -95,4 +96,4 @@ def get_host_zone_channels(chain_id: str) -> dict[str, Any]: ...
 def query_wasm_contract(contract_address: str, query_msg: dict, api_endpoint: str, block_height: int = 0) -> dict: ...
 def get_module_accounts(api_endpoint: str = ...) -> list[dict]: ...
 def query_evm_method(method: str, params: list[str], rpc_endpoint: str) -> str: ...
-def get_evm_balance(address: str, rpc_endpoint: str) -> int: ...
+def get_evm_balance(address: str, rpc_endpoint: str, token_address: str | None = None, decimals: int = 16) -> int: ...

{strideutils-0.8.7 → strideutils-0.8.8}/strideutils-stubs/stride_requests.pyi

@@ -41,6 +41,7 @@ MULTISIG_REDEMPTION_RECORDS_QUERY: str
 COSMWASM_CONTRACT_QUERY: str
 CHANNELS_SUMMARY_QUERY_BY_HOST: str
 CHANNELS_SUMMARY_QUERY: str
+ERC20_ABI: Incomplete
 requests_session: Incomplete
 retries: Incomplete
 
@@ -137,4 +138,4 @@ def get_host_zone_channels(chain_id: str) -> dict[str, Any]: ...
 def query_wasm_contract(contract_address: str, query_msg: dict, api_endpoint: str, block_height: int = 0) -> dict: ...
 def get_module_accounts(api_endpoint: str = ...) -> list[dict]: ...
 def query_evm_method(method: str, params: list[str], rpc_endpoint: str) -> str: ...
-def get_evm_balance(address: str, rpc_endpoint: str) -> int: ...
+def get_evm_balance(address: str, rpc_endpoint: str, token_address: str | None = None, decimals: int = 16) -> int: ...

{strideutils-0.8.7 → strideutils-0.8.8}/strideutils.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: strideutils
-Version: 0.8.7
+Version: 0.8.8
 Summary: Utility functions for monitoring the Stride blockchain
 Author-email: Stride Labs <hello@stridelabs.co>
 License: MIT License
@@ -44,6 +44,7 @@ Requires-Dist: python-dotenv==1.0.1
 Requires-Dist: pyyaml>=6.0.1
 Requires-Dist: requests>=2.32.3
 Requires-Dist: pycryptodome>=3.20.0
+Requires-Dist: web3==7.6.1
 Provides-Extra: dev
 Requires-Dist: pytest>=8.1.1; extra == "dev"
 Requires-Dist: mypy>=1.10.0; extra == "dev"

{strideutils-0.8.7 → strideutils-0.8.8}/strideutils.egg-info/requires.txt

@@ -14,6 +14,7 @@ python-dotenv==1.0.1
 pyyaml>=6.0.1
 requests>=2.32.3
 pycryptodome>=3.20.0
+web3==7.6.1
 
 [build]
 build>=1.2.2.post1