mm-balance 0.1.16__tar.gz → 0.1.18__tar.gz

mm_balance-0.1.18/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: mm-balance
+Version: 0.1.18
+Requires-Python: >=3.12
+Requires-Dist: mm-aptos==0.1.5
+Requires-Dist: mm-btc==0.2.1
+Requires-Dist: mm-eth==0.2.3
+Requires-Dist: mm-solana==0.2.3
+Requires-Dist: typer>=0.15.1

{mm_balance-0.1.16 → mm_balance-0.1.18}/justfile

@@ -6,14 +6,14 @@ clean:
     rm -rf .pytest_cache .mypy_cache .ruff_cache .coverage dist build src/*.egg-info
 
 build: clean lint audit test
-    uvx --from build pyproject-build --installer uv
+    uv build
 
 format:
     uv run ruff check --select I --fix src tests
     uv run ruff format src tests
 
 test:
-    uv run coverage run -m pytest -n auto tests
+    uv run pytest -n auto tests
 
 lint: format
     uv run ruff check src tests

mm_balance-0.1.18/pyproject.toml

@@ -0,0 +1,72 @@
+[project]
+name = "mm-balance"
+version = "0.1.18"
+description = ""
+requires-python = ">=3.12"
+dependencies = [
+    "mm-btc==0.2.1",
+    "mm-eth==0.2.3",
+    "mm-solana==0.2.3",
+    "mm-aptos==0.1.5",
+    "typer>=0.15.1",
+]
+[project.scripts]
+mm-balance = "mm_balance.cli:app"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.uv]
+dev-dependencies = [
+    "pytest~=8.3.4",
+    "pytest-xdist~=3.6.1",
+    "ruff~=0.9.2",
+    "pip-audit~=2.7.3",
+    "bandit~=1.8.2",
+    "mypy~=1.14.1",
+    "types-PyYAML~=6.0.12.20241230",
+]
+
+[tool.mypy]
+python_version = "3.13"
+warn_no_return = false
+strict = true
+exclude = ["^tests/", "^tmp/"]
+
+[tool.ruff]
+line-length = 130
+target-version = "py313"
+[tool.ruff.lint]
+select = ["ALL"]
+ignore = [
+    "A005", # flake8-builtins: stdlib-module-shadowing
+    "ERA001", # eradicate: commented-out-code
+    "PT", # flake8-pytest-style
+    "D", # pydocstyle
+    "FIX", # flake8-fixme
+    "PLR0911", # pylint: too-many-return-statements
+    "PLR0912", # pylint: too-many-branches
+    "PLR0913", # pylint: too-many-arguments
+    "PLR2004", # pylint: magic-value-comparison
+    "PLC0414", # pylint: useless-import-alias
+    "FBT", # flake8-boolean-trap
+    "EM", # flake8-errmsg
+    "TRY003", # tryceratops: raise-vanilla-args
+    "C901", # mccabe: complex-structure,
+    "BLE001", # flake8-blind-except
+    "S311", # bandit: suspicious-non-cryptographic-random-usage
+    "TD002", # flake8-todos: missing-todo-author
+    "TD003", # flake8-todos: missing-todo-link
+    "RET503", # flake8-return: implicit-return
+    "COM812", # it's used in ruff formatter
+]
+[tool.ruff.lint.per-file-ignores]
+"tests/*.py" = ["ANN", "S"]
+[tool.ruff.format]
+quote-style = "double"
+indent-style = "space"
+
+[tool.bandit]
+exclude_dirs = ["tests"]
+skips = ["B311"]

{mm_balance-0.1.16 → mm_balance-0.1.18}/src/mm_balance/cli.py

@@ -46,7 +46,7 @@ def cli(
     zip_password = ""  # nosec
     if config_path.name.endswith(".zip"):
         zip_password = getpass.getpass("zip password")
-    config = Config.read_config(config_path, zip_password=zip_password)
+    config = Config.read_config_or_exit(config_path, zip_password=zip_password)
 
     if print_format is not None:
         config.print_format = print_format

{mm_balance-0.1.16 → mm_balance-0.1.18}/src/mm_balance/config.py

@@ -30,10 +30,12 @@ class Group(BaseConfig):
         result += " / " + self.network
         return result
 
+    @classmethod
     @field_validator("ticker", mode="after")
     def ticker_validator(cls, v: str) -> str:
         return v.upper()
 
+    @classmethod
     @field_validator("addresses", mode="before")
     def to_list_validator(cls, v: str | list[str] | None) -> list[str]:
         return cls.to_list_str_validator(v, unique=True, remove_comments=True, split_line=True)
@@ -67,6 +69,7 @@ class AddressGroup(BaseConfig):
     name: str
     addresses: list[str]
 
+    @classmethod
     @field_validator("addresses", mode="before")
     def to_list_validator(cls, v: str | list[str] | None) -> list[str]:
         return cls.to_list_str_validator(v, unique=True, remove_comments=True, split_line=True)

{mm_balance-0.1.16 → mm_balance-0.1.18}/src/mm_balance/constants.py

@@ -1,5 +1,3 @@
-from typing import Any
-
 from pydantic import GetCoreSchemaHandler
 from pydantic_core import CoreSchema, core_schema
 
@@ -11,11 +9,13 @@ TIMEOUT_DECIMALS = 5
 
 
 class Network(str):
+    __slots__ = ()
+
     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:
+    def __get_pydantic_core_schema__(cls, _source_type: object, handler: GetCoreSchemaHandler) -> CoreSchema:
         return core_schema.no_info_after_validator_function(cls, handler(str))
 
 

{mm_balance-0.1.16 → mm_balance-0.1.18}/src/mm_balance/output/utils.py

@@ -7,7 +7,7 @@ def format_number(value: Decimal, separator: str, extra: str | None = None) -> s
     str_value = f"{value:,}".replace(",", separator)
     if extra == "$":
         return "$" + str_value
-    elif extra == "%":
+    if extra == "%":
         return str_value + "%"
     return str_value
 

mm_balance-0.1.18/src/mm_balance/price.py

@@ -0,0 +1,48 @@
+from collections import defaultdict
+from decimal import Decimal
+
+import pydash
+from mm_std import hr
+from mm_std.random_ import random_str_choice
+
+from mm_balance.config import Config, Group
+from mm_balance.constants import RETRIES_COINGECKO_PRICES, TICKER_TO_COINGECKO_ID
+
+
+class Prices(defaultdict[str, Decimal]):
+    """
+    A Prices class representing a mapping from coin names to their prices.
+
+    Inherits from:
+        Dict[str, Decimal]: A dictionary with coin names as keys and their prices as Decimal values.
+    """
+
+
+def get_prices(config: Config) -> Prices:
+    result = Prices()
+
+    coingecko_map: dict[str, str] = {}  # ticker -> coingecko_id
+
+    for group in config.groups:
+        coingecko_id = get_coingecko_id(group)
+        if coingecko_id:
+            coingecko_map[group.ticker] = coingecko_id
+
+    url = f"https://api.coingecko.com/api/v3/simple/price?ids={','.join(coingecko_map.values())}&vs_currencies=usd"
+    for _ in range(RETRIES_COINGECKO_PRICES):
+        res = hr(url, proxy=random_str_choice(config.proxies))
+        if res.code != 200:
+            continue
+
+        for ticker, coingecko_id in coingecko_map.items():
+            if coingecko_id in res.json:
+                result[ticker] = Decimal(str(pydash.get(res.json, f"{coingecko_id}.usd")))
+        break
+
+    return result
+
+
+def get_coingecko_id(group: Group) -> str | None:
+    if group.coingecko_id:
+        return group.coingecko_id
+    return TICKER_TO_COINGECKO_ID.get(group.ticker)

{mm_balance-0.1.16 → mm_balance-0.1.18}/src/mm_balance/result.py

@@ -102,17 +102,16 @@ def _create_group_result(config: Config, group: Group, tasks: list[Task], prices
         balance: Balance | str
         if task.balance is None:
             balance = "balance is None! Something went wrong."
+        elif isinstance(task.balance, Ok):
+            coin_value = task.balance.ok
+            usd_value = Decimal(0)
+            if group.ticker in prices:
+                usd_value = round(coin_value * prices[group.ticker], config.round_ndigits)
+            balance = Balance(balance=coin_value, usd_value=usd_value)
+            balance_sum += balance.balance
+            usd_sum += balance.usd_value
         else:
-            if isinstance(task.balance, Ok):
-                coin_value = task.balance.ok
-                usd_value = Decimal(0)
-                if group.ticker in prices:
-                    usd_value = round(coin_value * prices[group.ticker], config.round_ndigits)
-                balance = Balance(balance=coin_value, usd_value=usd_value)
-                balance_sum += balance.balance
-                usd_sum += balance.usd_value
-            else:
-                balance = task.balance.err
+            balance = task.balance.err
         addresses.append(AddressBalance(address=task.wallet_address, balance=balance))
 
     balance_sum_share = balance_sum * group.share

{mm_balance-0.1.16 → mm_balance-0.1.18}/src/mm_balance/rpc/aptos.py

@@ -10,7 +10,7 @@ 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
+        token = "0x1::aptos_coin::AptosCoin"  # noqa: S105 # nosec
     return balance.get_decimal_balance_with_retries(
         RETRIES_BALANCE,
         nodes,

{mm_balance-0.1.16 → mm_balance-0.1.18}/src/mm_balance/token_decimals.py

@@ -28,9 +28,7 @@ def get_token_decimals(config: Config) -> TokenDecimals:
                 result[group.network][None] = 18
             elif group.network == NETWORK_SOLANA:
                 result[group.network][None] = 9
-            elif group.network == NETWORK_BITCOIN:
-                result[group.network][None] = 8
-            elif group.network == NETWORK_APTOS:
+            elif group.network in (NETWORK_BITCOIN, NETWORK_APTOS):
                 result[group.network][None] = 8
             else:
                 fatal(f"Can't get token decimals for native token on network: {group.network}")

{mm_balance-0.1.16 → mm_balance-0.1.18}/src/mm_balance/utils.py

@@ -5,6 +5,6 @@ def fnumber(value: Decimal, separator: str, extra: str | None = None) -> str:
     str_value = f"{value:,}".replace(",", separator)
     if extra == "$":
         return "$" + str_value
-    elif extra == "%":
+    if extra == "%":
         return str_value + "%"
     return str_value

{mm_balance-0.1.16 → mm_balance-0.1.18}/src/mm_balance/workers.py

@@ -1,8 +1,8 @@
 from dataclasses import dataclass
 from decimal import Decimal
+from typing import TYPE_CHECKING
 
 from mm_std import ConcurrentTasks, PrintFormat, Result
-from rich.progress import TaskID
 
 from mm_balance.config import Config
 from mm_balance.constants import NETWORK_APTOS, NETWORK_BITCOIN, NETWORK_SOLANA, Network
@@ -10,6 +10,9 @@ from mm_balance.output import utils
 from mm_balance.rpc import aptos, btc, evm, solana
 from mm_balance.token_decimals import TokenDecimals
 
+if TYPE_CHECKING:
+    from rich.progress import TaskID
+
 
 @dataclass
 class Task:
@@ -20,7 +23,7 @@ class Task:
 
 
 class Workers:
-    def __init__(self, config: Config, token_decimals: TokenDecimals):
+    def __init__(self, config: Config, token_decimals: TokenDecimals) -> None:
         self.config = config
         self.token_decimals = token_decimals
         self.tasks: dict[Network, list[Task]] = {network: [] for network in config.networks()}
@@ -49,9 +52,7 @@ class Workers:
     def get_errors(self) -> list[Task]:
         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)
+            result.extend([task for task in self.tasks[network] if task.balance is not None and task.balance.is_err()])
         return result
 
     def _process_network(self, network: Network) -> None: