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

bittensor_cli/cli.py

@@ -74,8 +74,6 @@ except ImportError:
         pass
 
 
-
-
 _epilog = "Made with [bold red]:heart:[/bold red] by The Openτensor Foundaτion"
 
 np.set_printoptions(precision=8, suppress=True, floatmode="fixed")
@@ -481,7 +479,7 @@ def version_callback(value: bool):
                 f"{repo.active_branch.name}/"
                 f"{repo.commit()}"
             )
-        except (NameError, GitError):
+        except (TypeError, GitError):
             version = f"BTCLI version: {__version__}"
         typer.echo(version)
         raise typer.Exit()
@@ -3323,9 +3321,9 @@ class CLIManager:
     def stake_move(
         self,
         network: Optional[list[str]] = Options.network,
-        wallet_name=Options.wallet_name,
-        wallet_path=Options.wallet_path,
-        wallet_hotkey=Options.wallet_hotkey,
+        wallet_name: Optional[str] = Options.wallet_name,
+        wallet_path: Optional[str] = Options.wallet_path,
+        wallet_hotkey: Optional[str] = Options.wallet_hotkey,
         origin_netuid: Optional[int] = typer.Option(
             None, "--origin-netuid", help="Origin netuid"
         ),
@@ -3535,14 +3533,45 @@ class CLIManager:
         )
         self.verbosity_handler(quiet, verbose)
 
+        if not wallet_name:
+            wallet_name = Prompt.ask(
+                "Enter the [blue]origin wallet name[/blue]",
+                default=self.config.get("wallet_name") or defaults.wallet.name,
+            )
         wallet = self.wallet_ask(
-            wallet_name,
-            wallet_path,
-            wallet_hotkey,
-            ask_for=[WO.NAME, WO.PATH, WO.HOTKEY],
-            validate=WV.WALLET_AND_HOTKEY,
+            wallet_name, wallet_path, wallet_hotkey, ask_for=[WO.NAME]
         )
 
+        if not wallet_hotkey:
+            origin_hotkey = Prompt.ask(
+                "Enter the [blue]origin hotkey[/blue] name or "
+                "[blue]ss58 address[/blue] where the stake will be moved from",
+                default=self.config.get("wallet_hotkey") or defaults.wallet.hotkey,
+            )
+            if is_valid_ss58_address(origin_hotkey):
+                origin_hotkey = origin_hotkey
+            else:
+                wallet = self.wallet_ask(
+                    wallet_name,
+                    wallet_path,
+                    origin_hotkey,
+                    ask_for=[WO.NAME, WO.PATH, WO.HOTKEY],
+                    validate=WV.WALLET_AND_HOTKEY,
+                )
+                origin_hotkey = wallet.hotkey.ss58_address
+        else:
+            if is_valid_ss58_address(wallet_hotkey):
+                origin_hotkey = wallet_hotkey
+            else:
+                wallet = self.wallet_ask(
+                    wallet_name,
+                    wallet_path,
+                    wallet_hotkey,
+                    ask_for=[],
+                    validate=WV.WALLET_AND_HOTKEY,
+                )
+                origin_hotkey = wallet.hotkey.ss58_address
+
         if not dest_ss58:
             dest_ss58 = Prompt.ask(
                 "Enter the [blue]destination wallet name[/blue] or [blue]coldkey SS58 address[/blue]"
@@ -3580,6 +3609,7 @@ class CLIManager:
             move_stake.transfer_stake(
                 wallet=wallet,
                 subtensor=self.initialize_chain(network),
+                origin_hotkey=origin_hotkey,
                 origin_netuid=origin_netuid,
                 dest_netuid=dest_netuid,
                 dest_coldkey_ss58=dest_ss58,

bittensor_cli/src/bittensor/subtensor_interface.py

@@ -1327,20 +1327,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/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:
@@ -621,10 +622,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 +695,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 +708,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.
@@ -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

bittensor_cli/version.py

@@ -1,5 +1,6 @@
 import re
 
+
 def version_as_int(version):
     _core_version = re.match(r"^\d+\.\d+\.\d+", version).group(0)
     _version_split = _core_version.split(".")
@@ -14,5 +15,5 @@ def version_as_int(version):
     __new_signature_version__ = 360
     return __version_as_int__
 
-__version__ = "9.0.2"
+__version__ = "9.0.3"
 __version_as_int__ = version_as_int(__version__)

{bittensor_cli-9.0.2.dist-info → bittensor_cli-9.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: bittensor-cli
-Version: 9.0.2
+Version: 9.0.3
 Summary: Bittensor CLI
 Home-page: https://github.com/opentensor/btcli
 Author: bittensor.com

{bittensor_cli-9.0.2.dist-info → bittensor_cli-9.0.3.dist-info}/RECORD

@@ -1,14 +1,14 @@
 bittensor_cli/__init__.py,sha256=Lpv4NkbAQgwrfqFOnTMuR_S-fqGdaWCSLhxnFnGTHM0,1232
-bittensor_cli/cli.py,sha256=hiCu9JQFnu22Q1WjpApkhzNvP9EuR3grNYXWcvGoufc,192627
+bittensor_cli/cli.py,sha256=i9KKs432ZN9mb7JDIsFrKD6qS_jfQQaBUn6haTwD-F4,194036
 bittensor_cli/doc_generation_helper.py,sha256=GexqjEIKulWg84hpNBEchJ840oOgOi7DWpt447nsdNI,91
-bittensor_cli/version.py,sha256=W091sz9aNjfMKBgfd7SGgZHim3RwQSv-M8lo7mgPIEY,621
+bittensor_cli/version.py,sha256=34wkPxh1A8pTDhsf7Fo0mtk1I7GfmHzJTaampx1V_nQ,622
 bittensor_cli/src/__init__.py,sha256=rTtP85aCliJTLdWv2DYFNgw5Fi0KB2--80Xta5_WsXE,26427
 bittensor_cli/src/bittensor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bittensor_cli/src/bittensor/balances.py,sha256=EZI8hX-eCkTfuTZq5GMvRYU6DCdl8V_zyZIQWi05K5g,11084
 bittensor_cli/src/bittensor/chain_data.py,sha256=WV_zX_05Zam1yp_oMX9VIYxu99cGPUGEnhCyH67vNtE,30856
 bittensor_cli/src/bittensor/minigraph.py,sha256=BIzmSVLfBYiRAeGD_i1LAC8Cw7zxp38a91SIFEPMqYc,10479
 bittensor_cli/src/bittensor/networking.py,sha256=pZLMs8YXpZzDMLXWMBb_Bj6TVkm_q9khyY-lnbwVMuE,462
-bittensor_cli/src/bittensor/subtensor_interface.py,sha256=mprNkG462WW57eRUhMmDE4gHGuMoeIvBeQLLSxIppLI,53098
+bittensor_cli/src/bittensor/subtensor_interface.py,sha256=T9zBmWTR2fhotCrLQ1X0ihdMiy4vymHSkLBRbYy8Fqs,53457
 bittensor_cli/src/bittensor/utils.py,sha256=E57t4SiASuXCjiXLLU63vK6bK6aKqj9JtIIi9lDOi-k,46238
 bittensor_cli/src/bittensor/extrinsics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bittensor_cli/src/bittensor/extrinsics/registration.py,sha256=3mJZ3hw_wZEa-8I0R8WVuKjMQi4Y9EV5FjTCvbY37Iw,63780
@@ -16,20 +16,20 @@ bittensor_cli/src/bittensor/extrinsics/root.py,sha256=N9Fg4VaveRRP1ZN4EZjIWCe04F
 bittensor_cli/src/bittensor/extrinsics/transfer.py,sha256=FyrRo3yk-065toN4f-1Xes23CE5tqP5KU0dBJkKofUc,8476
 bittensor_cli/src/bittensor/templates/table.j2,sha256=P2EFiksnO1cQsB8zjK6hVJwUryHTsLslzRE0YtobAV8,10601
 bittensor_cli/src/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bittensor_cli/src/commands/sudo.py,sha256=yjmUuOrKkz0Ft2TqHWl4Lw96aYRCEk9kdvtyXCTOASE,31295
+bittensor_cli/src/commands/sudo.py,sha256=erQVUDj101iJcmDL1Ej7tGQHNfWRP5BHpx0dXLHQYbE,31487
 bittensor_cli/src/commands/wallets.py,sha256=hv9gzh-zPS8gXzgJIfENQVR6BFYxh6abjxmHUjqMhsY,51086
 bittensor_cli/src/commands/weights.py,sha256=uI7aACKD90JOtYt61VdKL76z7Fe_wh4WtdwMXL6ydD4,16269
 bittensor_cli/src/commands/stake/__init__.py,sha256=uxomMv_QrYt5Qn3_X5UWFFh45ISjB0JmDmCFxVyX8nQ,6495
-bittensor_cli/src/commands/stake/add.py,sha256=ZPFuvlTUT2z23urITj_RLUBe51jqyfZ8l22-zdd2Nzg,25074
-bittensor_cli/src/commands/stake/children_hotkeys.py,sha256=9CMc56eg_6-tpRz9x5ml4Brc5SSbhC7KDlJqYgQ4XiU,29576
+bittensor_cli/src/commands/stake/add.py,sha256=F93eM-RBjrjeoSQAq0iBj6LJwBF9BQJVDd4f9bub1f8,24979
+bittensor_cli/src/commands/stake/children_hotkeys.py,sha256=STUeMc9PdzJPx_nfFzPLBUkSc4Wf9b40PJZZZEAKHfI,29630
 bittensor_cli/src/commands/stake/list.py,sha256=AnzBhUjuwckFAXYwgv9Yk1JHUKu2w1h8_m8-hf77g80,28625
-bittensor_cli/src/commands/stake/move.py,sha256=rFCHpkdyUc6tSp1YiJ3fOVG_s6La7YVwQPhXDIBD_FE,37406
-bittensor_cli/src/commands/stake/remove.py,sha256=6RnxandM05xAG1QJ_vRtWRI0xlzfZC_4kEJgOPKua0M,47069
+bittensor_cli/src/commands/stake/move.py,sha256=HiQtuighlAvZr6uhtGikn5L5ZXgEjSHS5DbKhIDuTbU,37155
+bittensor_cli/src/commands/stake/remove.py,sha256=gaQN7Gck0l0GCmtiYhZMWlD6OudcNy6cEIEkE8OzQRo,46863
 bittensor_cli/src/commands/subnets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bittensor_cli/src/commands/subnets/price.py,sha256=TWcRXUFeS_Q-pfyv0YIluAL8SE7d2gzTODK-9M2J5pw,29878
-bittensor_cli/src/commands/subnets/subnets.py,sha256=ObB2g-iTCPkmrHomR0RHdvqX8FSHXaywSIoIFXko1h4,84118
-bittensor_cli-9.0.2.dist-info/METADATA,sha256=a-FAYGb9717y58lnFPAtwbKdVBCDwC4edgLf7sQd-GI,6855
-bittensor_cli-9.0.2.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-bittensor_cli-9.0.2.dist-info/entry_points.txt,sha256=hBTLGLbVxmAKy69XSKaUZvjTCmyEzDGZKq4S8UOto8I,49
-bittensor_cli-9.0.2.dist-info/top_level.txt,sha256=DvgvXpmTtI_Q1BbDZMlK90LFcGFCreN1daViEPV2iFw,14
-bittensor_cli-9.0.2.dist-info/RECORD,,
+bittensor_cli/src/commands/subnets/subnets.py,sha256=YuJ0pmd63O2dg9j3WmTwAeZQ7CGweqw6-ORF6ZaaNh0,84041
+bittensor_cli-9.0.3.dist-info/METADATA,sha256=4_hNxQ1PdM4VNJes38PSeRMZFitO59g0wLPko8xPKpY,6855
+bittensor_cli-9.0.3.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+bittensor_cli-9.0.3.dist-info/entry_points.txt,sha256=hBTLGLbVxmAKy69XSKaUZvjTCmyEzDGZKq4S8UOto8I,49
+bittensor_cli-9.0.3.dist-info/top_level.txt,sha256=DvgvXpmTtI_Q1BbDZMlK90LFcGFCreN1daViEPV2iFw,14
+bittensor_cli-9.0.3.dist-info/RECORD,,