mm-eth 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

mm_eth/cli/rpc_helpers.py

@@ -0,0 +1,140 @@
+from loguru import logger
+from mm_std import Err, fatal
+
+from mm_eth import erc20, rpc
+from mm_eth.utils import from_wei_str
+
+from . import calcs
+
+
+def get_nonce(nodes: list[str] | str, address: str, log_prefix: str | None = None) -> int | None:
+    res = rpc.eth_get_transaction_count(nodes, address, attempts=5)
+    prefix = log_prefix or address
+    logger.debug(f"{prefix}: nonce={res.ok_or_err()}")
+    if isinstance(res, Err):
+        logger.info(f"{prefix}: nonce error, {res.err}")
+        return None
+    return res.ok
+
+
+def check_nodes_for_chain_id(nodes: list[str], chain_id: int) -> None:
+    for node in nodes:
+        res = rpc.eth_chain_id(node, timeout=7)
+        if isinstance(res, Err):
+            fatal(f"can't get chain_id for {node}, error={res.err}")
+        if res.ok != chain_id:
+            fatal(f"node {node} has a wrong chain_id: {res.ok}")
+
+
+def get_base_fee(nodes: list[str], log_prefix: str | None = None) -> int | None:
+    res = rpc.get_base_fee_per_gas(nodes)
+    prefix = _get_prefix(log_prefix)
+    logger.debug(f"{prefix}base_fee={res.ok_or_err()}")
+    if isinstance(res, Err):
+        logger.info(f"{prefix}base_fee error, {res.err}")
+        return None
+    return res.ok
+
+
+def calc_max_fee_per_gas(nodes: list[str], max_fee_per_gas: str, log_prefix: str | None = None) -> int | None:
+    if "base" in max_fee_per_gas.lower():
+        base_fee = get_base_fee(nodes, log_prefix)
+        if base_fee is None:
+            return None
+        return calcs.calc_var_wei_value(max_fee_per_gas, var_name="base", var_value=base_fee)
+    return calcs.calc_var_wei_value(max_fee_per_gas)
+
+
+def is_max_fee_per_gas_limit_exceeded(
+    max_fee_per_gas: int,
+    max_fee_per_gas_limit: str | None,
+    log_prefix: str | None = None,
+) -> bool:
+    if max_fee_per_gas_limit is None:
+        return False
+    max_fee_per_gas_limit_value = calcs.calc_var_wei_value(max_fee_per_gas_limit)
+    if max_fee_per_gas > max_fee_per_gas_limit_value:
+        prefix = _get_prefix(log_prefix)
+        logger.info(
+            "{}max_fee_per_gas_limit is exeeded, max_fee_per_gas={}, max_fee_per_gas_limit={}",
+            prefix,
+            from_wei_str(max_fee_per_gas, "gwei"),
+            from_wei_str(max_fee_per_gas_limit_value, "gwei"),
+        )
+        return True
+    return False
+
+
+def calc_gas(
+    *,
+    nodes: list[str],
+    gas: str,
+    from_address: str,
+    to_address: str,
+    value: int | None = None,
+    data: str | None = None,
+    log_prefix: str | None = None,
+) -> int | None:
+    estimate_value = None
+    if "estimate" in gas.lower():
+        prefix = _get_prefix(log_prefix)
+        res = rpc.eth_estimate_gas(nodes, from_address, to_address, data=data, value=value, attempts=5)
+        logger.debug(f"{prefix}gas_estimate={res.ok_or_err()}")
+        if isinstance(res, Err):
+            logger.info(f"{prefix}estimate_gas error, {res.err}")
+            return None
+        estimate_value = res.ok
+    return calcs.calc_var_wei_value(gas, var_name="estimate", var_value=estimate_value)
+
+
+def calc_eth_value(
+    *,
+    nodes: list[str],
+    value_str: str,
+    address: str,
+    gas: int | None = None,
+    max_fee_per_gas: int | None = None,
+    log_prefix: str | None = None,
+) -> int | None:
+    balance_value = None
+    if "balance" in value_str.lower():
+        prefix = _get_prefix(log_prefix)
+        res = rpc.eth_get_balance(nodes, address, attempts=5)
+        logger.debug(f"{prefix}balance={res.ok_or_err()}")
+        if isinstance(res, Err):
+            logger.info(f"{prefix}balance error, {res.err}")
+            return None
+        balance_value = res.ok
+    value = calcs.calc_var_wei_value(value_str, var_name="balance", var_value=balance_value)
+    if "balance" in value_str.lower() and gas is not None and max_fee_per_gas is not None:
+        value = value - gas * max_fee_per_gas
+    return value
+
+
+def calc_erc20_value(
+    *,
+    nodes: list[str],
+    value_str: str,
+    wallet_address: str,
+    token_address: str,
+    decimals: int,
+    log_prefix: str | None = None,
+) -> int | None:
+    value_str = value_str.lower()
+    balance_value = None
+    if "balance" in value_str:
+        prefix = _get_prefix(log_prefix)
+        res = erc20.get_balance(nodes, token_address, wallet_address, attempts=5)
+        logger.debug(f"{prefix}balance={res.ok_or_err()}")
+        if isinstance(res, Err):
+            logger.info(f"{prefix}balance error, {res.err}")
+            return None
+        balance_value = res.ok
+    return calcs.calc_var_wei_value(value_str, var_name="balance", var_value=balance_value, decimals=decimals)
+
+
+def _get_prefix(log_prefix: str | None) -> str:
+    prefix = log_prefix or ""
+    if prefix:
+        prefix += ": "
+    return prefix

mm_eth/cli/validators.py

@@ -0,0 +1,85 @@
+import os
+from decimal import Decimal, InvalidOperation
+from pathlib import Path
+
+from mm_std import str_to_list
+
+from mm_eth.utils import to_wei
+
+from . import calcs
+
+
+def wei_validator(v: str | None) -> int | None:
+    if v is None:
+        return None
+    return to_wei(v)
+
+
+def log_validator(v: str | None) -> str | None:
+    if v is None:
+        return None
+    log_file = Path(v).expanduser()
+    log_file.touch(exist_ok=True)
+    if not (log_file.is_file() and os.access(log_file, os.W_OK)):
+        raise ValueError(f"wrong log path: {v}")
+    return v
+
+
+def nodes_validator(v: str | list[str] | None) -> list[str]:
+    if v is None:
+        return []
+    if isinstance(v, str):
+        return str_to_list(v, unique=True, remove_comments=True, split_line=True)
+    return v
+
+
+def addresses_validator(v: str | list[str] | None) -> list[str]:
+    if v is None:
+        return []
+    if isinstance(v, str):
+        return str_to_list(v, unique=True, remove_comments=True, split_line=True, lower=True)
+    return v
+
+
+def delay_validator(v: str | Decimal) -> Decimal | tuple[Decimal, Decimal]:
+    if isinstance(v, int | float):
+        return Decimal(str(v))
+    if isinstance(v, str):
+        arr = [a.strip() for a in v.split("-")]
+        if len(arr) != 2:
+            raise ValueError("wrong delay value")
+        try:
+            return Decimal(arr[0]), Decimal(arr[1])
+        except InvalidOperation:
+            raise ValueError("wrong delay value") from None
+    raise ValueError("wrong delay value")
+
+
+def is_valid_calc_var_wei_value(value: str | None, base_name: str = "var", decimals: int | None = None) -> bool:
+    if value is None:
+        return True  # check for None on BaseModel.field type level
+    try:
+        calcs.calc_var_wei_value(value, var_value=123, var_name=base_name, decimals=decimals)
+        return True  # noqa: TRY300
+    except ValueError:
+        return False
+
+
+def is_valid_calc_decimal_value(value: str | None) -> bool:
+    if value is None:
+        return True  # check for None on BaseModel.field type level
+    try:
+        calcs.calc_decimal_value(value)
+        return True  # noqa: TRY300
+    except ValueError:
+        return False
+
+
+def is_valid_calc_function_args(value: str | None) -> bool:
+    if value is None:
+        return True
+    try:
+        calcs.calc_function_args(value)
+        return True  # noqa: TRY300
+    except ValueError:
+        return False

{mm_eth-0.2.1.dist-info → mm_eth-0.2.2.dist-info}/METADATA

@@ -1,7 +1,9 @@
 Metadata-Version: 2.4
 Name: mm-eth
-Version: 0.2.1
+Version: 0.2.2
 Requires-Python: >=3.12
-Requires-Dist: mm-std~=0.1.10
+Requires-Dist: loguru~=0.7.3
+Requires-Dist: mm-std~=0.1.11
+Requires-Dist: typer>=0.15.1
 Requires-Dist: web3~=7.7.0
 Requires-Dist: websocket-client~=1.8.0

mm_eth-0.2.2.dist-info/RECORD

@@ -0,0 +1,47 @@
+mm_eth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mm_eth/abi.py,sha256=Qf-QOsR9QexyQM9XWKNeTMkRarIL3XQJbaDbJ8ifMrw,4856
+mm_eth/account.py,sha256=IsajQ7156-CxqRFt2ntv_3nJKwi6tCZz0TmtjUCgmIg,2038
+mm_eth/anvil.py,sha256=98RCfI7dEpxFBTV6UErYvubWVP3n0ctUFn1--4kZ84U,1603
+mm_eth/deploy.py,sha256=SB3ruY808_5UnG8kHR34uVP66P3zOWZu0ImKD7UUv2s,691
+mm_eth/ens.py,sha256=WMxqC1v3zwDDuLH_oWekm22qrNYxCNcvZumQMT7SYds,623
+mm_eth/erc20.py,sha256=m0_yYzPYgOBx0oh4V330XGjRQBjMd5XlWhRL5qChvQE,6866
+mm_eth/ethernodes.py,sha256=9y_poTmFUj6cnWaT9mtfc6S9lAfVXTwLGRqxMQ8hT0Y,3080
+mm_eth/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mm_eth/rpc.py,sha256=F0jBzYw4wkh9LKwynXrzyhA08r5JTGrYAbwTH6C2Z7w,13872
+mm_eth/solc.py,sha256=dYRvT8PjZlLDZhNsc_-0790Eug_ZwU2G-iBfIdGj6wQ,1071
+mm_eth/tx.py,sha256=J5Wky8rmV6NlMFmAYF4997PscRYlUdkYFpEldH3BCNg,4014
+mm_eth/types.py,sha256=vXXP5Dc72BpHv5tsyws0KDZebG1W1-5HH0UjL7N2Mgc,113
+mm_eth/utils.py,sha256=NICgROKU5ZTrkY6nlSDLHjbbpYYswLoofyc3ozPdwBs,7804
+mm_eth/vault.py,sha256=h8NyiOQh5YFskh1lZA3KyvnJUnxl9769ME2ChplG0CM,1477
+mm_eth/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mm_eth/cli/calcs.py,sha256=yDx3VgFmCLRYWpZVqyk_-mXIgTVFDPKGTktx-7H2XDw,4426
+mm_eth/cli/cli.py,sha256=FfEgosWXa9gIGRuwSbPILvaDQ2Crh9hqTU50CqatvgY,8956
+mm_eth/cli/cli_utils.py,sha256=H8aasm4CBuKoXPA_ZGBTdTBs2ABAR7niFGqQjXOrYOU,3706
+mm_eth/cli/print_helpers.py,sha256=yOiOFjTKloumwf07AqNIHQswUo8t0yuT9bpeGBGl60Q,1470
+mm_eth/cli/rpc_helpers.py,sha256=tAJBUHwpH0jPvg0Hrm-cwpOrpp9GUb91e4ZdwixNUQg,4723
+mm_eth/cli/validators.py,sha256=CWOXorI2HGajhLvQ354d2cCl5S3JC7tKUxWmmVn6j4Y,2417
+mm_eth/cli/cmd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mm_eth/cli/cmd/balance_cmd.py,sha256=SjRSbGF8QFMBHncEyzPdVUi_rjOwNW2wxfbFsRBV2A0,2097
+mm_eth/cli/cmd/balances_cmd.py,sha256=XOjcyqJLSQL1Jj40fnw3la2ZRGwpZU-NIBpZbCEQNhc,4328
+mm_eth/cli/cmd/call_contract_cmd.py,sha256=lI1QnFTydYTFhZmxmQWvyMCnhTodckck9KGXUBq5PU4,1250
+mm_eth/cli/cmd/config_example_cmd.py,sha256=hMfrFxjIP0xZllJRRxi99qhF8tzJI3lrp-XudFEWF1g,270
+mm_eth/cli/cmd/deploy_cmd.py,sha256=543wR11I5GchYzTqWbUuyxcWMvZs0ZbcT0V_OLdwsew,1260
+mm_eth/cli/cmd/encode_input_data_cmd.py,sha256=9UQ1MKPEFQJ8j_COsP3KGKhwOf9tT3feBezI8vvxTlw,267
+mm_eth/cli/cmd/mnemonic_cmd.py,sha256=Mb0H0inSNCa93GEPxGYQi7BnPe11mnLjnspfe7h54I4,972
+mm_eth/cli/cmd/node_cmd.py,sha256=mUqixPGNHzuKCJvk1Fd5VaUGumpbR2AE7raGmXZscg4,1943
+mm_eth/cli/cmd/private_key_cmd.py,sha256=Fv_2OLog1h32pIP7PJITwl_pHdy3BXvaDRcXZsxY1xo,241
+mm_eth/cli/cmd/rpc_cmd.py,sha256=02q82YqgbPezfEBmV_QBCIeNReE7ktkPych8Xr9ann8,2186
+mm_eth/cli/cmd/send_contract_cmd.py,sha256=fknRzpCKV4nWMus26npeCVCb2JfiHTsSMFGPnP-465o,9141
+mm_eth/cli/cmd/solc_cmd.py,sha256=tBpeMdPfGs2iQIMaIJAAhMh1a3KyXHwyninfXPVpsgs,677
+mm_eth/cli/cmd/token_cmd.py,sha256=4y6ZQpLOJ33_iNuKpm9tZXh4RntWhmPUcizgaNNBzaw,1102
+mm_eth/cli/cmd/transfer_erc20_cmd.py,sha256=EXOy303fKd3cuC8QEVghAjJw0qrLoO4GFfT37whMt2E,10296
+mm_eth/cli/cmd/transfer_eth_cmd.py,sha256=rBFZzDntXX0ouGb1B15rRaEiEoF88C-It2aiJhSVfzQ,9470
+mm_eth/cli/cmd/vault_cmd.py,sha256=rRTclz0U6N_87KFsssdvZhi3_f6tmkHiYxMsIoVeBGc,532
+mm_eth/cli/config_examples/balances.yml,sha256=fpXJfoOSqOrkoWpqO7-HrTTb5OBs9PM-PpZ1ulqWUnY,399
+mm_eth/cli/config_examples/call_contract.yml,sha256=E0XuWuBnbhyTYfxNqaoxH6Cy7UCYqMLDwoEu_4OSV3M,233
+mm_eth/cli/config_examples/transfer_erc20.yml,sha256=mCUpUfqzZjTvst8kqd3EGzJY0S7cr48ySqS6yZhzLdI,1324
+mm_eth/cli/config_examples/transfer_eth.yml,sha256=i4hh3-LTcpl7JsrSuuM6gDi23B5ZnCeEnaYazQO3bzQ,1181
+mm_eth-0.2.2.dist-info/METADATA,sha256=PUaC1zS4IBlq6bwzjhGXqykcW2hyD9sQ0PKwTHM2DQY,228
+mm_eth-0.2.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mm_eth-0.2.2.dist-info/entry_points.txt,sha256=aGhpsozl8NIrkuUcX5fSgURCcDhr3ShUdeTSIrJq4oc,46
+mm_eth-0.2.2.dist-info/RECORD,,

mm_eth-0.2.2.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+mm-eth = mm_eth.cli.cli:app

mm_eth-0.2.1.dist-info/RECORD

@@ -1,18 +0,0 @@
-mm_eth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mm_eth/abi.py,sha256=Qf-QOsR9QexyQM9XWKNeTMkRarIL3XQJbaDbJ8ifMrw,4856
-mm_eth/account.py,sha256=IsajQ7156-CxqRFt2ntv_3nJKwi6tCZz0TmtjUCgmIg,2038
-mm_eth/anvil.py,sha256=98RCfI7dEpxFBTV6UErYvubWVP3n0ctUFn1--4kZ84U,1603
-mm_eth/deploy.py,sha256=SB3ruY808_5UnG8kHR34uVP66P3zOWZu0ImKD7UUv2s,691
-mm_eth/ens.py,sha256=WMxqC1v3zwDDuLH_oWekm22qrNYxCNcvZumQMT7SYds,623
-mm_eth/erc20.py,sha256=m0_yYzPYgOBx0oh4V330XGjRQBjMd5XlWhRL5qChvQE,6866
-mm_eth/ethernodes.py,sha256=9y_poTmFUj6cnWaT9mtfc6S9lAfVXTwLGRqxMQ8hT0Y,3080
-mm_eth/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mm_eth/rpc.py,sha256=F0jBzYw4wkh9LKwynXrzyhA08r5JTGrYAbwTH6C2Z7w,13872
-mm_eth/solc.py,sha256=dYRvT8PjZlLDZhNsc_-0790Eug_ZwU2G-iBfIdGj6wQ,1071
-mm_eth/tx.py,sha256=J5Wky8rmV6NlMFmAYF4997PscRYlUdkYFpEldH3BCNg,4014
-mm_eth/types.py,sha256=vXXP5Dc72BpHv5tsyws0KDZebG1W1-5HH0UjL7N2Mgc,113
-mm_eth/utils.py,sha256=NICgROKU5ZTrkY6nlSDLHjbbpYYswLoofyc3ozPdwBs,7804
-mm_eth/vault.py,sha256=h8NyiOQh5YFskh1lZA3KyvnJUnxl9769ME2ChplG0CM,1477
-mm_eth-0.2.1.dist-info/METADATA,sha256=VZ2A_K7tjKAWxIzkq6K3zk3PQv6hXDRs89oBvGgApB0,170
-mm_eth-0.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mm_eth-0.2.1.dist-info/RECORD,,