mm-btc 0.2.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

mm_btc/blockstream.py

@@ -1,5 +1,4 @@
 from collections.abc import Sequence
-from typing import TypeAlias
 
 from mm_std import Err, HResponse, Ok, Result, hr
 from mm_std.random_ import random_str_choice
@@ -8,10 +7,12 @@ from pydantic import BaseModel
 MAINNET_BASE_URL = "https://blockstream.info/api"
 TESTNET_BASE_URL = "https://blockstream.info/testnet/api"
 
-ERROR_INVALID_ADDRESS = "INVALID_ADDRESS"
-ERROR_INVALID_NETWORK = "INVALID_NETWORK"
+# ERROR_INVALID_ADDRESS = "INVALID_ADDRESS"
+# ERROR_INVALID_NETWORK = "INVALID_NETWORK"
 
-Proxy: TypeAlias = str | Sequence[str] | None
+ERROR_400_BAD_REQUEST = "400 Bad Request"
+
+type Proxy = str | Sequence[str] | None
 
 
 class Mempool(BaseModel):
@@ -54,7 +55,7 @@ class Utxo(BaseModel):
 
 
 class BlockstreamClient:
-    def __init__(self, testnet: bool = False, timeout: int = 10, proxies: Proxy = None, attempts: int = 1):
+    def __init__(self, testnet: bool = False, timeout: int = 10, proxies: Proxy = None, attempts: int = 1) -> None:
         self.testnet = testnet
         self.timeout = timeout
         self.proxies = proxies
@@ -68,12 +69,8 @@ class BlockstreamClient:
             try:
                 res = self._request(f"/address/{address}")
                 data = res.to_dict()
-                if res.code == 400 and (
-                    "invalid bitcoin address" in res.body.lower() or "bech32 segwit decoding error" in res.body.lower()
-                ):
-                    return Err(ERROR_INVALID_ADDRESS, data=data)
-                elif res.code == 400 and "invalid network" in res.body.lower():
-                    return Err(ERROR_INVALID_NETWORK, data=data)
+                if res.code == 400:
+                    return Err("400 Bad Request", data=data)
                 return Ok(Address(**res.json), data=data)
             except Exception as err:
                 result = Err(err, data=data)

mm_btc/cli/cli.py

@@ -0,0 +1,80 @@
+import importlib.metadata
+from pathlib import Path
+from typing import Annotated
+
+import typer
+import typer.core
+from mm_std import print_plain
+
+from mm_btc.wallet import AddressType
+
+from .cmd import address_cmd, create_tx_cmd, decode_tx_cmd, mnemonic_cmd, utxo_cmd
+
+app = typer.Typer(no_args_is_help=True, pretty_exceptions_enable=False, add_completion=False)
+
+
+@app.command("mnemonic")
+@app.command(name="m", hidden=True)
+def mnemonic_command(  # nosec B107:hardcoded_password_default
+    mnemonic: Annotated[str, typer.Option("--mnemonic", "-m", help="")] = "",
+    passphrase: Annotated[str, typer.Option("--passphrase", "-p")] = "",
+    path: Annotated[str, typer.Option("--path", help="Derivation path. Examples: bip44, bip88, m/44'/0'/0'/0")] = "bip44",
+    address_type: Annotated[AddressType, typer.Option("--address-type", "-a", help="Bitcoin address type")] = AddressType.P2WPKH,
+    hex_: Annotated[bool, typer.Option("--hex", help="Print private key in hex format instead of WIF")] = False,
+    words: int = typer.Option(12, "--words", "-w", help="Number of mnemonic words"),
+    limit: int = typer.Option(10, "--limit", "-l"),
+    testnet: bool = typer.Option(False, "--testnet", "-t", help="Testnet network"),
+) -> None:
+    """Generate keys based on a mnemonic"""
+    mnemonic_cmd.run(
+        mnemonic_cmd.Args(
+            mnemonic=mnemonic,
+            passphrase=passphrase,
+            words=words,
+            limit=limit,
+            path=path,
+            address_type=address_type,
+            hex=hex_,
+            testnet=testnet,
+        )
+    )
+
+
+@app.command(name="address")
+@app.command(name="a", hidden=True)
+def address_command(address: str) -> None:
+    """Get address info from Blockstream API"""
+    address_cmd.run(address)
+
+
+@app.command("create-tx")
+def create_tx_command(config_path: Annotated[Path, typer.Argument(exists=True)]) -> None:
+    """Create a transaction"""
+    create_tx_cmd.run(config_path)
+
+
+@app.command("decode-tx")
+def decode_tx_command(tx_hex: str, testnet: Annotated[bool, typer.Option("--testnet", "-t")] = False) -> None:
+    """Decode a transaction"""
+    decode_tx_cmd.run(tx_hex, testnet)
+
+
+@app.command("utxo")
+def utxo_command(address: str) -> None:
+    """Get UTXOs from Blockstream API"""
+    utxo_cmd.run(address)
+
+
+def version_callback(value: bool) -> None:
+    if value:
+        print_plain(f"mm-btc: v{importlib.metadata.version('mm-btc')}")
+        raise typer.Exit
+
+
+@app.callback()
+def main(_version: bool = typer.Option(None, "--version", callback=version_callback, is_eager=True)) -> None:
+    pass
+
+
+if __name__ == "__main_":
+    app()

mm_btc/cli/cmd/address_cmd.py

@@ -0,0 +1,10 @@
+from mm_std import print_json
+
+from mm_btc.blockstream import BlockstreamClient
+from mm_btc.wallet import is_testnet_address
+
+
+def run(address: str) -> None:
+    client = BlockstreamClient(testnet=is_testnet_address(address))
+    res = client.get_address(address)
+    print_json(res)

mm_btc/cli/cmd/create_tx_cmd.py

@@ -0,0 +1,27 @@
+from pathlib import Path
+
+from bit import PrivateKey, PrivateKeyTestnet
+from mm_std import BaseConfig, print_console
+
+from mm_btc.wallet import is_testnet_address
+
+
+class Config(BaseConfig):
+    class Output(BaseConfig):
+        address: str
+        amount: int
+
+    from_address: str
+    private: str
+    outputs: list[Output]
+
+
+def run(config_path: Path) -> None:
+    config = Config.read_toml_config_or_exit(config_path)
+    testnet = is_testnet_address(config.from_address)
+    key = PrivateKeyTestnet(config.private) if testnet else PrivateKey(config.private)
+
+    outputs = [(o.address, o.amount, "satoshi") for o in config.outputs]
+
+    tx = key.create_transaction(outputs)
+    print_console(tx)

mm_btc/cli/cmd/decode_tx_cmd.py

@@ -0,0 +1,8 @@
+from mm_std import print_json
+
+from mm_btc.tx import decode_tx
+
+
+def run(tx_hex: str, testnet: bool = False) -> None:
+    res = decode_tx(tx_hex, testnet)
+    print_json(res)

mm_btc/cli/cmd/mnemonic_cmd.py

@@ -0,0 +1,49 @@
+from dataclasses import dataclass
+from enum import Enum
+
+from mm_std import print_plain
+
+from mm_btc.wallet import AddressType, derive_accounts, generate_mnemonic
+
+
+class PrivateType(str, Enum):
+    hex = "hex"
+    wif = "wif"
+
+
+@dataclass
+class Args:
+    mnemonic: str
+    passphrase: str
+    words: int
+    limit: int
+    hex: bool  # Print private key in hex format instead of WIF
+    address_type: AddressType
+    path: str
+    testnet: bool
+
+
+def run(args: Args) -> None:
+    mnemonic = args.mnemonic or generate_mnemonic()
+    passphrase = args.passphrase
+    path = get_derivation_path_prefix(args.path, args.testnet)
+    accounts = derive_accounts(mnemonic, passphrase, path, args.address_type, args.limit)
+
+    print_plain(f"{mnemonic}")
+    if passphrase:
+        print_plain(f"{passphrase}")
+    for acc in accounts:
+        private = acc.private if args.hex else acc.wif
+        print_plain(f"{acc.path} {acc.address} {private}")
+
+
+def get_derivation_path_prefix(path: str, testnet: bool) -> str:
+    if path.startswith("m/"):
+        return path
+    coin = "1" if testnet else "0"
+    if path == "bip44":
+        return f"m/44'/{coin}'/0'/0"
+    if path == "bip84":
+        return f"m/84'/{coin}'/0'/0"
+
+    raise ValueError("Invalid path")

mm_btc/cli/cmd/utxo_cmd.py

@@ -0,0 +1,11 @@
+from mm_std import print_json
+
+from mm_btc.blockstream import BlockstreamClient
+from mm_btc.wallet import is_testnet_address
+
+
+def run(address: str) -> None:
+    client = BlockstreamClient(testnet=is_testnet_address(address))
+    res = client.get_utxo(address)
+
+    print_json(res.ok_or_err())

mm_btc/wallet.py

@@ -39,13 +39,9 @@ def generate_mnemonic(language: str = "english", words: int = 12) -> str:
 
 def derive_accounts(mnemonic: str, passphrase: str, path_prefix: str, address_type: AddressType, limit: int) -> list[Account]:
     coin = Bitcoin
-    if path_prefix.startswith("m/84'/1'"):
+    if path_prefix.startswith(("m/84'/1'", "m/44'/1'")):
         network = coin.NETWORKS.TESTNET
-    elif path_prefix.startswith("m/44'/1'"):
-        network = coin.NETWORKS.TESTNET
-    elif path_prefix.startswith("m/84'/0'"):
-        network = coin.NETWORKS.MAINNET
-    elif path_prefix.startswith("m/44'/0'"):
+    elif path_prefix.startswith(("m/84'/0'", "m/44'/0'")):
         network = coin.NETWORKS.MAINNET
     else:
         raise ValueError("Invalid path")

mm_btc-0.3.0.dist-info/METADATA

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: mm-btc
+Version: 0.3.0
+Requires-Python: >=3.12
+Requires-Dist: bitcoinlib~=0.7.1
+Requires-Dist: bit~=0.8.0
+Requires-Dist: hdwallet~=3.2.3
+Requires-Dist: mm-crypto-utils>=0.1.3
+Requires-Dist: mnemonic~=0.21
+Requires-Dist: typer~=0.15.1

mm_btc-0.3.0.dist-info/RECORD

@@ -0,0 +1,17 @@
+mm_btc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mm_btc/blockstream.py,sha256=hnZ4pSPUbXCoPA0zPaEazwibDvy-WXHCXk0hrix-grA,3347
+mm_btc/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mm_btc/tx.py,sha256=7KTMNuL1n8rRj_245BV1bH9M2_PctNvlvPsLEsRTYx4,245
+mm_btc/wallet.py,sha256=_pGTuAf6Y9KzaWTTSg_tn6WEoRsGqhxJuPof0_xyrmM,2333
+mm_btc/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mm_btc/cli/cli.py,sha256=-UQ_StfyHS0p5gSD7bWWsibjJh3FJDSaWkVnlRUvZds,2615
+mm_btc/cli/cmd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mm_btc/cli/cmd/address_cmd.py,sha256=fipX9FiQ2EgCEq8nVN6ELnDMoR5KTIX7bpAaRK3t_yg,284
+mm_btc/cli/cmd/create_tx_cmd.py,sha256=QHhfA91FGWf5r6DkWTaXInZJi6HYLfxueA1GtcBD2-s,703
+mm_btc/cli/cmd/decode_tx_cmd.py,sha256=0jGlUjnA1X2Q2aYwSBeAjqVNZIBsW5X2lEwIpSfWJsU,175
+mm_btc/cli/cmd/mnemonic_cmd.py,sha256=CY6tPsqOTNfM-4WhKDhqitCi2OeqSIUMEUTFIRGHwIg,1254
+mm_btc/cli/cmd/utxo_cmd.py,sha256=mp-lVLURpXFqO3IeBIEwnHusEvT5tFjUKxnRYC-rFqQ,294
+mm_btc-0.3.0.dist-info/METADATA,sha256=07SoipqPS9Ogpdp2jM5tu29r9I6xX07NQ837ijTchkY,261
+mm_btc-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mm_btc-0.3.0.dist-info/entry_points.txt,sha256=KphzLNE9eb9H1DO-L0gvBQaQT5ImedyVCN4a6svfiRg,46
+mm_btc-0.3.0.dist-info/RECORD,,

mm_btc-0.3.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+mm-btc = mm_btc.cli.cli:app

mm_btc-0.2.0.dist-info/METADATA

@@ -1,9 +0,0 @@
-Metadata-Version: 2.4
-Name: mm-btc
-Version: 0.2.0
-Requires-Python: >=3.12
-Requires-Dist: bitcoinlib~=0.7.1
-Requires-Dist: bit~=0.8.0
-Requires-Dist: hdwallet~=3.2.0
-Requires-Dist: mm-std~=0.1.9
-Requires-Dist: mnemonic>=0.21

mm_btc-0.2.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-mm_btc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mm_btc/blockstream.py,sha256=U2_pc6RYAkFcCvH_WrxjIm13VNavyY3WgK8ePowV-Ic,3620
-mm_btc/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mm_btc/tx.py,sha256=7KTMNuL1n8rRj_245BV1bH9M2_PctNvlvPsLEsRTYx4,245
-mm_btc/wallet.py,sha256=E9XKZMQx48JiGRDG8ij7cbilNPVmcpmS6jAGBzXfV9U,2475
-mm_btc-0.2.0.dist-info/METADATA,sha256=drZ2vHDl89rFCn6SJDDaELptn97EiCTGnXIwD92rWXg,223
-mm_btc-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mm_btc-0.2.0.dist-info/RECORD,,