mm-eth 0.5.6__py3-none-any.whl → 0.5.8__py3-none-any.whl

mm_eth/async_rpc.py

@@ -0,0 +1,93 @@
+import json
+from typing import Any
+
+import websockets
+from mm_crypto_utils import Nodes, Proxies, random_node, random_proxy
+from mm_std import Err, Ok, Result, ahr
+
+from mm_eth.utils import hex_str_to_int
+
+
+async def rpc_call(
+    *,
+    nodes: Nodes,
+    method: str,
+    params: list[object],
+    id_: int = 1,
+    timeout: int = 10,
+    proxies: Proxies = None,
+    attempts: int = 1,
+) -> Result[Any]:
+    data = {"jsonrpc": "2.0", "method": method, "params": params, "id": id_}
+    res: Result[Any] = Err("not started yet")
+    for _ in range(attempts):
+        node = random_node(nodes)
+        res = (
+            await _http_call(node, data, timeout, random_proxy(proxies))
+            if node.startswith("http")
+            else await _ws_call(node, data, timeout)
+        )
+        if isinstance(res, Ok):
+            return res
+    return res
+
+
+async def _http_call(node: str, data: dict[str, object], timeout: int, proxy: str | None) -> Result[Any]:
+    res = await ahr(node, method="POST", proxy=proxy, timeout=timeout, params=data, json_params=True)
+    if res.is_error():
+        return res.to_err_result()
+    try:
+        err = res.json.get("error", {}).get("message", "")
+        if err:
+            return res.to_err_result(f"service_error: {err}")
+        if "result" in res.json:
+            return res.to_ok_result(res.json["result"])
+        return res.to_err_result("unknown_response")
+    except Exception as err:
+        return res.to_err_result(f"exception: {err}")
+
+
+async def _ws_call(node: str, data: dict[str, object], timeout: int) -> Result[Any]:
+    try:
+        async with websockets.connect(node, timeout=timeout) as ws:
+            await ws.send(json.dumps(data))
+            response = json.loads(await ws.recv())
+
+        err = response.get("error", {}).get("message", "")
+        if err:
+            return Err(f"service_error: {err}")
+        if "result" in response:
+            return Ok(response["result"], response)
+        return Err(f"unknown_response: {response}")
+    except TimeoutError:
+        return Err("timeout")
+    except Exception as err:
+        return Err(f"exception: {err}")
+
+
+async def eth_block_number(rpc_urls: Nodes, timeout: int = 10, proxies: Proxies = None, attempts: int = 1) -> Result[int]:
+    return (
+        await rpc_call(
+            nodes=rpc_urls,
+            method="eth_blockNumber",
+            params=[],
+            timeout=timeout,
+            proxies=proxies,
+            attempts=attempts,
+        )
+    ).and_then(hex_str_to_int)
+
+
+async def eth_get_balance(
+    rpc_urls: Nodes, address: str, timeout: int = 10, proxies: Proxies = None, attempts: int = 1
+) -> Result[int]:
+    return (
+        await rpc_call(
+            nodes=rpc_urls,
+            method="eth_getBalance",
+            params=[address, "latest"],
+            timeout=timeout,
+            proxies=proxies,
+            attempts=attempts,
+        )
+    ).and_then(hex_str_to_int)

mm_eth/ens.py

@@ -1,7 +1,7 @@
 from mm_crypto_utils import Nodes, Proxies, random_node, random_proxy
 from mm_std import Err, Ok, Result
 
-from mm_eth.utils import get_w3
+from mm_eth.utils import get_async_w3, get_w3
 
 
 def get_name_with_retries(
@@ -21,3 +21,24 @@ def get_name(rpc_url: str, address: str, timeout: float = 5, proxy: str | None =
         return Ok(w3.ens.name(w3.to_checksum_address(address)))  # type: ignore[union-attr]
     except Exception as e:
         return Err(e)
+
+
+async def async_get_name(rpc_url: str, address: str, timeout: float = 5, proxy: str | None = None) -> Result[str | None]:
+    try:
+        w3 = await get_async_w3(rpc_url, timeout=timeout, proxy=proxy)
+        res = await w3.ens.name(w3.to_checksum_address(address))  # type: ignore[union-attr]
+        await w3.provider.disconnect()
+        return Ok(res)
+    except Exception as e:
+        return Err(e)
+
+
+async def async_get_name_with_retries(
+    rpc_urls: Nodes, address: str, retries: int, timeout: float = 5, proxies: Proxies = None
+) -> Result[str | None]:
+    res: Result[str | None] = Err("not started yet")
+    for _ in range(retries):
+        res = await async_get_name(random_node(rpc_urls), address, timeout=timeout, proxy=random_proxy(proxies))
+        if res.is_ok():
+            return res
+    return res

mm_eth/erc20.py

@@ -11,7 +11,7 @@ from eth_utils import to_checksum_address, to_hex
 from mm_crypto_utils import Nodes, Proxies
 from mm_std import Err, Ok, Result
 
-from mm_eth import rpc
+from mm_eth import async_rpc, rpc
 from mm_eth.rpc import Log
 from mm_eth.tx import SignedTx, sign_legacy_tx, sign_tx
 from mm_eth.utils import hex_str_to_int, hex_to_bytes, log_topic_to_address
@@ -70,6 +70,27 @@ def get_balance(
     ).and_then(hex_str_to_int)
 
 
+async def async_get_balance(
+    rpc_urls: Nodes,
+    token_address: str,
+    user_address: str,
+    timeout: int = 10,
+    proxies: Proxies = None,
+    attempts: int = 1,
+) -> Result[int]:
+    data = "0x70a08231000000000000000000000000" + user_address[2:]
+    return (
+        await async_rpc.rpc_call(
+            nodes=rpc_urls,
+            method="eth_call",
+            params=[{"to": token_address, "data": data}, "latest"],
+            timeout=timeout,
+            proxies=proxies,
+            attempts=attempts,
+        )
+    ).and_then(hex_str_to_int)
+
+
 def get_name(rpc_urls: Nodes, address: str, timeout: int = 10, proxies: Proxies = None, attempts: int = 1) -> Result[str]:
     return rpc.rpc_call(
         nodes=rpc_urls,

mm_eth/ethernodes.py

@@ -1,6 +1,6 @@
 from datetime import datetime
 
-from mm_std import CHROME_USER_AGENT, Result, hr
+from mm_std import CHROME_USER_AGENT, Ok, Result, hr
 from pydantic import BaseModel, Field
 
 
@@ -29,6 +29,6 @@ def search_nodes(offset: int = 0, proxy: str | None = None) -> Result[SearchResu
     if res.is_error():
         return res.to_err_result()
     try:
-        return res.to_ok_result(SearchResult(**res.json))
+        return Ok(SearchResult(**res.json))
     except Exception as e:
         return res.to_err_result(f"exception: {e}")

mm_eth/utils.py

@@ -2,13 +2,15 @@ import re
 from decimal import Decimal, localcontext
 from typing import Any, cast
 
+import aiohttp
 import eth_utils
 import pydash
+from aiohttp_socks import ProxyConnector
 from eth_typing import HexStr
 from hexbytes import HexBytes
 from mm_std import Err, Ok, Result, number_with_separator
 from pydantic import BaseModel
-from web3 import Web3
+from web3 import AsyncWeb3, Web3
 from web3.types import Wei
 
 
@@ -193,6 +195,20 @@ def get_w3(rpc_url: str, timeout: float | None = None, proxy: str | None = None)
     return Web3(Web3.HTTPProvider(rpc_url, request_kwargs=request_kwargs))
 
 
+async def get_async_w3(rpc_url: str, timeout: float | None = None, proxy: str | None = None) -> AsyncWeb3:
+    request_kwargs: dict[str, object] = {"timeout": timeout}
+    if proxy and proxy.startswith("http"):
+        request_kwargs["proxy"] = proxy
+    provider = AsyncWeb3.AsyncHTTPProvider(rpc_url, request_kwargs=request_kwargs, exception_retry_configuration=None)
+    w3 = AsyncWeb3(provider)
+
+    if proxy and proxy.startswith("socks"):
+        session = aiohttp.ClientSession(connector=ProxyConnector.from_url(proxy))
+        await provider.cache_async_session(session)
+
+    return w3
+
+
 def name_network(chain_id: int) -> str:
     match chain_id:
         case 1:

{mm_eth-0.5.6.dist-info → mm_eth-0.5.8.dist-info}/METADATA

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: mm-eth
-Version: 0.5.6
+Version: 0.5.8
 Requires-Python: >=3.12
-Requires-Dist: mm-crypto-utils>=0.2.6
-Requires-Dist: requests[socks]>=2.32.0
+Requires-Dist: aiohttp-socks~=0.10.1
+Requires-Dist: mm-crypto-utils>=0.2.9
 Requires-Dist: typer>=0.15.2
-Requires-Dist: web3~=7.8.0
+Requires-Dist: web3~=7.9.0
 Requires-Dist: websocket-client~=1.8.0

{mm_eth-0.5.6.dist-info → mm_eth-0.5.8.dist-info}/RECORD

@@ -2,17 +2,18 @@ mm_eth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mm_eth/abi.py,sha256=Qf-QOsR9QexyQM9XWKNeTMkRarIL3XQJbaDbJ8ifMrw,4856
 mm_eth/account.py,sha256=k0MNMatBe0zo1iKZiB_Tq6zyEIo22IncD6ewNUJp3dY,2075
 mm_eth/anvil.py,sha256=98RCfI7dEpxFBTV6UErYvubWVP3n0ctUFn1--4kZ84U,1603
+mm_eth/async_rpc.py,sha256=8rMtg3Hr6Pn7NaFROxxWsXQSM8wJ92EUYlOOoc6_TOk,2934
 mm_eth/constants.py,sha256=Cy_G-IleBH4gAZ4ok8AGHHlqmdW0ZM7ZldyVpzAfWLs,54
 mm_eth/deploy.py,sha256=SB3ruY808_5UnG8kHR34uVP66P3zOWZu0ImKD7UUv2s,691
-mm_eth/ens.py,sha256=4sEglXdgemKm9fYLCXEvsEeUpdUCqec1GXPklyp19eQ,851
-mm_eth/erc20.py,sha256=V__yPDG4ABXWfkSdIqffE5qQDjjZl93287YunllN52A,6804
-mm_eth/ethernodes.py,sha256=9y_poTmFUj6cnWaT9mtfc6S9lAfVXTwLGRqxMQ8hT0Y,3080
+mm_eth/ens.py,sha256=amMs4xbBia1hVHKglKwqeheL_8NpukKNAZIsg65wpDA,1689
+mm_eth/erc20.py,sha256=LZmTkjmXpN7sUhOtSLiTqb8Ubet9C9q32EQwTS78OqE,7378
+mm_eth/ethernodes.py,sha256=V4VVbC6Nr9jhwT7blxtLugXC5KfXqE8n-kP0VvGHbqo,3070
 mm_eth/json_encoder.py,sha256=S4oD-qfTVztMb4sRpY1puhBQwOBofTyQXWszmdXk4og,433
 mm_eth/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mm_eth/rpc.py,sha256=k0eHxo_Dp6G0fHQ_rD-QbwOJz5ngC6kxBjl5NEHnyw8,13832
 mm_eth/solc.py,sha256=dYRvT8PjZlLDZhNsc_-0790Eug_ZwU2G-iBfIdGj6wQ,1071
 mm_eth/tx.py,sha256=efSoMCoWkenbGdHo1_LX66_Edz1HvED5-J_i3wrHwMw,4051
-mm_eth/utils.py,sha256=gLWT5I7CxFvfHgTQrZSSHUk3YyaDzdmtHwLHFjwuV3I,7407
+mm_eth/utils.py,sha256=bkh4EBaxTYEnrR1s7zQ2hzZCwXw1r-2HmfN-dpTXwRc,8069
 mm_eth/vault.py,sha256=h8NyiOQh5YFskh1lZA3KyvnJUnxl9769ME2ChplG0CM,1477
 mm_eth/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mm_eth/cli/calcs.py,sha256=cLFTYNAN-I53tUiSg-zFfVr2afjIZPftDDjHj16FBz0,1068
@@ -41,7 +42,7 @@ mm_eth/cli/cmd/wallet/private_key_cmd.py,sha256=Fv_2OLog1h32pIP7PJITwl_pHdy3BXva
 mm_eth/cli/examples/balances.toml,sha256=i_ALpiEcf8-0TFiUg1cgJhxxfHYeBl9x0b3tnUWjswU,421
 mm_eth/cli/examples/call_contract.toml,sha256=ZQWK-409V_vLIZ2bsRD5RCWPPzShPz2KJTTRQY4YaGw,248
 mm_eth/cli/examples/transfer.toml,sha256=8mWuphDquoSDJw-hb9VJqtonjmv3kJ5Ip8jJ4t5YnfM,1810
-mm_eth-0.5.6.dist-info/METADATA,sha256=R49ouAdttwFUlqby9en4HmcJsu1oxjWqp21vKKn9l_s,246
-mm_eth-0.5.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mm_eth-0.5.6.dist-info/entry_points.txt,sha256=aGhpsozl8NIrkuUcX5fSgURCcDhr3ShUdeTSIrJq4oc,46
-mm_eth-0.5.6.dist-info/RECORD,,
+mm_eth-0.5.8.dist-info/METADATA,sha256=8YvcBQc-8hK5tLsYKGyxmUtfLl9o8WeEXtn4verNNFY,244
+mm_eth-0.5.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mm_eth-0.5.8.dist-info/entry_points.txt,sha256=aGhpsozl8NIrkuUcX5fSgURCcDhr3ShUdeTSIrJq4oc,46
+mm_eth-0.5.8.dist-info/RECORD,,