mm-balance 0.1.10__tar.gz → 0.1.11__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (27) hide show
  1. {mm_balance-0.1.10 → mm_balance-0.1.11}/PKG-INFO +1 -1
  2. {mm_balance-0.1.10 → mm_balance-0.1.11}/pyproject.toml +1 -1
  3. {mm_balance-0.1.10 → mm_balance-0.1.11}/src/mm_balance/balances.py +2 -2
  4. {mm_balance-0.1.10 → mm_balance-0.1.11}/src/mm_balance/cli.py +2 -1
  5. {mm_balance-0.1.10 → mm_balance-0.1.11}/src/mm_balance/config/example.yml +2 -2
  6. mm_balance-0.1.11/src/mm_balance/config.py +146 -0
  7. mm_balance-0.1.10/src/mm_balance/types.py → mm_balance-0.1.11/src/mm_balance/constants.py +19 -12
  8. {mm_balance-0.1.10 → mm_balance-0.1.11}/src/mm_balance/output.py +2 -2
  9. mm_balance-0.1.11/src/mm_balance/price.py +55 -0
  10. mm_balance-0.1.11/src/mm_balance/rpc/btc.py +16 -0
  11. {mm_balance-0.1.10 → mm_balance-0.1.11}/src/mm_balance/rpc/eth.py +6 -21
  12. mm_balance-0.1.11/src/mm_balance/rpc/solana.py +26 -0
  13. {mm_balance-0.1.10 → mm_balance-0.1.11}/src/mm_balance/token_decimals.py +1 -1
  14. {mm_balance-0.1.10 → mm_balance-0.1.11}/src/mm_balance/total.py +1 -1
  15. {mm_balance-0.1.10 → mm_balance-0.1.11}/uv.lock +1 -1
  16. mm_balance-0.1.10/src/mm_balance/config.py +0 -153
  17. mm_balance-0.1.10/src/mm_balance/price.py +0 -83
  18. mm_balance-0.1.10/src/mm_balance/rpc/btc.py +0 -20
  19. mm_balance-0.1.10/src/mm_balance/rpc/solana.py +0 -22
  20. {mm_balance-0.1.10 → mm_balance-0.1.11}/.gitignore +0 -0
  21. {mm_balance-0.1.10 → mm_balance-0.1.11}/README.md +0 -0
  22. {mm_balance-0.1.10 → mm_balance-0.1.11}/justfile +0 -0
  23. {mm_balance-0.1.10 → mm_balance-0.1.11}/src/mm_balance/__init__.py +0 -0
  24. {mm_balance-0.1.10 → mm_balance-0.1.11}/src/mm_balance/rpc/__init__.py +0 -0
  25. {mm_balance-0.1.10 → mm_balance-0.1.11}/tests/__init__.py +0 -0
  26. {mm_balance-0.1.10 → mm_balance-0.1.11}/tests/conftest.py +0 -0
  27. {mm_balance-0.1.10 → mm_balance-0.1.11}/tests/test_dummy.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: mm-balance
3
- Version: 0.1.10
3
+ Version: 0.1.11
4
4
  Requires-Python: >=3.12
5
5
  Requires-Dist: mm-btc==0.1.0
6
6
  Requires-Dist: mm-eth==0.1.3
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "mm-balance"
3
- version = "0.1.10"
3
+ version = "0.1.11"
4
4
  description = ""
5
5
  requires-python = ">=3.12"
6
6
  dependencies = [
@@ -8,9 +8,9 @@ from rich.progress import TaskID
8
8
 
9
9
  from mm_balance import output
10
10
  from mm_balance.config import Config
11
+ from mm_balance.constants import Network
11
12
  from mm_balance.rpc import btc, eth, solana
12
13
  from mm_balance.token_decimals import TokenDecimals
13
- from mm_balance.types import Network
14
14
 
15
15
 
16
16
  class Balances:
@@ -58,7 +58,7 @@ class Balances:
58
58
  token_decimals = self.token_decimals[network][token_address] if token_address else -1
59
59
  match network:
60
60
  case Network.BTC:
61
- res = btc.get_balance(wallet_address, self.config)
61
+ res = btc.get_balance(wallet_address, proxies, round_ndigits)
62
62
  case Network.ETH:
63
63
  if token_address is None:
64
64
  res = eth.get_native_balance(nodes, wallet_address, proxies, round_ndigits)
@@ -32,11 +32,12 @@ def cli(
32
32
  config = Config.read_config(config_path, zip_password=zip_password)
33
33
 
34
34
  prices = get_prices(config) if config.price else Prices()
35
+ output.print_prices(config, prices)
36
+
35
37
  token_decimals = get_token_decimals(config)
36
38
  balances = Balances(config, token_decimals)
37
39
  balances.process()
38
40
 
39
- output.print_prices(config, prices)
40
41
  output.print_groups(balances, config, prices)
41
42
  output.print_total(config, balances, prices)
42
43
 
@@ -1,4 +1,4 @@
1
- groups:
1
+ coins:
2
2
  - coin: sol
3
3
  addresses:
4
4
  - 2ojv9BAiHUrvsm9gxDe7fJSzbNZSJcxZvf8dqmWGHG8S # binance
@@ -53,4 +53,4 @@ addresses:
53
53
  #- http://123.123.123.123
54
54
  #- http://123.123.123.124
55
55
  #round_ndigits: 4
56
- #price: no
56
+ #price: yes
@@ -0,0 +1,146 @@
1
+ from __future__ import annotations
2
+
3
+ from decimal import Decimal
4
+ from typing import Any, Self
5
+
6
+ import pydash
7
+ from mm_std import BaseConfig, PrintFormat, fatal, hr
8
+ from pydantic import Field, field_validator, model_validator
9
+
10
+ from mm_balance.constants import DEFAULT_ETH_NODES, DEFAULT_SOL_NODES, EthTokenAddress, Network, SolTokenAddress
11
+
12
+
13
+ class Group(BaseConfig):
14
+ comment: str = ""
15
+ coin: str
16
+ network: Network
17
+ token_address: str | None = None
18
+ coingecko_id: str | None = None
19
+ addresses: list[str] = Field(default_factory=list)
20
+ share: Decimal = Decimal(1)
21
+
22
+ @property
23
+ def name(self) -> str:
24
+ result = self.coin
25
+ if self.comment:
26
+ result += " / " + self.comment
27
+ return result
28
+
29
+ @field_validator("coin", mode="after")
30
+ def coin_validator(cls, v: str) -> str:
31
+ return v.upper()
32
+
33
+ @field_validator("addresses", mode="before")
34
+ def to_list_validator(cls, v: str | list[str] | None) -> list[str]:
35
+ return cls.to_list_str_validator(v, unique=True, remove_comments=True, split_line=True)
36
+
37
+ @model_validator(mode="before")
38
+ def before_all(cls, data: Any) -> Any:
39
+ if "network" not in data:
40
+ data["network"] = detect_network(data["coin"])
41
+ return data
42
+
43
+ @model_validator(mode="after")
44
+ def final_validator(self) -> Self:
45
+ if self.token_address is None:
46
+ self.token_address = detect_token_address(self.coin, self.network)
47
+ if self.token_address is not None and self.network is Network.ETH:
48
+ self.token_address = self.token_address.lower()
49
+ return self
50
+
51
+ def process_addresses(self, address_groups: list[AddressGroup]) -> None:
52
+ addresses: list[str] = []
53
+ for address in self.addresses:
54
+ if address_group := pydash.find(address_groups, lambda g: g.name == address): # noqa: B023
55
+ addresses.extend(address_group.addresses)
56
+ else:
57
+ # TODO: check address is valid
58
+ addresses.append(address)
59
+ self.addresses = addresses
60
+
61
+
62
+ class AddressGroup(BaseConfig):
63
+ name: str
64
+ addresses: list[str]
65
+
66
+ @field_validator("addresses", mode="before")
67
+ def to_list_validator(cls, v: str | list[str] | None) -> list[str]:
68
+ return cls.to_list_str_validator(v, unique=True, remove_comments=True, split_line=True)
69
+
70
+
71
+ class Config(BaseConfig):
72
+ groups: list[Group] = Field(alias="coins")
73
+ addresses: list[AddressGroup] = Field(default_factory=list)
74
+
75
+ proxies_url: str | None = None
76
+ proxies: list[str] = Field(default_factory=list)
77
+ round_ndigits: int = 4
78
+ nodes: dict[Network, list[str]] = Field(default_factory=dict)
79
+ print_format: PrintFormat = PrintFormat.TABLE
80
+ price: bool = True
81
+
82
+ workers: dict[Network, int] = {network: 5 for network in Network}
83
+
84
+ def has_share(self) -> bool:
85
+ return any(g.share != Decimal(1) for g in self.groups)
86
+
87
+ @model_validator(mode="after")
88
+ def final_validator(self) -> Self:
89
+ # load from proxies_url
90
+ if self.proxies_url is not None:
91
+ self.proxies = get_proxies(self.proxies_url)
92
+
93
+ # load addresses from address_group
94
+ for group in self.groups:
95
+ group.process_addresses(self.addresses)
96
+
97
+ # load default rpc nodes
98
+ if Network.BTC not in self.nodes:
99
+ self.nodes[Network.BTC] = []
100
+ if Network.ETH not in self.nodes:
101
+ self.nodes[Network.ETH] = DEFAULT_ETH_NODES
102
+ if Network.SOL not in self.nodes:
103
+ self.nodes[Network.SOL] = DEFAULT_SOL_NODES
104
+
105
+ return self
106
+
107
+
108
+ def detect_network(coin: str) -> Network:
109
+ coin = coin.lower()
110
+ if coin == "btc":
111
+ return Network.BTC
112
+ if coin == "eth":
113
+ return Network.ETH
114
+ if coin == "sol":
115
+ return Network.SOL
116
+ return Network.ETH
117
+ # TODO: raise ValueError(f"can't get network for the coin: {coin}")
118
+
119
+
120
+ def detect_token_address(coin: str, network: str) -> str | None:
121
+ if network == Network.ETH.lower():
122
+ if coin.lower() == "usdt":
123
+ return EthTokenAddress.USDT
124
+ if coin.lower() == "usdc":
125
+ return EthTokenAddress.USDC
126
+
127
+ if network == Network.SOL.lower():
128
+ if coin.lower() == "usdt":
129
+ return SolTokenAddress.USDT
130
+ if coin.lower() == "usdc":
131
+ return SolTokenAddress.USDC
132
+
133
+
134
+ def get_proxies(proxies_url: str) -> list[str]:
135
+ try:
136
+ res = hr(proxies_url)
137
+ if res.is_error():
138
+ fatal(f"Can't get proxies: {res.error}")
139
+ proxies = [p.strip() for p in res.body.splitlines() if p.strip()]
140
+ return pydash.uniq(proxies)
141
+ except Exception as err:
142
+ fatal(f"Can't get proxies: {err}")
143
+
144
+
145
+ def get_address_group_by_name(address_groups: list[AddressGroup], name: str) -> AddressGroup | None:
146
+ return pydash.find(address_groups, lambda g: g.name == name)
@@ -2,6 +2,25 @@ from __future__ import annotations
2
2
 
3
3
  from enum import Enum, unique
4
4
 
5
+ RETRIES_BALANCE = 5
6
+ RETRIES_DECIMALS = 5
7
+ RETRIES_COINGECKO_PRICES = 5
8
+ TIMEOUT_BALANCE = 5
9
+ TIMEOUT_DECIMALS = 5
10
+
11
+
12
+ @unique
13
+ class EthTokenAddress(str, Enum):
14
+ USDT = "0xdac17f958d2ee523a2206206994597c13d831ec7"
15
+ USDC = "0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48"
16
+
17
+
18
+ @unique
19
+ class SolTokenAddress(str, Enum):
20
+ USDT = "Es9vMFrzaCERmJfrF4H2FYD4KCoNkY11McCe8BenwNYB"
21
+ USDC = "EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v"
22
+
23
+
5
24
  DEFAULT_ETH_NODES = ["https://ethereum.publicnode.com", "https://rpc.ankr.com/eth"]
6
25
  DEFAULT_SOL_NODES = ["https://api.mainnet-beta.solana.com"]
7
26
 
@@ -24,15 +43,3 @@ class Network(str, Enum):
24
43
  BTC = "btc"
25
44
  ETH = "eth"
26
45
  SOL = "sol"
27
-
28
-
29
- @unique
30
- class EthTokenAddress(str, Enum):
31
- USDT = "0xdac17f958d2ee523a2206206994597c13d831ec7"
32
- USDC = "0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48"
33
-
34
-
35
- @unique
36
- class SolTokenAddress(str, Enum):
37
- USDT = "Es9vMFrzaCERmJfrF4H2FYD4KCoNkY11McCe8BenwNYB"
38
- USDC = "EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v"
@@ -4,7 +4,7 @@ from mm_std import Ok, print_table
4
4
  from rich.progress import BarColumn, MofNCompleteColumn, Progress, TaskID, TextColumn
5
5
 
6
6
  from mm_balance.balances import Balances
7
- from mm_balance.config import Config
7
+ from mm_balance.config import Config, Group
8
8
  from mm_balance.price import Prices
9
9
  from mm_balance.total import Total
10
10
 
@@ -15,7 +15,7 @@ def print_groups(balances: Balances, config: Config, prices: Prices) -> None:
15
15
  _print_group(group, group_balances, config, prices)
16
16
 
17
17
 
18
- def _print_group(group: Config.Group, group_balances: list[Balances.Balance], config: Config, prices: Prices) -> None:
18
+ def _print_group(group: Group, group_balances: list[Balances.Balance], config: Config, prices: Prices) -> None:
19
19
  rows = []
20
20
  balance_sum = Decimal(0)
21
21
  usd_sum = Decimal(0)
@@ -0,0 +1,55 @@
1
+ from decimal import Decimal
2
+
3
+ import pydash
4
+ from mm_std import fatal, hr
5
+ from mm_std.random_ import random_str_choice
6
+
7
+ from mm_balance.config import Config, Group
8
+ from mm_balance.constants import RETRIES_COINGECKO_PRICES, EthTokenAddress, Network
9
+
10
+
11
+ class Prices(dict[str, Decimal]):
12
+ """
13
+ A Prices class representing a mapping from coin names to their prices.
14
+
15
+ Inherits from:
16
+ Dict[str, Decimal]: A dictionary with coin names as keys and their prices as Decimal values.
17
+ """
18
+
19
+
20
+ def get_prices(config: Config) -> Prices:
21
+ result = Prices()
22
+
23
+ coins = pydash.uniq([group.coin for group in config.groups])
24
+ coingecko_ids = pydash.uniq([get_coingecko_id(group) for group in config.groups])
25
+
26
+ url = f"https://api.coingecko.com/api/v3/simple/price?ids={",".join(coingecko_ids)}&vs_currencies=usd"
27
+ for _ in range(RETRIES_COINGECKO_PRICES):
28
+ res = hr(url, proxy=random_str_choice(config.proxies))
29
+ if res.code != 200:
30
+ continue
31
+
32
+ for idx, coin in enumerate(coins):
33
+ if coingecko_ids[idx] in res.json:
34
+ result[coin] = Decimal(str(pydash.get(res.json, f"{coingecko_ids[idx]}.usd")))
35
+ else:
36
+ fatal("Can't get price for {coin} from coingecko, coingecko_id={coingecko_ids[idx]}")
37
+
38
+ return result
39
+
40
+
41
+ def get_coingecko_id(group: Group) -> str:
42
+ if group.coingecko_id:
43
+ return group.coingecko_id
44
+ elif group.network is Network.BTC:
45
+ return "bitcoin"
46
+ elif group.network is Network.ETH and group.token_address is None:
47
+ return "ethereum"
48
+ elif group.coin.lower() == "usdt" or (group.token_address is not None and group.token_address == EthTokenAddress.USDT):
49
+ return "tether"
50
+ elif group.coin.lower() == "usdc" or (group.token_address is not None and group.token_address == EthTokenAddress.USDC):
51
+ return "usd-coin"
52
+ elif group.coin.lower() == "sol":
53
+ return "solana"
54
+
55
+ raise ValueError(f"can't get coingecko_id for {group.coin}")
@@ -0,0 +1,16 @@
1
+ from decimal import Decimal
2
+
3
+ from mm_btc.blockstream import BlockstreamClient
4
+ from mm_std import Ok, Result
5
+
6
+ from mm_balance.constants import RETRIES_BALANCE
7
+
8
+
9
+ def get_balance(address: str, proxies: list[str], round_ndigits: int) -> Result[Decimal]:
10
+ return (
11
+ BlockstreamClient(proxies=proxies, attempts=RETRIES_BALANCE)
12
+ .get_confirmed_balance(address)
13
+ .and_then(
14
+ lambda b: Ok(round(Decimal(b / 100_000_000), round_ndigits)),
15
+ )
16
+ )
@@ -3,9 +3,11 @@ from decimal import Decimal
3
3
  from mm_eth import erc20, rpc
4
4
  from mm_std import Ok, Result
5
5
 
6
+ from mm_balance.constants import RETRIES_BALANCE, RETRIES_DECIMALS, TIMEOUT_BALANCE, TIMEOUT_DECIMALS
7
+
6
8
 
7
9
  def get_native_balance(nodes: list[str], address: str, proxies: list[str], round_ndigits: int) -> Result[Decimal]:
8
- return rpc.eth_get_balance(nodes, address, proxies=proxies, attempts=5, timeout=10).and_then(
10
+ return rpc.eth_get_balance(nodes, address, proxies=proxies, attempts=RETRIES_BALANCE, timeout=TIMEOUT_BALANCE).and_then(
9
11
  lambda b: Ok(round(Decimal(b / 10**18), round_ndigits)),
10
12
  )
11
13
 
@@ -18,29 +20,12 @@ def get_token_balance(
18
20
  token_address,
19
21
  wallet_address,
20
22
  proxies=proxies,
21
- attempts=5,
22
- timeout=10,
23
+ attempts=RETRIES_BALANCE,
24
+ timeout=TIMEOUT_BALANCE,
23
25
  ).and_then(
24
26
  lambda b: Ok(round(Decimal(b / 10**decimals), round_ndigits)),
25
27
  )
26
28
 
27
29
 
28
30
  def get_token_decimals(nodes: list[str], token_address: str, proxies: list[str]) -> Result[int]:
29
- return erc20.get_decimals(nodes, token_address, timeout=10, proxies=proxies, attempts=5)
30
-
31
-
32
- # def get_balance(
33
- # address: str, token_address: str | None, config: Config, progress: Progress | None = None, task_id: TaskID | None = None
34
- # ) -> Result[Decimal]:
35
- # res: Result[Decimal]
36
- #
37
- # if token_address is not None:
38
- #
39
- # else:
40
- # res = rpc.eth_get_balance(config.nodes[Network.ETH], address, proxies=config.proxies, attempts=5, timeout=10).and_then(
41
- # lambda b: Ok(round(Decimal(b / 10 ** 18), config.round_ndigits)),
42
- # )
43
- #
44
- # if task_id is not None and progress is not None:
45
- # progress.update(task_id, advance=1)
46
- # return res
31
+ return erc20.get_decimals(nodes, token_address, timeout=TIMEOUT_DECIMALS, proxies=proxies, attempts=RETRIES_DECIMALS)
@@ -0,0 +1,26 @@
1
+ from decimal import Decimal
2
+
3
+ from mm_solana import balance, token
4
+ from mm_std import Ok, Result
5
+
6
+ from mm_balance.constants import RETRIES_BALANCE, RETRIES_DECIMALS, TIMEOUT_BALANCE, TIMEOUT_DECIMALS
7
+
8
+
9
+ def get_native_balance(nodes: list[str], address: str, proxies: list[str], round_ndigits: int) -> Result[Decimal]:
10
+ return balance.get_balance_with_retries(
11
+ nodes, address, retries=RETRIES_BALANCE, timeout=TIMEOUT_BALANCE, proxies=proxies
12
+ ).and_then(lambda b: Ok(round(Decimal(b / 1_000_000_000), round_ndigits)))
13
+
14
+
15
+ def get_token_balance(
16
+ nodes: list[str], wallet_address: str, token_address: str, decimals: int, proxies: list[str], round_ndigits: int
17
+ ) -> Result[Decimal]:
18
+ return token.get_balance_with_retries(
19
+ nodes, wallet_address, token_address, retries=RETRIES_BALANCE, timeout=TIMEOUT_BALANCE, proxies=proxies
20
+ ).and_then(lambda b: Ok(round(Decimal(b / 10**decimals), round_ndigits)))
21
+
22
+
23
+ def get_token_decimals(nodes: list[str], token_address: str, proxies: list[str]) -> Result[int]:
24
+ return token.get_decimals_with_retries(
25
+ nodes, token_address, retries=RETRIES_DECIMALS, timeout=TIMEOUT_DECIMALS, proxies=proxies
26
+ )
@@ -1,8 +1,8 @@
1
1
  from mm_std import Err, fatal
2
2
 
3
3
  from mm_balance.config import Config
4
+ from mm_balance.constants import Network
4
5
  from mm_balance.rpc import eth, solana
5
- from mm_balance.types import Network
6
6
 
7
7
 
8
8
  class TokenDecimals(dict[Network, dict[str, int]]):
@@ -7,8 +7,8 @@ from mm_std import Ok, PrintFormat, print_table
7
7
 
8
8
  from mm_balance.balances import Balances
9
9
  from mm_balance.config import Config
10
+ from mm_balance.constants import Coin
10
11
  from mm_balance.price import Prices
11
- from mm_balance.types import Coin
12
12
 
13
13
 
14
14
  @dataclass
@@ -925,7 +925,7 @@ wheels = [
925
925
 
926
926
  [[package]]
927
927
  name = "mm-balance"
928
- version = "0.1.10"
928
+ version = "0.1.11"
929
929
  source = { editable = "." }
930
930
  dependencies = [
931
931
  { name = "mm-btc" },
@@ -1,153 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from decimal import Decimal
4
- from typing import Any, Self
5
-
6
- import pydash
7
- from mm_std import BaseConfig, PrintFormat, fatal, hr
8
- from pydantic import Field, field_validator, model_validator
9
-
10
- from mm_balance.types import DEFAULT_ETH_NODES, DEFAULT_SOL_NODES, EthTokenAddress, Network, SolTokenAddress
11
-
12
-
13
- class Config(BaseConfig):
14
- class Group(BaseConfig):
15
- comment: str = ""
16
- coin: str
17
- network: Network
18
- token_address: str | None = None
19
- coingecko_id: str | None = None
20
- addresses: list[str] = Field(default_factory=list)
21
- share: Decimal = Decimal(1)
22
-
23
- @property
24
- def name(self) -> str:
25
- result = self.coin
26
- if self.comment:
27
- result += " / " + self.comment
28
- return result
29
-
30
- @field_validator("coin", mode="after")
31
- def coin_validator(cls, v: str) -> str:
32
- return v.upper()
33
-
34
- @field_validator("addresses", mode="before")
35
- def to_list_validator(cls, v: str | list[str] | None) -> list[str]:
36
- return cls.to_list_str_validator(v, unique=True, remove_comments=True, split_line=True)
37
-
38
- @model_validator(mode="before")
39
- def before_all(cls, data: Any) -> Any:
40
- if "network" not in data:
41
- data["network"] = detect_network(data["coin"])
42
- return data
43
-
44
- @model_validator(mode="after")
45
- def final_validator(self) -> Self:
46
- if self.token_address is None:
47
- self.token_address = detect_token_address(self.coin, self.network)
48
- if self.token_address is not None and self.network is Network.ETH:
49
- self.token_address = self.token_address.lower()
50
- return self
51
-
52
- def process_addresses(self, address_groups: list[Config.AddressGroup]) -> None:
53
- addresses: list[str] = []
54
- for address in self.addresses:
55
- if address_group := pydash.find(address_groups, lambda g: g.name == address): # noqa: B023
56
- addresses.extend(address_group.addresses)
57
- else:
58
- # TODO: check address is valid
59
- addresses.append(address)
60
- self.addresses = addresses
61
-
62
- class AddressGroup(BaseConfig):
63
- name: str
64
- addresses: list[str]
65
-
66
- @field_validator("addresses", mode="before")
67
- def to_list_validator(cls, v: str | list[str] | None) -> list[str]:
68
- return cls.to_list_str_validator(v, unique=True, remove_comments=True, split_line=True)
69
-
70
- groups: list[Group]
71
- addresses: list[AddressGroup] = Field(default_factory=list)
72
-
73
- proxies_url: str | None = None
74
- proxies: list[str] = Field(default_factory=list)
75
- round_ndigits: int = 4
76
- nodes: dict[Network, list[str]] = Field(default_factory=dict)
77
- print_format: PrintFormat = PrintFormat.TABLE
78
- price: bool = True
79
-
80
- workers: dict[Network, int] = {network: 5 for network in Network}
81
-
82
- def btc_groups(self) -> list[Group]:
83
- return [g for g in self.groups if g.network == Network.BTC]
84
-
85
- def eth_groups(self) -> list[Group]:
86
- return [g for g in self.groups if g.network == Network.ETH]
87
-
88
- def sol_groups(self) -> list[Group]:
89
- return [g for g in self.groups if g.network == Network.SOL]
90
-
91
- def has_share(self) -> bool:
92
- return any(g.share != Decimal(1) for g in self.groups)
93
-
94
- @model_validator(mode="after")
95
- def final_validator(self) -> Self:
96
- # load from proxies_url
97
- if self.proxies_url is not None:
98
- self.proxies = get_proxies(self.proxies_url)
99
-
100
- # load addresses from address_group
101
- for group in self.groups:
102
- group.process_addresses(self.addresses)
103
-
104
- # load default rpc nodes
105
- if Network.BTC not in self.nodes:
106
- self.nodes[Network.BTC] = []
107
- if Network.ETH not in self.nodes:
108
- self.nodes[Network.ETH] = DEFAULT_ETH_NODES
109
- if Network.SOL not in self.nodes:
110
- self.nodes[Network.SOL] = DEFAULT_SOL_NODES
111
-
112
- return self
113
-
114
-
115
- def detect_network(coin: str) -> Network:
116
- coin = coin.lower()
117
- if coin == "btc":
118
- return Network.BTC
119
- if coin == "eth":
120
- return Network.ETH
121
- if coin == "sol":
122
- return Network.SOL
123
- return Network.ETH
124
- # raise ValueError(f"can't get network for the coin: {coin}")
125
-
126
-
127
- def detect_token_address(coin: str, network: str) -> str | None:
128
- if network == Network.ETH.lower():
129
- if coin.lower() == "usdt":
130
- return EthTokenAddress.USDT
131
- if coin.lower() == "usdc":
132
- return EthTokenAddress.USDC
133
-
134
- if network == Network.SOL.lower():
135
- if coin.lower() == "usdt":
136
- return SolTokenAddress.USDT
137
- if coin.lower() == "usdc":
138
- return SolTokenAddress.USDC
139
-
140
-
141
- def get_proxies(proxies_url: str) -> list[str]:
142
- try:
143
- res = hr(proxies_url)
144
- if res.is_error():
145
- fatal(f"Can't get proxies: {res.error}")
146
- proxies = [p.strip() for p in res.body.splitlines() if p.strip()]
147
- return pydash.uniq(proxies)
148
- except Exception as err:
149
- fatal(f"Can't get proxies: {err}")
150
-
151
-
152
- def get_address_group_by_name(address_groups: list[Config.AddressGroup], name: str) -> Config.AddressGroup | None:
153
- return pydash.find(address_groups, lambda g: g.name == name)
@@ -1,83 +0,0 @@
1
- import time
2
- from decimal import Decimal
3
-
4
- import pydash
5
- from mm_std import Err, Ok, Result, fatal, hr
6
- from mm_std.random_ import random_str_choice
7
-
8
- from mm_balance import output
9
- from mm_balance.config import Config
10
- from mm_balance.types import EthTokenAddress, Network
11
-
12
-
13
- class Prices(dict[str, Decimal]):
14
- """
15
- A Prices class representing a mapping from coin names to their prices.
16
-
17
- Inherits from:
18
- Dict[str, Decimal]: A dictionary with coin names as keys and their prices as Decimal values.
19
- """
20
-
21
-
22
- def get_prices(config: Config) -> Prices:
23
- result = Prices()
24
- coins_total = len(pydash.uniq([group.coin for group in config.groups]))
25
-
26
- progress = output.create_progress_bar()
27
-
28
- with progress:
29
- task_id = output.create_progress_task(progress, "prices", total=coins_total)
30
-
31
- for group in config.groups:
32
- if group.coin in result:
33
- continue
34
-
35
- coingecko_id = get_coingecko_id(group)
36
- res = get_asset_price(coingecko_id, config.proxies)
37
- if isinstance(res, Ok):
38
- result[group.coin] = res.ok
39
- progress.update(task_id, advance=1)
40
- else:
41
- fatal(res.err)
42
- # raise ValueError(res.err)
43
-
44
- return result
45
-
46
-
47
- def get_asset_price(coingecko_asset_id: str, proxies: list[str]) -> Result[Decimal]:
48
- url = f"https://api.coingecko.com/api/v3/simple/price?ids={coingecko_asset_id}&vs_currencies=usd"
49
- data = None
50
- error = f"error: can't get price for {coingecko_asset_id} from coingecko"
51
- for _ in range(3):
52
- res = hr(url, proxy=random_str_choice(proxies))
53
-
54
- # Check for Rate Limit
55
- if res.code == 429:
56
- error = f"error: can't get price for {coingecko_asset_id} from coingecko. You've exceeded the Rate Limit. Please add more proxies." # noqa: E501
57
- if not proxies:
58
- fatal(error) # Exit immidiately if no proxies are provided
59
-
60
- data = res.to_dict()
61
- if res.json and coingecko_asset_id in coingecko_asset_id in res.json:
62
- return Ok(Decimal(pydash.get(res.json, f"{coingecko_asset_id}.usd")))
63
-
64
- if not proxies:
65
- time.sleep(10)
66
- return Err(error, data=data)
67
-
68
-
69
- def get_coingecko_id(group: Config.Group) -> str:
70
- if group.coingecko_id:
71
- return group.coingecko_id
72
- elif group.network is Network.BTC:
73
- return "bitcoin"
74
- elif group.network is Network.ETH and group.token_address is None:
75
- return "ethereum"
76
- elif group.coin.lower() == "usdt" or (group.token_address is not None and group.token_address == EthTokenAddress.USDT):
77
- return "tether"
78
- elif group.coin.lower() == "usdc" or (group.token_address is not None and group.token_address == EthTokenAddress.USDC):
79
- return "usd-coin"
80
- elif group.coin.lower() == "sol":
81
- return "solana"
82
-
83
- raise ValueError(f"can't get coingecko_id for {group.coin}")
@@ -1,20 +0,0 @@
1
- from decimal import Decimal
2
-
3
- from mm_btc.blockstream import BlockstreamClient
4
- from mm_std import Ok, Result
5
- from rich.progress import Progress, TaskID
6
-
7
- from mm_balance.config import Config
8
-
9
-
10
- def get_balance(address: str, config: Config, progress: Progress | None = None, task_id: TaskID | None = None) -> Result[Decimal]:
11
- res: Result[Decimal] = (
12
- BlockstreamClient(proxies=config.proxies, attempts=3)
13
- .get_confirmed_balance(address)
14
- .and_then(
15
- lambda b: Ok(round(Decimal(b / 100_000_000), config.round_ndigits)),
16
- )
17
- )
18
- if task_id is not None and progress is not None:
19
- progress.update(task_id, advance=1)
20
- return res
@@ -1,22 +0,0 @@
1
- from decimal import Decimal
2
-
3
- from mm_solana import balance, token
4
- from mm_std import Ok, Result
5
-
6
-
7
- def get_native_balance(nodes: list[str], address: str, proxies: list[str], round_ndigits: int) -> Result[Decimal]:
8
- return balance.get_balance_with_retries(nodes, address, retries=5, timeout=5, proxies=proxies).and_then(
9
- lambda b: Ok(round(Decimal(b / 1_000_000_000), round_ndigits)),
10
- )
11
-
12
-
13
- def get_token_balance(
14
- nodes: list[str], wallet_address: str, token_address: str, decimals: int, proxies: list[str], round_ndigits: int
15
- ) -> Result[Decimal]:
16
- return token.get_balance_with_retries(nodes, wallet_address, token_address, retries=5, timeout=5, proxies=proxies).and_then(
17
- lambda b: Ok(round(Decimal(b / 10**decimals), round_ndigits))
18
- )
19
-
20
-
21
- def get_token_decimals(nodes: list[str], token_address: str, proxies: list[str]) -> Result[int]:
22
- return token.get_decimals_with_retries(nodes, token_address, retries=5, timeout=5, proxies=proxies)
File without changes
File without changes
File without changes