mm-balance 0.1.12__tar.gz → 0.1.14__tar.gz

{mm_balance-0.1.12 → mm_balance-0.1.14}/PKG-INFO

@@ -1,8 +1,9 @@
 Metadata-Version: 2.3
 Name: mm-balance
-Version: 0.1.12
+Version: 0.1.14
 Requires-Python: >=3.12
+Requires-Dist: mm-aptos==0.1.2
 Requires-Dist: mm-btc==0.1.0
 Requires-Dist: mm-eth==0.1.3
-Requires-Dist: mm-solana==0.1.4
+Requires-Dist: mm-solana==0.1.5
 Requires-Dist: typer>=0.12.5

{mm_balance-0.1.12 → mm_balance-0.1.14}/pyproject.toml

@@ -1,12 +1,13 @@
 [project]
 name = "mm-balance"
-version = "0.1.12"
+version = "0.1.14"
 description = ""
 requires-python = ">=3.12"
 dependencies = [
     "mm-btc==0.1.0",
     "mm-eth==0.1.3",
-    "mm-solana==0.1.4",
+    "mm-solana==0.1.5",
+    "mm-aptos==0.1.2",
     "typer>=0.12.5",
 ]
 [project.scripts]
@@ -22,10 +23,10 @@ dev-dependencies = [
     "pytest-xdist~=3.6.1",
     "pytest-httpserver~=1.1.0",
     "coverage~=7.6.3",
-    "ruff~=0.6.9",
+    "ruff~=0.7.0",
     "pip-audit~=2.7.0",
     "bandit~=1.7.10",
-    "mypy~=1.12.0",
+    "mypy~=1.12.1",
     "types-python-dateutil~=2.9.0.20241003",
     "types-PyYAML~=6.0.12.20240917",
 ]

{mm_balance-0.1.12 → mm_balance-0.1.14}/src/mm_balance/balances.py

@@ -8,8 +8,8 @@ from rich.progress import TaskID
 
 from mm_balance import output
 from mm_balance.config import Config
-from mm_balance.constants import Network
-from mm_balance.rpc import btc, eth, solana
+from mm_balance.constants import NETWORK_APTOS, NETWORK_BITCOIN, NETWORK_SOLANA, Network
+from mm_balance.rpc import aptos, btc, evm, solana
 from mm_balance.token_decimals import TokenDecimals
 
 
@@ -23,7 +23,7 @@ class Balances:
     def __init__(self, config: Config, token_decimals: TokenDecimals):
         self.config = config
         self.token_decimals = token_decimals
-        self.tasks: dict[Network, list[Balances.Balance]] = {network: [] for network in Network}
+        self.tasks: dict[Network, list[Balances.Balance]] = {network: [] for network in config.networks()}
         self.progress_bar = output.create_progress_bar()
         self.progress_bar_task: dict[Network, TaskID] = {}
 
@@ -31,17 +31,17 @@ class Balances:
             task_list = [Balances.Balance(group_index=idx, address=a, token_address=group.token_address) for a in group.addresses]
             self.tasks[group.network].extend(task_list)
 
-        for network in Network:
+        for network in config.networks():
             if self.tasks[network]:
                 self.progress_bar_task[network] = output.create_progress_task(
-                    self.progress_bar, network.value, len(self.tasks[network])
+                    self.progress_bar, network, len(self.tasks[network])
                 )
 
     def process(self) -> None:
         with self.progress_bar:
             job = ConcurrentTasks(max_workers=10)
-            for network in Network:
-                job.add_task(network.value, self._process_network, args=(network,))
+            for network in self.config.networks():
+                job.add_task(network, self._process_network, args=(network,))
             job.execute()
 
     def _process_network(self, network: Network) -> None:
@@ -49,6 +49,7 @@ class Balances:
         for idx, task in enumerate(self.tasks[network]):
             job.add_task(str(idx), self._get_balance, args=(network, task.address, task.token_address))
         job.execute()
+        # TODO: print job.exceptions if present
         for idx, _task in enumerate(self.tasks[network]):
             self.tasks[network][idx].balance = job.result.get(str(idx))  # type: ignore[assignment]
 
@@ -56,23 +57,18 @@ class Balances:
         nodes = self.config.nodes[network]
         round_ndigits = self.config.round_ndigits
         proxies = self.config.proxies
-        token_decimals = self.token_decimals[network][token_address] if token_address else -1
-        match network:
-            case Network.BITCOIN:
-                res = btc.get_balance(wallet_address, proxies, round_ndigits)
-            case Network.ETHEREUM | Network.ARBITRUM_ONE | Network.OP_MAINNET:
-                if token_address is None:
-                    res = eth.get_native_balance(nodes, wallet_address, proxies, round_ndigits)
-                else:
-                    res = eth.get_token_balance(nodes, wallet_address, token_address, token_decimals, proxies, round_ndigits)
-            case Network.SOLANA:
-                if token_address is None:
-                    res = solana.get_native_balance(nodes, wallet_address, proxies, round_ndigits)
-                else:
-                    res = solana.get_token_balance(nodes, wallet_address, token_address, token_decimals, proxies, round_ndigits)
+        token_decimals = self.token_decimals[network][token_address]
 
-            case _:
-                raise ValueError
+        if network.is_evm_network():
+            res = evm.get_balance(nodes, wallet_address, token_address, token_decimals, proxies, round_ndigits)
+        elif network == NETWORK_BITCOIN:
+            res = btc.get_balance(wallet_address, proxies, round_ndigits)
+        elif network == NETWORK_APTOS:
+            res = aptos.get_balance(nodes, wallet_address, token_address, token_decimals, proxies, round_ndigits)
+        elif network == NETWORK_SOLANA:
+            res = solana.get_balance(nodes, wallet_address, token_address, token_decimals, proxies, round_ndigits)
+        else:
+            raise ValueError(f"Unsupported network: {network}")
 
         self.progress_bar.update(self.progress_bar_task[network], advance=1)
         return res
@@ -80,3 +76,11 @@ class Balances:
     def get_group_balances(self, group_index: int, network: Network) -> list[Balance]:
         # TODO: can we get network by group_index?
         return [b for b in self.tasks[network] if b.group_index == group_index]
+
+    def get_errors(self) -> list[Balance]:
+        result = []
+        for network in self.tasks:
+            for task in self.tasks[network]:
+                if task.balance is not None and task.balance.is_err():
+                    result.append(task)
+        return result

{mm_balance-0.1.12 → mm_balance-0.1.14}/src/mm_balance/cli.py

@@ -8,7 +8,7 @@ import typer
 from mm_balance import output
 from mm_balance.balances import Balances
 from mm_balance.config import Config
-from mm_balance.constants import Network
+from mm_balance.constants import NETWORKS
 from mm_balance.price import Prices, get_prices
 from mm_balance.token_decimals import get_token_decimals
 
@@ -24,7 +24,7 @@ def example_callback(value: bool) -> None:
 
 def networks_callback(value: bool) -> None:
     if value:
-        for network in Network:
+        for network in NETWORKS:
             typer.echo(network)
         raise typer.Exit
 
@@ -42,15 +42,22 @@ def cli(
         zip_password = getpass.getpass("zip password")
     config = Config.read_config(config_path, zip_password=zip_password)
 
+    if config.print_debug:
+        output.print_nodes(config)
+
+    token_decimals = get_token_decimals(config)
+    if config.print_debug:
+        output.print_token_decimals(token_decimals)
+
     prices = get_prices(config) if config.price else Prices()
     output.print_prices(config, prices)
 
-    token_decimals = get_token_decimals(config)
     balances = Balances(config, token_decimals)
     balances.process()
 
     output.print_groups(balances, config, prices)
     output.print_total(config, balances, prices)
+    output.print_errors(config, balances)
 
 
 if __name__ == "__main__":

{mm_balance-0.1.12 → mm_balance-0.1.14}/src/mm_balance/config/example.yml

@@ -38,6 +38,14 @@ coins:
     comment: binance
     addresses: binance_eth
 
+  - ticker: USDC
+    comment: okx aptos
+    network: aptos
+    token_address: 0xf22bede237a07e121b56d91a491eb7bcdfd1f5907926a9e58338f964a01b17fa::asset::USDC
+    token_decimals: 6
+    addresses: "0x834d639b10d20dcb894728aa4b9b572b2ea2d97073b10eacb111f338b20ea5d7" # for a single line it's necessary to use quotes
+
+
 addresses:
   - name: okx_eth
     addresses: |
@@ -60,3 +68,6 @@ addresses:
 #- http://123.123.123.124
 #round_ndigits: 4
 #price: yes
+#skip_empty: no
+#print_debug: no
+#format_number_separator: ","

{mm_balance-0.1.12 → mm_balance-0.1.14}/src/mm_balance/config.py

@@ -1,26 +1,22 @@
 from __future__ import annotations
 
 from decimal import Decimal
+from pathlib import Path
 from typing import Self
 
 import pydash
 from mm_std import BaseConfig, PrintFormat, fatal, hr
 from pydantic import Field, field_validator, model_validator
 
-from mm_balance.constants import (
-    DEFAULT_ARBITRUM_ONE_NODES,
-    DEFAULT_ETHEREUM_NODES,
-    DEFAULT_SOLANA_NODES,
-    TOKEN_ADDRESS,
-    Network,
-)
+from mm_balance.constants import DEFAULT_NODES, TOKEN_ADDRESS, Network
 
 
 class Group(BaseConfig):
     comment: str = ""
     ticker: str
     network: Network
-    token_address: str | None = None
+    token_address: str | None = None  # If None, it's a native token, for example ETH
+    token_decimals: int | None = None
     coingecko_id: str | None = None
     addresses: list[str] = Field(default_factory=list)
     share: Decimal = Decimal(1)
@@ -30,11 +26,11 @@ class Group(BaseConfig):
         result = self.ticker
         if self.comment:
             result += " / " + self.comment
-        result += " / " + self.network.value
+        result += " / " + self.network
         return result
 
     @field_validator("ticker", mode="after")
-    def coin_validator(cls, v: str) -> str:
+    def ticker_validator(cls, v: str) -> str:
         return v.upper()
 
     @field_validator("addresses", mode="before")
@@ -51,7 +47,7 @@ class Group(BaseConfig):
     def final_validator(self) -> Self:
         if self.token_address is None:
             self.token_address = detect_token_address(self.ticker, self.network)
-        if self.token_address is not None and self.network is Network.ETHEREUM:
+        if self.token_address is not None and self.network.is_evm_network():
             self.token_address = self.token_address.lower()
         return self
 
@@ -63,7 +59,7 @@ class Group(BaseConfig):
             else:
                 # TODO: check address is valid
                 addresses.append(address)
-        self.addresses = addresses
+        self.addresses = pydash.uniq(process_file_addresses(addresses))
 
 
 class AddressGroup(BaseConfig):
@@ -85,12 +81,17 @@ class Config(BaseConfig):
     nodes: dict[Network, list[str]] = Field(default_factory=dict)
     print_format: PrintFormat = PrintFormat.TABLE
     price: bool = True
-
-    workers: dict[Network, int] = {network: 5 for network in Network}
+    skip_empty: bool = False  # don't print the address with an empty balance
+    print_debug: bool = False  # print debug info: nodes, token_decimals
+    format_number_separator: str = ","  #  as thousands separators
+    workers: dict[Network, int] = Field(default_factory=dict)
 
     def has_share(self) -> bool:
         return any(g.share != Decimal(1) for g in self.groups)
 
+    def networks(self) -> list[Network]:
+        return pydash.uniq([group.network for group in self.groups])
+
     @model_validator(mode="after")
     def final_validator(self) -> Self:
         # load from proxies_url
@@ -102,36 +103,21 @@ class Config(BaseConfig):
             group.process_addresses(self.addresses)
 
         # load default rpc nodes
-        if Network.BITCOIN not in self.nodes:
-            self.nodes[Network.BITCOIN] = []
-        if Network.ETHEREUM not in self.nodes:
-            self.nodes[Network.ETHEREUM] = DEFAULT_ETHEREUM_NODES
-        if Network.ARBITRUM_ONE not in self.nodes:
-            self.nodes[Network.ARBITRUM_ONE] = DEFAULT_ARBITRUM_ONE_NODES
-        if Network.OP_MAINNET not in self.nodes:
-            self.nodes[Network.OP_MAINNET] = DEFAULT_ARBITRUM_ONE_NODES
-        if Network.SOLANA not in self.nodes:
-            self.nodes[Network.SOLANA] = DEFAULT_SOLANA_NODES
-
-        return self
+        for network in self.networks():
+            if network not in self.nodes:
+                self.nodes[network] = DEFAULT_NODES[network]
 
+        # load default workers
+        for network in self.networks():
+            if network not in self.workers:
+                self.workers[network] = 5
 
-# def detect_network(coin: str) -> Network:
-#
-#     # coin = coin.lower()
-#     # if coin == "btc":
-#     #     return Network.BTC
-#     # if coin == "eth":
-#     #     return Network.ETH
-#     # if coin == "sol":
-#     #     return Network.SOL
-#     # return Network.ETH
-#     # # TODO: raise ValueError(f"can't get network for the coin: {coin}")
+        return self
 
 
-def detect_token_address(coin: str, network: Network) -> str | None:
+def detect_token_address(ticker: str, network: Network) -> str | None:
     if network in TOKEN_ADDRESS:
-        return TOKEN_ADDRESS[network].get(coin)
+        return TOKEN_ADDRESS[network].get(ticker)
 
 
 def get_proxies(proxies_url: str) -> list[str]:
@@ -147,3 +133,17 @@ def get_proxies(proxies_url: str) -> list[str]:
 
 def get_address_group_by_name(address_groups: list[AddressGroup], name: str) -> AddressGroup | None:
     return pydash.find(address_groups, lambda g: g.name == name)
+
+
+def process_file_addresses(addresses: list[str]) -> list[str]:
+    result = []
+    for address in addresses:
+        if address.startswith("file://"):
+            path = Path(address.removeprefix("file://"))
+            if path.is_file():
+                result.extend(path.read_text().strip().splitlines())
+            else:
+                fatal(f"File with addresses not found: {path}")
+        else:
+            result.append(address)
+    return result

mm_balance-0.1.14/src/mm_balance/constants.py

@@ -0,0 +1,69 @@
+from typing import Any
+
+from pydantic import GetCoreSchemaHandler
+from pydantic_core import CoreSchema, core_schema
+
+RETRIES_BALANCE = 5
+RETRIES_DECIMALS = 5
+RETRIES_COINGECKO_PRICES = 5
+TIMEOUT_BALANCE = 5
+TIMEOUT_DECIMALS = 5
+
+
+class Network(str):
+    def is_evm_network(self) -> bool:
+        return self in [NETWORK_ETHEREUM, NETWORK_ARBITRUM_ONE, NETWORK_OP_MAINNET] or self.startswith("evm-")
+
+    @classmethod
+    def __get_pydantic_core_schema__(cls, _source_type: Any, handler: GetCoreSchemaHandler) -> CoreSchema:
+        return core_schema.no_info_after_validator_function(cls, handler(str))
+
+
+NETWORK_APTOS = Network("aptos")
+NETWORK_ARBITRUM_ONE = Network("arbitrum-one")
+NETWORK_BITCOIN = Network("bitcoin")
+NETWORK_ETHEREUM = Network("ethereum")
+NETWORK_SOLANA = Network("solana")
+NETWORK_OP_MAINNET = Network("op-mainnet")
+NETWORKS = [NETWORK_APTOS, NETWORK_ARBITRUM_ONE, NETWORK_BITCOIN, NETWORK_ETHEREUM, NETWORK_SOLANA, NETWORK_OP_MAINNET]
+
+
+TOKEN_ADDRESS: dict[Network, dict[str, str]] = {
+    NETWORK_ETHEREUM: {
+        "USDT": "0xdac17f958d2ee523a2206206994597c13d831ec7",
+        "USDC": "0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48",
+    },
+    NETWORK_SOLANA: {
+        "USDT": "Es9vMFrzaCERmJfrF4H2FYD4KCoNkY11McCe8BenwNYB",
+        "USDC": "EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v",
+    },
+    NETWORK_ARBITRUM_ONE: {
+        "USDT": "0xfd086bc7cd5c481dcc9c85ebe478a1c0b69fcbb9",
+        "USDC": "0xff970a61a04b1ca14834a43f5de4533ebddb5cc8",
+    },
+    NETWORK_OP_MAINNET: {
+        "USDT": "0x94b008aA00579c1307B0EF2c499aD98a8ce58e58",
+        "USDC": "0x7f5c764cbc14f9669b88837ca1490cca17c31607",
+    },
+}
+
+TICKER_TO_COINGECKO_ID = {
+    "BTC": "bitcoin",
+    "ETH": "ethereum",
+    "USDT": "tether",
+    "USDC": "usd-coin",
+    "SOL": "solana",
+    "APT": "aptos",
+    "POL": "matic-network",
+}
+
+USD_STABLECOINS = ["USDT", "USDC"]
+
+DEFAULT_NODES: dict[Network, list[str]] = {
+    NETWORK_ARBITRUM_ONE: ["https://arb1.arbitrum.io/rpc", "https://arbitrum.llamarpc.com"],
+    NETWORK_BITCOIN: [],
+    NETWORK_ETHEREUM: ["https://ethereum.publicnode.com", "https://rpc.ankr.com/eth"],
+    NETWORK_SOLANA: ["https://api.mainnet-beta.solana.com"],
+    NETWORK_OP_MAINNET: ["https://mainnet.optimism.io", "https://optimism.llamarpc.com"],
+    NETWORK_APTOS: ["https://fullnode.mainnet.aptoslabs.com/v1"],
+}

mm_balance-0.1.14/src/mm_balance/output.py

@@ -0,0 +1,109 @@
+from decimal import Decimal
+
+from mm_std import Err, Ok, fatal, print_table
+from rich.progress import BarColumn, MofNCompleteColumn, Progress, TaskID, TextColumn
+
+from mm_balance.balances import Balances
+from mm_balance.config import Config, Group
+from mm_balance.price import Prices
+from mm_balance.token_decimals import TokenDecimals
+from mm_balance.total import Total
+from mm_balance.utils import fnumber
+
+
+def print_groups(balances: Balances, config: Config, prices: Prices) -> None:
+    for group_index, group in enumerate(config.groups):
+        group_balances = balances.get_group_balances(group_index, group.network)
+        _print_group(group, group_balances, config, prices)
+
+
+def _print_group(group: Group, group_balances: list[Balances.Balance], config: Config, prices: Prices) -> None:
+    rows = []
+    balance_sum = Decimal(0)
+    usd_sum = Decimal(0)
+    for address_task in group_balances:
+        if isinstance(address_task.balance, Err):
+            row = [address_task.address, address_task.balance.err]
+        elif isinstance(address_task.balance, Ok):
+            balance = address_task.balance.ok
+            balance_sum += balance
+            if config.skip_empty and balance == Decimal(0):
+                continue
+            row = [address_task.address, fnumber(balance, config.format_number_separator)]
+            if config.price:
+                balance_usd = round(balance * prices[group.ticker], config.round_ndigits)
+                usd_sum += balance_usd
+                row.append(fnumber(balance_usd, config.format_number_separator, "$"))
+        else:
+            fatal("address_task is None!")
+        rows.append(row)
+
+    balance_sum_str = fnumber(round(balance_sum, config.round_ndigits), config.format_number_separator)
+    sum_row = ["sum", balance_sum_str]
+    if config.price:
+        usd_sum_str = fnumber(round(usd_sum, config.round_ndigits), config.format_number_separator, "$")
+        sum_row.append(usd_sum_str)
+    rows.append(sum_row)
+
+    if group.share < Decimal(1):
+        sum_share_str = fnumber(round(balance_sum * group.share, config.round_ndigits), config.format_number_separator)
+        sum_share_row = [f"sum_share, {group.share}", sum_share_str]
+        if config.price:
+            usd_sum_share_str = fnumber(round(usd_sum * group.share, config.round_ndigits), config.format_number_separator, "$")
+            sum_share_row.append(usd_sum_share_str)
+        rows.append(sum_share_row)
+
+    table_headers = ["address", "balance"]
+    if config.price:
+        table_headers += ["usd"]
+    print_table(group.name, table_headers, rows)
+
+
+def print_nodes(config: Config) -> None:
+    rows = []
+    for network, nodes in config.nodes.items():
+        rows.append([network, "\n".join(nodes)])
+    print_table("Nodes", ["network", "nodes"], rows)
+
+
+def print_token_decimals(token_decimals: TokenDecimals) -> None:
+    rows = []
+    for network, decimals in token_decimals.items():
+        rows.append([network, decimals])
+    print_table("Token Decimals", ["network", "decimals"], rows)
+
+
+def print_prices(config: Config, prices: Prices) -> None:
+    if config.price:
+        rows = []
+        for ticker, price in prices.items():
+            rows.append([ticker, fnumber(round(price, config.round_ndigits), config.format_number_separator, "$")])
+        print_table("Prices", ["coin", "usd"], rows)
+
+
+def print_total(config: Config, balances: Balances, prices: Prices) -> None:
+    total = Total.calc(balances, prices, config)
+    total.print()
+
+
+def print_errors(config: Config, balances: Balances) -> None:
+    error_balances = balances.get_errors()
+    if not error_balances:
+        return
+    rows = []
+    for balance in error_balances:
+        group = config.groups[balance.group_index]
+        rows.append([group.ticker + " / " + group.network, balance.address, balance.balance.err])  # type: ignore[union-attr]
+    print_table("Errors", ["coin", "address", "error"], rows)
+
+
+def create_progress_bar() -> Progress:
+    return Progress(
+        TextColumn("[progress.description]{task.description}"),
+        BarColumn(),
+        MofNCompleteColumn(),
+    )
+
+
+def create_progress_task(progress: Progress, description: str, total: int) -> TaskID:
+    return progress.add_task("[green]" + description, total=total)

{mm_balance-0.1.12 → mm_balance-0.1.14}/src/mm_balance/price.py

@@ -5,7 +5,7 @@ from mm_std import fatal, hr
 from mm_std.random_ import random_str_choice
 
 from mm_balance.config import Config, Group
-from mm_balance.constants import RETRIES_COINGECKO_PRICES, Network
+from mm_balance.constants import RETRIES_COINGECKO_PRICES, TICKER_TO_COINGECKO_ID
 
 
 class Prices(dict[str, Decimal]):
@@ -38,29 +38,11 @@ def get_prices(config: Config) -> Prices:
     return result
 
 
-TICKER_TO_COINGECKO_ID = {
-    "BTC": "bitcoin",
-    "ETH": "ethereum",
-    "USDT": "tether",
-    "USDC": "usd-coin",
-    "SOL": "solana",
-}
-
-
 def get_coingecko_id(group: Group) -> str:
     if group.coingecko_id:
         return group.coingecko_id
-    elif group.network is Network.BITCOIN:
-        return "bitcoin"
-    elif group.network is Network.ETHEREUM and group.token_address is None:
-        return "ethereum"
-    elif group.ticker == "ETH":
-        return "ethereum"
-    elif group.ticker == "USDT":
-        return "tether"
-    elif group.ticker == "USDC":
-        return "usd-coin"
-    elif group.ticker == "SOL":
-        return "solana"
+    coingecko_id = TICKER_TO_COINGECKO_ID.get(group.ticker)
+    if coingecko_id:
+        return coingecko_id
 
-    raise ValueError(f"can't get coingecko_id for {group.ticker}")
+    fatal(f"Can't get coingecko_id for {group.ticker}. Please add coingecko_id to the config.")

mm_balance-0.1.14/src/mm_balance/rpc/aptos.py

@@ -0,0 +1,23 @@
+from decimal import Decimal
+
+from mm_aptos import balance
+from mm_std import Result
+
+from mm_balance.constants import RETRIES_BALANCE, TIMEOUT_BALANCE
+
+
+def get_balance(
+    nodes: list[str], wallet: str, token: str | None, decimals: int, proxies: list[str], round_ndigits: int
+) -> Result[Decimal]:
+    if token is None:
+        token = "0x1::aptos_coin::AptosCoin"  # nosec
+    return balance.get_decimal_balance_with_retries(
+        RETRIES_BALANCE,
+        nodes,
+        wallet,
+        token,
+        decimals=decimals,
+        timeout=TIMEOUT_BALANCE,
+        proxies=proxies,
+        round_ndigits=round_ndigits,
+    )

mm_balance-0.1.14/src/mm_balance/rpc/evm.py

@@ -0,0 +1,27 @@
+from decimal import Decimal
+
+from mm_eth import erc20, rpc
+from mm_std import Ok, Result
+
+from mm_balance.constants import RETRIES_BALANCE, RETRIES_DECIMALS, TIMEOUT_BALANCE, TIMEOUT_DECIMALS
+
+
+def get_balance(
+    nodes: list[str], wallet: str, token: str | None, decimals: int, proxies: list[str], round_ndigits: int
+) -> Result[Decimal]:
+    if token is not None:
+        res = erc20.get_balance(
+            nodes,
+            token,
+            wallet,
+            proxies=proxies,
+            attempts=RETRIES_BALANCE,
+            timeout=TIMEOUT_BALANCE,
+        )
+    else:
+        res = rpc.eth_get_balance(nodes, wallet, proxies=proxies, attempts=RETRIES_BALANCE, timeout=TIMEOUT_BALANCE)
+    return res.and_then(lambda b: Ok(round(Decimal(b / 10**decimals), round_ndigits)))
+
+
+def get_token_decimals(nodes: list[str], token_address: str, proxies: list[str]) -> Result[int]:
+    return erc20.get_decimals(nodes, token_address, timeout=TIMEOUT_DECIMALS, proxies=proxies, attempts=RETRIES_DECIMALS)

mm_balance-0.1.14/src/mm_balance/rpc/solana.py

@@ -0,0 +1,26 @@
+from decimal import Decimal
+
+from mm_solana import balance
+from mm_solana import token as solana_token
+from mm_std import Ok, Result
+
+from mm_balance.constants import RETRIES_BALANCE, RETRIES_DECIMALS, TIMEOUT_BALANCE, TIMEOUT_DECIMALS
+
+
+def get_balance(
+    nodes: list[str], wallet: str, token: str | None, decimals: int, proxies: list[str], round_ndigits: int
+) -> Result[Decimal]:
+    if token is None:
+        res = balance.get_balance_with_retries(nodes, wallet, retries=RETRIES_BALANCE, timeout=TIMEOUT_BALANCE, proxies=proxies)
+    else:
+        res = solana_token.get_balance_with_retries(
+            nodes, wallet, token, retries=RETRIES_BALANCE, timeout=TIMEOUT_BALANCE, proxies=proxies
+        )
+
+    return res.and_then(lambda b: Ok(round(Decimal(b / 10**decimals), round_ndigits)))
+
+
+def get_token_decimals(nodes: list[str], token_address: str, proxies: list[str]) -> Result[int]:
+    return solana_token.get_decimals_with_retries(
+        nodes, token_address, retries=RETRIES_DECIMALS, timeout=TIMEOUT_DECIMALS, proxies=proxies
+    )