mm-sol 0.5.9__py3-none-any.whl → 0.6.1__py3-none-any.whl

mm_sol/solana_cli.py

@@ -1,252 +0,0 @@
-import json
-import random
-from decimal import Decimal
-from pathlib import Path
-from typing import Literal
-
-import pydash
-from mm_std import CommandResult, Err, Ok, Result, run_command, run_ssh_command
-from pydantic import BaseModel, ConfigDict, Field, field_validator
-
-
-class ValidatorInfo(BaseModel):
-    identity_address: str
-    info_address: str
-    name: str | None
-    keybase: str | None
-    website: str | None
-    details: str | None
-
-
-class StakeAccount(BaseModel):
-    type: str = Field(..., alias="stakeType")
-    balance: float | None = Field(..., alias="accountBalance")
-    withdrawer: str
-    staker: str
-    vote: str | None = Field(None, alias="delegatedVoteAccountAddress")
-
-    @field_validator("balance")
-    def from_lamports_to_sol(cls, v: int | None) -> float | None:
-        if v:
-            return v / 1_000_000_000
-
-
-class Stake(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-
-    stake_address: str = Field(..., alias="stakePubkey")
-    withdrawer_address: str = Field(..., alias="withdrawer")
-    vote_address: str | None = Field(None, alias="delegatedVoteAccountAddress")
-    balance: float | None = Field(..., alias="accountBalance")
-    delegated: float | None = Field(None, alias="delegatedStake")
-    active: float | None = Field(None, alias="activeStake")
-    lock_time: int | None = Field(None, alias="unixTimestamp")
-
-    @field_validator("balance", "delegated", "active")
-    def from_lamports_to_sol(cls, v: int | None) -> float | None:
-        if v:
-            return v / 1_000_000_000
-
-
-def get_balance(
-    *,
-    address: str,
-    solana_dir: str = "",
-    url: str = "localhost",
-    ssh_host: str | None = None,
-    ssh_key_path: str | None = None,
-    timeout: int = 60,
-) -> Result[Decimal]:
-    solana_dir = _solana_dir(solana_dir)
-    cmd = f"{solana_dir}solana balance {address} -u {url}"
-    res = _exec_cmd(cmd, ssh_host, ssh_key_path, timeout)
-    data = {"cmd": cmd, "stdout": res.stdout, "stderr": res.stderr}
-    try:
-        return Ok(Decimal(res.stdout.replace("SOL", "").strip()), data=data)
-    except Exception as e:
-        return Err(e, data=data)
-
-
-def get_stake_account(
-    *,
-    address: str,
-    solana_dir: str = "",
-    url: str = "localhost",
-    ssh_host: str | None = None,
-    ssh_key_path: str | None = None,
-    timeout: int = 60,
-) -> Result[StakeAccount]:
-    solana_dir = _solana_dir(solana_dir)
-    cmd = f"{solana_dir}solana stake-account --output json -u {url} {address}"
-    res = _exec_cmd(cmd, ssh_host, ssh_key_path, timeout)
-    data = {"cmd": cmd, "stdout": res.stdout, "stderr": res.stderr}
-    try:
-        json_res = json.loads(res.stdout)
-        return Ok(StakeAccount(**json_res), data=data)
-    except Exception as e:
-        return Err(e, data=data)
-
-
-def transfer_with_private_key_file(
-    *,
-    recipient: str,
-    amount: Decimal,
-    private_key_path: str,
-    solana_dir: str = "",
-    url: str = "localhost",
-    ssh_host: str | None = None,
-    ssh_key_path: str | None = None,
-    allow_unfunded_recipient: bool = True,
-    timeout: int = 60,
-) -> Result[str]:
-    solana_dir = _solana_dir(solana_dir)
-    cmd = f"{solana_dir}solana transfer {recipient} {amount} --from {private_key_path} --fee-payer {private_key_path}"
-    if allow_unfunded_recipient:
-        cmd += " --allow-unfunded-recipient"
-    cmd += f" -u {url} --output json"
-    res = _exec_cmd(cmd, ssh_host, ssh_key_path, timeout)
-    data = {"cmd": cmd, "stdout": res.stdout, "stderr": res.stderr}
-    try:
-        json_res = json.loads(res.stdout)
-        return Ok(json_res["signature"], data=data)
-    except Exception as e:
-        return Err(e, data=data)
-
-
-def transfer_with_private_key_str(
-    *,
-    recipient: str,
-    amount: Decimal,
-    private_key: str,
-    tmp_dir_path: str,
-    solana_dir: str = "",
-    url: str = "localhost",
-    ssh_host: str | None = None,
-    ssh_key_path: str | None = None,
-    timeout: int = 60,
-) -> Result[str]:
-    # make private_key file
-    private_key_path = Path(f"{tmp_dir_path}/solana__{random.randint(1, 10_000_000_000)}.json")
-    private_key_path.write_text(private_key)
-
-    try:
-        return transfer_with_private_key_file(
-            recipient=recipient,
-            amount=amount,
-            private_key_path=private_key_path.as_posix(),
-            solana_dir=solana_dir,
-            url=url,
-            ssh_host=ssh_host,
-            ssh_key_path=ssh_key_path,
-            timeout=timeout,
-        )
-    finally:
-        private_key_path.unlink()
-
-
-def withdraw_from_vote_account(
-    *,
-    recipient: str,
-    amount: Decimal | Literal["ALL"],
-    vote_key_path: str,
-    fee_payer_key_path: str,
-    solana_dir: str = "",
-    url: str = "localhost",
-    ssh_host: str | None = None,
-    ssh_key_path: str | None = None,
-    timeout: int = 60,
-) -> Result[str]:
-    solana_dir = _solana_dir(solana_dir)
-    cmd = f"{solana_dir}solana withdraw-from-vote-account --keypair {fee_payer_key_path} -u {url} --output json {vote_key_path} {recipient} {amount}"  # noqa: E501
-    res = _exec_cmd(cmd, ssh_host, ssh_key_path, timeout)
-    data = {"cmd": cmd, "stdout": res.stdout, "stderr": res.stderr}
-    try:
-        json_res = json.loads(res.stdout)
-        return Ok(json_res["signature"], data=data)
-    except Exception as e:
-        return Err(e, data=data)
-
-
-def get_validators_info(
-    *,
-    solana_dir: str = "",
-    url: str = "localhost",
-    ssh_host: str | None = None,
-    ssh_key_path: str | None = None,
-    timeout: int = 60,
-) -> Result[list[ValidatorInfo]]:
-    solana_dir = _solana_dir(solana_dir)
-    cmd = f"{solana_dir}solana validator-info get --output json -u {url}"
-    res = _exec_cmd(cmd, ssh_host, ssh_key_path, timeout)
-    data = {"cmd": cmd, "stdout": res.stdout, "stderr": res.stderr}
-    try:
-        validators = []
-        for v in json.loads(res.stdout):
-            validators.append(  # noqa: PERF401
-                ValidatorInfo(
-                    info_address=v["infoPubkey"],
-                    identity_address=v["identityPubkey"],
-                    name=pydash.get(v, "info.name"),
-                    keybase=pydash.get(v, "info.keybaseUsername"),
-                    details=pydash.get(v, "info.details"),
-                    website=pydash.get(v, "info.website"),
-                ),
-            )
-        return Ok(validators, data=data)
-    except Exception as e:
-        return Err(e, data=data)
-
-
-def get_vote_account_rewards(
-    *,
-    address: str,
-    solana_dir: str = "",
-    url: str = "localhost",
-    ssh_host: str | None = None,
-    ssh_key_path: str | None = None,
-    num_rewards_epochs: int = 10,
-    timeout: int = 60,
-) -> Result[dict[int, float]]:
-    solana_dir = _solana_dir(solana_dir)
-    cmd = f"{solana_dir}solana vote-account {address} --with-rewards --num-rewards-epochs={num_rewards_epochs} -u {url}"
-    cmd += " --output json 2>/dev/null"
-    res = _exec_cmd(cmd, ssh_host, ssh_key_path, timeout)
-    data = {"cmd": cmd, "stdout": res.stdout, "stderr": res.stderr}
-    try:
-        rewards: dict[int, float] = {}
-        for r in reversed(json.loads(res.stdout)["epochRewards"]):
-            rewards[r["epoch"]] = r["amount"] / 10**9
-        return Ok(rewards, data=data)
-    except Exception as e:
-        return Err(e, data=data)
-
-
-def get_stakes(
-    *,
-    vote_address: str = "",
-    solana_dir: str = "",
-    url: str = "localhost",
-    ssh_host: str | None = None,
-    ssh_key_path: str | None = None,
-    timeout: int = 60,
-) -> Result[list[Stake]]:
-    solana_dir = _solana_dir(solana_dir)
-    cmd = f"{solana_dir}solana stakes --output json -u {url} {vote_address}"
-    res = _exec_cmd(cmd, ssh_host, ssh_key_path, timeout)
-    data = {"stdout": res.stdout, "stderr": res.stderr}
-    try:
-        return Ok([Stake(**x) for x in json.loads(res.stdout)], data=data)
-    except Exception as e:
-        return Err(e, data=data)
-
-
-def _exec_cmd(cmd: str, ssh_host: str | None, ssh_key_path: str | None, timeout: int) -> CommandResult:
-    if ssh_host:
-        return run_ssh_command(ssh_host, cmd, ssh_key_path, timeout=timeout)
-    return run_command(cmd, timeout=timeout)
-
-
-def _solana_dir(solana_dir: str) -> str:
-    if solana_dir and not solana_dir.endswith("/"):
-        solana_dir += "/"
-    return solana_dir

mm_sol/token.py

@@ -1,33 +0,0 @@
-import mm_crypto_utils
-from mm_crypto_utils import Nodes, Proxies
-from mm_std import Err, Ok, Result
-from solders.pubkey import Pubkey
-
-from mm_sol.utils import get_client
-
-
-def get_decimals(node: str, token_mint_address: str, timeout: float = 10, proxy: str | None = None) -> Result[int]:
-    data = None
-    try:
-        client = get_client(node, proxy=proxy, timeout=timeout)
-        res = client.get_token_supply(Pubkey.from_string(token_mint_address))
-        data = res
-        return Ok(res.value.decimals)
-    except Exception as e:
-        return Err(e, data=data)
-
-
-def get_decimals_with_retries(
-    nodes: Nodes, token_mint_address: str, retries: int, timeout: float = 10, proxies: Proxies = None
-) -> Result[int]:
-    res: Result[int] = Err("not started yet")
-    for _ in range(retries):
-        res = get_decimals(
-            node=mm_crypto_utils.random_node(nodes),
-            token_mint_address=token_mint_address,
-            timeout=timeout,
-            proxy=mm_crypto_utils.random_proxy(proxies),
-        )
-        if res.is_ok():
-            return res
-    return res

mm_sol/token_async.py

@@ -1,40 +0,0 @@
-import httpx
-from mm_std import DataResult
-from solana.exceptions import SolanaRpcException
-from solana.rpc.core import RPCException
-from solders.solders import InvalidParamsMessage, Pubkey, get_associated_token_address
-
-from mm_sol.utils import get_async_client
-
-
-async def get_token_balance(
-    node: str,
-    owner_address: str,
-    token_mint_address: str,
-    token_account: str | None = None,
-    timeout: float = 10,
-    proxy: str | None = None,
-) -> DataResult[int]:
-    try:
-        client = get_async_client(node, proxy=proxy, timeout=timeout)
-        if not token_account:
-            token_account = str(
-                get_associated_token_address(Pubkey.from_string(owner_address), Pubkey.from_string(token_mint_address))
-            )
-
-        res = await client.get_token_account_balance(Pubkey.from_string(token_account))
-
-        # Sometimes it not raise an error, but it returns this :(
-        if isinstance(res, InvalidParamsMessage) and "could not find account" in res.message:
-            return DataResult.ok(0, {"res": res.to_json()})
-        return DataResult.ok(int(res.value.amount), {"res": res.to_json()})
-    except RPCException as err:
-        if "could not find account" in str(err):
-            return DataResult.ok(0, {"rpc_exception": str(err)})
-        return DataResult.exception(err)
-    except httpx.HTTPStatusError as err:
-        return DataResult.err(f"http error: {err}")
-    except SolanaRpcException as err:
-        return DataResult.err(err.error_msg)
-    except Exception as err:
-        return DataResult.exception(err)

mm_sol-0.5.9.dist-info/RECORD

@@ -1,35 +0,0 @@
-mm_sol/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mm_sol/account.py,sha256=9aDP5si-QJfSF8fz8K6geGUGh9GL83YJNkdU1qd4Xu8,4515
-mm_sol/async_rpc.py,sha256=3_pEu4ywBjA_ej6cX5q4WJcN1oaPDeIAFa5xNceFCGw,1406
-mm_sol/balance.py,sha256=5sSIQbipNUeB_Hn7piwrYt-PuSEf4ZKB8puUmC986iY,5112
-mm_sol/block.py,sha256=4Lc4TANgpGvPflVumC9MR-3vIl1dedGyci3cgzczuds,1794
-mm_sol/constants.py,sha256=WSpfz5_cq_8XbIrNFJGu9okwbfPTL00zsyR_k9-7O0o,29
-mm_sol/converters.py,sha256=rBxe3SIADZS8hG7TYl4FgjmvKH-ykaTmNbnWWQDiFZ4,1430
-mm_sol/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mm_sol/rpc.py,sha256=Tw7THbU0lShjTGNGe4-Mc6q1U5E2dgjY2LIfrMTSfeA,8057
-mm_sol/rpc_async.py,sha256=_cKqyd5oLArm9VWWuYNqq2PbkqEu7KrsH2r4IAEevGY,2476
-mm_sol/solana_cli.py,sha256=ig3OoTvmkrl7MFQSZjRHIraLSmtse0_9kn5Nsw8_zb0,8258
-mm_sol/token.py,sha256=O8z3UE3iZGYLWw8fnd9weYMcoQO0m88noqbRO_jntGg,1092
-mm_sol/token_async.py,sha256=6IWlQOnyTWA2e3_bccxa10Y4BhbgQpXUAS1NrTyw67E,1572
-mm_sol/transfer.py,sha256=taf2NTLpo-bxISeaILARXEGLldUvqvP-agp5IDva7Hw,5825
-mm_sol/utils.py,sha256=oD06NsMSMhN6lqsM6mSgLTtiKwA1uAsen9WR82ofRTE,923
-mm_sol/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mm_sol/cli/calcs.py,sha256=-r9RlsQyOziTDf84uIsvTgZmsUdNrVeazu3vTj9hhNA,1887
-mm_sol/cli/cli.py,sha256=a28Cy1X2bRSIAAJu2HHoPoTJYOPvff-anVcL0BJnwRI,4610
-mm_sol/cli/cli_utils.py,sha256=nFdY8tJFZxyssEBEFCc3VTNJt447e6vMnugx4GBPL4o,1840
-mm_sol/cli/validators.py,sha256=M_Rr7JoG3TUYTDAGkjQLDH6l9i9FOrSpss30KdY3UlM,1379
-mm_sol/cli/cmd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mm_sol/cli/cmd/balance_cmd.py,sha256=DfUZY3-Hr-F7Y0xp1faol0yLnPu6iDU3b839VhdwAbw,2587
-mm_sol/cli/cmd/balances_cmd.py,sha256=gMRZXAnCOPESsKRqz33ALPzSOqGGeZF225kMRO6sV-c,2876
-mm_sol/cli/cmd/example_cmd.py,sha256=ZLTy1-cmapiCyYvjFInVE-pQCGKZzDgYKUhsOwtbSIY,234
-mm_sol/cli/cmd/node_cmd.py,sha256=-T3AvUOHPKEQWET7gyDVtwNT24SFX0_9rsgfSjnM_Cw,331
-mm_sol/cli/cmd/transfer_cmd.py,sha256=eQzD7LispR4KFMUlXdXMBGbXs7b0A-iHJHs7VkxH1UA,11021
-mm_sol/cli/cmd/wallet/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mm_sol/cli/cmd/wallet/keypair_cmd.py,sha256=cRHVVTs9zNYmUozZ8ZlJoutn9V6r8I1AEHBrszR7dTE,538
-mm_sol/cli/cmd/wallet/mnemonic_cmd.py,sha256=IiON_fJT5AFfIr_E1LR6_iDYZ3c_jWCFc-wSYqk61V8,648
-mm_sol/cli/examples/balances.toml,sha256=333g2EkyYBDW7OWFGMIWVZGkdFQMMo0Ag-bg-BvS4Zg,349
-mm_sol/cli/examples/transfer.toml,sha256=kOCdmuwmhlOam4LVtlcYTKF0PoZYHWMlv9gWxNSXMOk,1624
-mm_sol-0.5.9.dist-info/METADATA,sha256=rPkIVLM-UIwT_o9HLo8E_EqputbOZ44gfIGxw8_rstA,321
-mm_sol-0.5.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mm_sol-0.5.9.dist-info/entry_points.txt,sha256=MrYnosumy9nsITSAw5TiR3WXDwsdoF0YvUIlZ38TLLs,46
-mm_sol-0.5.9.dist-info/RECORD,,