mm-apt 0.3.1__tar.gz

mm_apt-0.3.1/.env.example

@@ -0,0 +1 @@
+MAINNET_RPC_URL=https://fullnode.mainnet.aptoslabs.com/v1

mm_apt-0.3.1/.gitignore

@@ -0,0 +1,14 @@
+.idea
+.venv
+.env
+.coverage
+/htmlcov
+__pycache__
+*.egg-info
+pip-wheel-metadata
+.pytest_cache
+.mypy_cache
+.ruff_cache
+/dist
+/build
+/tmp

mm_apt-0.3.1/PKG-INFO

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

mm_apt-0.3.1/README.txt

@@ -0,0 +1,2 @@
+https://aptos.dev/
+https://aptoscan.com/

mm_apt-0.3.1/dict.dic

@@ -0,0 +1 @@
+ndigits

mm_apt-0.3.1/justfile

@@ -0,0 +1,33 @@
+set dotenv-load
+version := `uv run python -c 'import tomllib; print(tomllib.load(open("pyproject.toml", "rb"))["project"]["version"])'`
+
+
+clean:
+    rm -rf .pytest_cache .mypy_cache .ruff_cache .coverage dist build src/*.egg-info
+
+build: clean lint audit test
+    uv build
+
+format:
+    uv run ruff check --select I --fix src tests
+    uv run ruff format src tests
+
+test:
+    uv run pytest -n auto tests
+
+lint: format
+    uv run ruff check src tests
+    uv run mypy src
+
+audit:
+    uv run pip-audit
+    uv run bandit -r -c "pyproject.toml" src
+
+publish: build
+    git diff-index --quiet HEAD
+    uvx twine upload dist/**
+    git tag -a 'v{{version}}' -m 'v{{version}}'
+    git push origin v{{version}}
+
+sync:
+    uv sync

mm_apt-0.3.1/pyproject.toml

@@ -0,0 +1,72 @@
+[project]
+name = "mm-apt"
+version = "0.3.1"
+description = ""
+requires-python = ">=3.12"
+dependencies = [
+    "mm-crypto-utils>=0.3.1",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.uv]
+dev-dependencies = [
+    "pytest~=8.3.5",
+    "pytest-xdist~=3.6.1",
+    "ruff~=0.11.5",
+    "pip-audit~=2.9.0",
+    "bandit~=1.8.3",
+    "mypy~=1.15.0",
+    "pytest-asyncio>=0.26.0",
+]
+
+[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 = [
+    "TC", # flake8-type-checking, TYPE_CHECKING is dangerous, for example it doesn't work with pydantic
+    "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
+    "ASYNC109",
+    "G004"
+]
+[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"]
+
+[tool.pytest.ini_options]
+asyncio_default_fixture_loop_scope = "function"

mm_apt-0.3.1/src/mm_apt/ans.py

@@ -0,0 +1,27 @@
+from mm_std import Result, http_request
+
+
+async def address_to_name(address: str, timeout: float = 5, proxy: str | None = None) -> Result[str | None]:
+    url = f"https://www.aptosnames.com/api/mainnet/v1/name/{address}"
+    res = await http_request(url, proxy=proxy, timeout=timeout)
+    if res.is_error():
+        return res.to_result_failure()
+    json_res = res.parse_json_body()
+    if res.status_code == 200 and json_res == {}:
+        return res.to_result_success(None)
+    if "name" in json_res:
+        return res.to_result_success(json_res["name"])
+    return res.to_result_failure("unknown_response")
+
+
+async def address_to_primary_name(address: str, timeout: float = 5, proxy: str | None = None) -> Result[str | None]:
+    url = f"https://www.aptosnames.com/api/mainnet/v1/primary-name/{address}"
+    res = await http_request(url, proxy=proxy, timeout=timeout)
+    if res.is_error():
+        return res.to_result_failure()
+    json_res = res.parse_json_body()
+    if res.status_code == 200 and json_res == {}:
+        return res.to_result_success(None)
+    if "name" in json_res:
+        return res.to_result_success(json_res["name"])
+    return res.to_result_failure("unknown_response")

mm_apt-0.3.1/src/mm_apt/balance.py

@@ -0,0 +1,13 @@
+from mm_std import Result, http_request
+
+
+async def get_balance(node: str, account: str, coin_type: str, timeout: float = 5, proxy: str | None = None) -> Result[int]:
+    url = f"{node}/accounts/{account}/resource/0x1::coin::CoinStore%3C{coin_type}%3E"
+    res = await http_request(url, proxy=proxy, timeout=timeout)
+    try:
+        json_res = res.parse_json_body()
+        if json_res.get("error_code") == "resource_not_found":
+            return res.to_result_success(0)
+        return res.to_result_success(int(json_res["data"]["coin"]["value"]))
+    except Exception as e:
+        return res.to_result_failure(e)

mm_apt-0.3.1/src/mm_apt/coin.py

@@ -0,0 +1,2 @@
+APTOS_COIN_TYPE = "0x1::aptos_coin::AptosCoin"
+APTOS_COIN_DECIMALS = 8

mm_apt-0.3.1/tests/conftest.py

@@ -0,0 +1,17 @@
+import pytest
+from mm_std import get_dotenv
+
+
+@pytest.fixture
+def anyio_backend() -> str:
+    return "asyncio"
+
+
+@pytest.fixture
+def mainnet_rpc_url() -> str:
+    return get_dotenv("MAINNET_RPC_URL")
+
+
+@pytest.fixture
+def okx_address() -> str:
+    return "0x834d639b10d20dcb894728aa4b9b572b2ea2d97073b10eacb111f338b20ea5d7"

mm_apt-0.3.1/tests/test_ans.py

@@ -0,0 +1,21 @@
+import pytest
+
+from mm_apt import ans
+
+pytestmark = pytest.mark.asyncio
+
+
+async def test_address_to_primary_name():
+    address = "0x68e6982c788b50e3caccc834a4764763381d7201be996914e1310139a35854ed"
+    assert (await ans.address_to_primary_name(address)).unwrap() == "vitalik"
+
+    address = "0xabfabdec0732564bd906fb94e467410a131c6e6040f7bca860458e2026e3b14e"
+    assert (await ans.address_to_primary_name(address)).unwrap() is None
+
+
+async def test_address_to_name_async():
+    address = "0x68e6982c788b50e3caccc834a4764763381d7201be996914e1310139a35854ed"
+    assert (await ans.address_to_name(address)).ok == "vitalik"
+
+    address = "0xabfabdec0732564bd906fb94e467410a131c6e6040f7bca860458e2026e3b14e"
+    assert (await ans.address_to_primary_name(address)).ok is None

mm_apt-0.3.1/tests/test_balance.py

@@ -0,0 +1,11 @@
+import pytest
+
+from mm_apt.balance import get_balance
+from mm_apt.coin import APTOS_COIN_TYPE
+
+pytestmark = pytest.mark.asyncio
+
+
+async def test_get_balance(mainnet_rpc_url, okx_address):
+    res = await get_balance(mainnet_rpc_url, okx_address, APTOS_COIN_TYPE)
+    assert res.unwrap() > 1000