bittensor-cli 9.0.2__py3-none-any.whl → 9.1.0__py3-none-any.whl

bittensor_cli/src/bittensor/subtensor_interface.py

@@ -20,6 +20,7 @@ from bittensor_cli.src.bittensor.chain_data import (
     decode_hex_identity,
     DynamicInfo,
     SubnetState,
+    MetagraphInfo,
 )
 from bittensor_cli.src import DelegatesDetails
 from bittensor_cli.src.bittensor.balances import Balance, fixed_to_float
@@ -1252,6 +1253,38 @@ class SubtensorInterface:
         else:
             return Balance.from_rao(_result).set_unit(int(netuid))
 
+    async def get_metagraph_info(
+        self, netuid: int, block_hash: Optional[str] = None
+    ) -> Optional[MetagraphInfo]:
+        hex_bytes_result = await self.query_runtime_api(
+            runtime_api="SubnetInfoRuntimeApi",
+            method="get_metagraph",
+            params=[netuid],
+            block_hash=block_hash,
+        )
+
+        if hex_bytes_result is None:
+            return None
+
+        try:
+            bytes_result = bytes.fromhex(hex_bytes_result[2:])
+        except ValueError:
+            bytes_result = bytes.fromhex(hex_bytes_result)
+
+        return MetagraphInfo.from_any(bytes_result)
+
+    async def get_all_metagraphs_info(
+        self, block_hash: Optional[str] = None
+    ) -> list[MetagraphInfo]:
+        hex_bytes_result = await self.query_runtime_api(
+            runtime_api="SubnetInfoRuntimeApi",
+            method="get_all_metagraphs",
+            params=[],
+            block_hash=block_hash,
+        )
+
+        return MetagraphInfo.list_from_any(hex_bytes_result)
+
     async def multi_get_stake_for_coldkey_and_hotkey_on_netuid(
         self,
         hotkey_ss58s: list[str],
@@ -1327,20 +1360,29 @@ class SubtensorInterface:
         This function is useful for analyzing the stake distribution and delegation patterns of multiple
         accounts simultaneously, offering a broader perspective on network participation and investment strategies.
         """
-        result = await self.query_runtime_api(
-            runtime_api="StakeInfoRuntimeApi",
-            method="get_stake_info_for_coldkeys",
-            params=[coldkey_ss58_list],
-            block_hash=block_hash,
-        )
-        if result is None:
-            return None
+        BATCH_SIZE = 60
 
+        tasks = []
+        for i in range(0, len(coldkey_ss58_list), BATCH_SIZE):
+            ss58_chunk = coldkey_ss58_list[i : i + BATCH_SIZE]
+            tasks.append(
+                self.query_runtime_api(
+                    runtime_api="StakeInfoRuntimeApi",
+                    method="get_stake_info_for_coldkeys",
+                    params=[ss58_chunk],
+                    block_hash=block_hash,
+                )
+            )
+        results = await asyncio.gather(*tasks)
         stake_info_map = {}
-        for coldkey_bytes, stake_info_list in result:
-            coldkey_ss58 = decode_account_id(coldkey_bytes)
-            stake_info_map[coldkey_ss58] = StakeInfo.list_from_any(stake_info_list)
-        return stake_info_map
+        for result in results:
+            if result is None:
+                continue
+            for coldkey_bytes, stake_info_list in result:
+                coldkey_ss58 = decode_account_id(coldkey_bytes)
+                stake_info_map[coldkey_ss58] = StakeInfo.list_from_any(stake_info_list)
+
+        return stake_info_map if stake_info_map else None
 
     async def all_subnets(self, block_hash: Optional[str] = None) -> list[DynamicInfo]:
         result = await self.query_runtime_api(

bittensor_cli/src/bittensor/utils.py

@@ -1279,14 +1279,25 @@ def print_linux_dependency_message():
     """Prints the WebKit dependency message for Linux systems."""
     console.print("[red]This command requires WebKit dependencies on Linux.[/red]")
     console.print(
-        "\nPlease install the required packages using one of the following commands based on your distribution:"
+        "\nPlease make sure these packages are installed on your system for PyWry to work:"
     )
     console.print("\nArch Linux / Manjaro:")
     console.print("[green]sudo pacman -S webkit2gtk[/green]")
     console.print("\nDebian / Ubuntu:")
     console.print("[green]sudo apt install libwebkit2gtk-4.0-dev[/green]")
+    console.print("\nNote for Ubuntu 24.04+ & Debian 13+:")
+    console.print("You may need these additional steps to install libwebkit2gtk:")
+    console.print(
+        "\tCreate a new source file with: [green]sudo vim /etc/apt/sources.list.d/jammy-temp.list[/green]"
+    )
+    console.print(
+        "\tAdd this into the file and save: [green]deb http://archive.ubuntu.com/ubuntu jammy main universe[/green]"
+    )
+    console.print(
+        "\tUpdate the repository and install the webkit dependency: [green]sudo apt update && sudo apt install libwebkit2gtk-4.0-dev[/green]"
+    )
     console.print("\nFedora / CentOS / AlmaLinux:")
-    console.print("[green]sudo dnf install gtk3-devel webkit2gtk3-devel[/green]")
+    console.print("[green]sudo dnf install gtk3-devel webkit2gtk3-devel[/green]\n\n")
 
 
 def is_linux():

bittensor_cli/src/commands/stake/add.py

@@ -17,6 +17,7 @@ from bittensor_cli.src.bittensor.utils import (
     is_valid_ss58_address,
     print_error,
     print_verbose,
+    unlock_key,
 )
 from bittensor_wallet import Wallet
 from bittensor_wallet.errors import KeyFileError
@@ -338,10 +339,7 @@ async def stake_add(
     if prompt:
         if not Confirm.ask("Would you like to continue?"):
             raise typer.Exit()
-    try:
-        wallet.unlock_coldkey()
-    except KeyFileError:
-        err_console.print("Error decrypting coldkey (possibly incorrect password)")
+    if not unlock_key(wallet).success:
         return False
 
     if safe_staking:

bittensor_cli/src/commands/stake/children_hotkeys.py

@@ -671,6 +671,8 @@ async def childkey_take(
         chk_take = await get_childkey_take(
             subtensor=subtensor, netuid=netuid, hotkey=ss58
         )
+        if chk_take is None:
+            chk_take = 0
         chk_take = u16_to_float(chk_take)
         console.print(
             f"Child take for {ss58} is: {chk_take * 100:.2f}% on netuid {netuid}."

bittensor_cli/src/commands/stake/move.py

@@ -17,6 +17,7 @@ from bittensor_cli.src.bittensor.utils import (
     format_error_message,
     group_subnets,
     get_subnet_name,
+    unlock_key,
 )
 
 if TYPE_CHECKING:
@@ -334,8 +335,7 @@ async def stake_move_selection(
 
 
 async def stake_transfer_selection(
-    wallet: Wallet,
-    subtensor: "SubtensorInterface",
+    wallet: Wallet, subtensor: "SubtensorInterface", origin_hotkey: str
 ):
     """Selection interface for transferring stakes."""
     (
@@ -352,7 +352,7 @@ async def stake_transfer_selection(
 
     available_stakes = {}
     for stake in stakes:
-        if stake.stake.tao > 0 and stake.hotkey_ss58 == wallet.hotkey.ss58_address:
+        if stake.stake.tao > 0 and stake.hotkey_ss58 == origin_hotkey:
             available_stakes[stake.netuid] = {
                 "hotkey_ss58": stake.hotkey_ss58,
                 "stake": stake.stake,
@@ -621,10 +621,7 @@ async def move_stake(
             raise typer.Exit()
 
     # Perform moving operation.
-    try:
-        wallet.unlock_coldkey()
-    except KeyFileError:
-        err_console.print("Error decrypting coldkey (possibly incorrect password)")
+    if not unlock_key(wallet).success:
         return False
     with console.status(
         f"\n:satellite: Moving [blue]{amount_to_move_as_balance}[/blue] from [blue]{origin_hotkey}[/blue] on netuid: [blue]{origin_netuid}[/blue] \nto "
@@ -697,6 +694,7 @@ async def transfer_stake(
     wallet: Wallet,
     subtensor: "SubtensorInterface",
     amount: float,
+    origin_hotkey: str,
     origin_netuid: int,
     dest_netuid: int,
     dest_coldkey_ss58: str,
@@ -709,6 +707,7 @@ async def transfer_stake(
         wallet (Wallet): Bittensor wallet object.
         subtensor (SubtensorInterface): Subtensor interface instance.
         amount (float): Amount to transfer.
+        origin_hotkey (str): The hotkey SS58 to transfer the stake from.
         origin_netuid (int): The netuid to transfer stake from.
         dest_netuid (int): The netuid to transfer stake to.
         dest_coldkey_ss58 (str): The destination coldkey to transfer stake to.
@@ -718,8 +717,9 @@ async def transfer_stake(
     Returns:
         bool: True if transfer was successful, False otherwise.
     """
+    origin_hotkey = origin_hotkey or wallet.hotkey.ss58_address
     if interactive_selection:
-        selection = await stake_transfer_selection(wallet, subtensor)
+        selection = await stake_transfer_selection(wallet, subtensor, origin_hotkey)
         origin_netuid = selection["origin_netuid"]
         amount = selection["amount"]
         dest_netuid = selection["destination_netuid"]
@@ -739,16 +739,15 @@ async def transfer_stake(
         return False
 
     # Get current stake balances
-    hotkey_ss58 = wallet.hotkey.ss58_address
     with console.status(f"Retrieving stake data from {subtensor.network}..."):
         current_stake = await subtensor.get_stake(
             coldkey_ss58=wallet.coldkeypub.ss58_address,
-            hotkey_ss58=hotkey_ss58,
+            hotkey_ss58=origin_hotkey,
             netuid=origin_netuid,
         )
         current_dest_stake = await subtensor.get_stake(
             coldkey_ss58=dest_coldkey_ss58,
-            hotkey_ss58=hotkey_ss58,
+            hotkey_ss58=origin_hotkey,
             netuid=dest_netuid,
         )
     amount_to_transfer = Balance.from_tao(amount).set_unit(origin_netuid)
@@ -768,8 +767,8 @@ async def transfer_stake(
             subtensor=subtensor,
             origin_netuid=origin_netuid,
             destination_netuid=dest_netuid,
-            origin_hotkey=hotkey_ss58,
-            destination_hotkey=hotkey_ss58,
+            origin_hotkey=origin_hotkey,
+            destination_hotkey=origin_hotkey,
             amount_to_move=amount_to_transfer,
         )
 
@@ -777,10 +776,7 @@ async def transfer_stake(
             raise typer.Exit()
 
     # Perform transfer operation
-    try:
-        wallet.unlock_coldkey()
-    except KeyFileError:
-        err_console.print("Error decrypting coldkey (possibly incorrect password)")
+    if not unlock_key(wallet).success:
         return False
 
     with console.status("\n:satellite: Transferring stake ..."):
@@ -789,7 +785,7 @@ async def transfer_stake(
             call_function="transfer_stake",
             call_params={
                 "destination_coldkey": dest_coldkey_ss58,
-                "hotkey": hotkey_ss58,
+                "hotkey": origin_hotkey,
                 "origin_netuid": origin_netuid,
                 "destination_netuid": dest_netuid,
                 "alpha_amount": amount_to_transfer.rao,
@@ -820,12 +816,12 @@ async def transfer_stake(
     new_stake, new_dest_stake = await asyncio.gather(
         subtensor.get_stake(
             coldkey_ss58=wallet.coldkeypub.ss58_address,
-            hotkey_ss58=hotkey_ss58,
+            hotkey_ss58=origin_hotkey,
             netuid=origin_netuid,
         ),
         subtensor.get_stake(
             coldkey_ss58=dest_coldkey_ss58,
-            hotkey_ss58=hotkey_ss58,
+            hotkey_ss58=origin_hotkey,
             netuid=dest_netuid,
         ),
     )
@@ -933,10 +929,7 @@ async def swap_stake(
             raise typer.Exit()
 
     # Perform swap operation
-    try:
-        wallet.unlock_coldkey()
-    except KeyFileError:
-        err_console.print("Error decrypting coldkey (possibly incorrect password)")
+    if not unlock_key(wallet).success:
         return False
 
     with console.status(

bittensor_cli/src/commands/stake/remove.py

@@ -21,6 +21,7 @@ from bittensor_cli.src.bittensor.utils import (
     is_valid_ss58_address,
     format_error_message,
     group_subnets,
+    unlock_key,
 )
 
 if TYPE_CHECKING:
@@ -268,10 +269,7 @@ async def unstake(
             raise typer.Exit()
 
     # Execute extrinsics
-    try:
-        wallet.unlock_coldkey()
-    except KeyFileError:
-        err_console.print("Error decrypting coldkey (possibly incorrect password)")
+    if not unlock_key(wallet).success:
         return False
 
     with console.status("\n:satellite: Performing unstaking operations...") as status:
@@ -465,10 +463,7 @@ async def unstake_all(
     ):
         return False
 
-    try:
-        wallet.unlock_coldkey()
-    except KeyFileError:
-        err_console.print("Error decrypting coldkey (possibly incorrect password)")
+    if not unlock_key(wallet).success:
         return False
 
     with console.status("Unstaking all stakes...") as status:

bittensor_cli/src/commands/subnets/subnets.py

@@ -97,8 +97,10 @@ async def register_subnetwork_extrinsic(
     sn_burn_cost = await burn_cost(subtensor)
     if sn_burn_cost > your_balance:
         err_console.print(
-            f"Your balance of: [{COLOR_PALETTE['POOLS']['TAO']}]{your_balance}[{COLOR_PALETTE['POOLS']['TAO']}] is not enough to pay the subnet lock cost of: "
-            f"[{COLOR_PALETTE['POOLS']['TAO']}]{sn_burn_cost}[{COLOR_PALETTE['POOLS']['TAO']}]"
+            f"Your balance of: [{COLOR_PALETTE['POOLS']['TAO']}]{your_balance}[{COLOR_PALETTE['POOLS']['TAO']}]"
+            f" is not enough to burn "
+            f"[{COLOR_PALETTE['POOLS']['TAO']}]{sn_burn_cost}[{COLOR_PALETTE['POOLS']['TAO']}] "
+            f"to register a subnet."
         )
         return False
 
@@ -107,7 +109,7 @@ async def register_subnetwork_extrinsic(
             f"Your balance is: [{COLOR_PALETTE['POOLS']['TAO']}]{your_balance}"
         )
         if not Confirm.ask(
-            f"Do you want to register a subnet for [{COLOR_PALETTE['POOLS']['TAO']}]{sn_burn_cost}?"
+            f"Do you want to burn [{COLOR_PALETTE['POOLS']['TAO']}]{sn_burn_cost} to register a subnet?"
         ):
             return False
 
@@ -145,10 +147,7 @@ async def register_subnetwork_extrinsic(
                 )
                 return False
 
-    try:
-        wallet.unlock_coldkey()
-    except KeyFileError:
-        err_console.print("Error decrypting coldkey (possibly incorrect password)")
+    if not unlock_key(wallet).success:
         return False
 
     with console.status(":satellite: Registering subnet...", spinner="earth"):

bittensor_cli/src/commands/sudo.py

@@ -189,6 +189,11 @@ async def set_hyperparameter_extrinsic(
                 ":cross_mark: [red]Invalid hyperparameter specified.[/red]"
             )
             return False
+    if sudo_:
+        if not Confirm.ask(
+            "This hyperparam is only settable by root sudo users. If you are not, this will fail. Please confirm"
+        ):
+            return False
 
     substrate = subtensor.substrate
     msg_value = value if not arbitrary_extrinsic else call_params