mm-apt 0.3.3__tar.gz → 0.3.5__tar.gz

{mm_apt-0.3.3 → mm_apt-0.3.5}/PKG-INFO

@@ -1,5 +1,5 @@
 Metadata-Version: 2.4
 Name: mm-apt
-Version: 0.3.3
+Version: 0.3.5
 Requires-Python: >=3.12
-Requires-Dist: mm-crypto-utils>=0.3.5
+Requires-Dist: mm-crypto-utils>=0.3.7

{mm_apt-0.3.3 → mm_apt-0.3.5}/pyproject.toml

@@ -1,10 +1,10 @@
 [project]
 name = "mm-apt"
-version = "0.3.3"
+version = "0.3.5"
 description = ""
 requires-python = ">=3.12"
 dependencies = [
-    "mm-crypto-utils>=0.3.5",
+    "mm-crypto-utils>=0.3.7",
 ]
 
 [build-system]
@@ -15,7 +15,7 @@ build-backend = "hatchling.build"
 dev-dependencies = [
     "pytest~=8.3.5",
     "pytest-xdist~=3.6.1",
-    "ruff~=0.11.6",
+    "ruff~=0.11.8",
     "pip-audit~=2.9.0",
     "bandit~=1.8.3",
     "mypy~=1.15.0",

mm_apt-0.3.5/src/mm_apt/account.py

@@ -0,0 +1,36 @@
+import re
+
+# Maximum allowable Aptos account address value (256 bits)
+MAX_APTOS_ADDRESS = 2**256
+
+
+def is_valid_address(address: str) -> bool:
+    """
+    Check if the address is a valid Aptos account address.
+
+    Requirements:
+    - Must be a 32-byte (64 hex characters) string.
+    - Must be an entire 64-character hex string, padded with leading zeros as needed.
+    - Optional '0x' or '0X' prefix is allowed.
+    - Numeric value must be < 2**256.
+    """
+    # Ensure input is a string
+    if not isinstance(address, str):
+        return False
+
+    # Remove optional prefix
+    hex_part = address[2:] if address.startswith(("0x", "0X")) else address
+
+    # Must be exactly 64 hex characters
+    if len(hex_part) != 64:
+        return False
+    if not re.fullmatch(r"[0-9a-fA-F]{64}", hex_part):
+        return False
+
+    # Convert to integer and check range
+    try:
+        value = int(hex_part, 16)
+    except ValueError:
+        return False
+
+    return 0 <= value < MAX_APTOS_ADDRESS

mm_apt-0.3.5/src/mm_apt/retry.py

@@ -0,0 +1,12 @@
+from mm_crypto_utils import Nodes, Proxies, retry_with_node_and_proxy
+from mm_std import Result
+
+from mm_apt import balance
+
+
+async def get_balance(
+    retries: int, nodes: Nodes, proxies: Proxies, *, account: str, coin_type: str, timeout: float = 5
+) -> Result[int]:
+    return await retry_with_node_and_proxy(
+        retries, nodes, proxies, lambda node, proxy: balance.get_balance(node, account, coin_type, timeout, proxy)
+    )

mm_apt-0.3.5/tests/test_account.py

@@ -0,0 +1,55 @@
+import pytest
+
+from mm_apt.account import is_valid_address
+
+
+# Helper to generate a full 64-char hex string with a given repeat char
+def full_hex(ch):
+    return ch * 64
+
+
+def test_valid_full_length_addresses():
+    # All zeros
+    assert is_valid_address("0x" + full_hex("0")) is True
+    # Mixed zeros and ones
+    assert is_valid_address("0x" + "0" * 63 + "1") is True
+    # All Fs (max minus 1)
+    assert is_valid_address("0x" + full_hex("f")) is True
+    # Uppercase hex
+    assert is_valid_address("0X" + full_hex("A")) is True
+    # Without prefix
+    assert is_valid_address(full_hex("1")) is True
+
+
+def test_invalid_full_length_addresses():
+    # Too short
+    short_hex = "0x" + "1" * 63
+    assert is_valid_address(short_hex) is False
+    # Too long
+    long_hex = "0x" + "f" * 65
+    assert is_valid_address(long_hex) is False
+    # Invalid character
+    bad_char = "0x" + "g" + "0" * 63
+    assert is_valid_address(bad_char) is False
+    # Numeric type
+    assert is_valid_address(123) is False
+
+
+def test_address_out_of_range():
+    # Exactly 2**256 is out of range -> 1 followed by 64 zeros in hex is too large (65 hex digits)
+    out_of_range = "0x1" + "0" * 64
+    # It's invalid by length then by range
+    assert is_valid_address(out_of_range) is False
+    # Highest valid: 2**256 - 1 -> 64 hex 'f'
+    max_valid = "0x" + full_hex("f")
+    assert is_valid_address(max_valid) is True
+
+
+def test_missing_prefix_short_address():
+    # Even valid numeric value but missing prefix and not full-length
+    assert is_valid_address("1" * 1) is False
+    assert is_valid_address("a" * 10) is False
+
+
+if __name__ == "__main__":
+    pytest.main()