Updates

Author: ibraheem-abe

bittensor_cli/cli.py

@@ -2274,6 +2274,7 @@ def wallet_associate_hotkey(
         wallet_path: Optional[str] = Options.wallet_path,
         wallet_hotkey: Optional[str] = Options.wallet_hotkey_ss58,
         network: Optional[list[str]] = Options.network,
+        prompt: bool = Options.prompt,
         quiet: bool = Options.quiet,
         verbose: bool = Options.verbose,
     ):
@@ -2296,33 +2297,33 @@ def wallet_associate_hotkey(
                 default=self.config.get("wallet_name") or defaults.wallet.name,
             )
         if not wallet_hotkey:
-            hotkey_or_wallet = Prompt.ask(
+            wallet_hotkey = Prompt.ask(
                 "Enter the [blue]hotkey[/blue] name or "
                 "[blue]hotkey ss58 address[/blue] [dim](to associate with your coldkey)[/dim]"
             )
-            if is_valid_ss58_address(hotkey_or_wallet):
-                hotkey_ss58 = hotkey_or_wallet
-                wallet = self.wallet_ask(
-                    wallet_name,
-                    wallet_path,
-                    wallet_hotkey,
-                    ask_for=[WO.NAME, WO.PATH],
-                    validate=WV.WALLET,
-                )
-            else:
-                wallet = self.wallet_ask(
-                    wallet_name,
-                    wallet_path,
-                    hotkey_or_wallet,
-                    ask_for=[WO.NAME, WO.PATH, WO.HOTKEY],
-                    validate=WV.WALLET_AND_HOTKEY,
-                )
-                hotkey_ss58 = wallet.hotkey.ss58_address
+
+        if is_valid_ss58_address(wallet_hotkey):
+            hotkey_ss58 = wallet_hotkey
+            wallet = self.wallet_ask(
+                wallet_name,
+                wallet_path,
+                None,
+                ask_for=[WO.NAME, WO.PATH],
+                validate=WV.WALLET,
+            )
+        else:
+            wallet = self.wallet_ask(
+                wallet_name,
+                wallet_path,
+                wallet_hotkey,
+                ask_for=[WO.NAME, WO.PATH, WO.HOTKEY],
+                validate=WV.WALLET_AND_HOTKEY,
+            )
+            hotkey_ss58 = wallet.hotkey.ss58_address
+
         return self._run_command(
             wallets.associate_hotkey(
-                wallet,
-                self.initialize_chain(network),
-                hotkey_ss58,
+                wallet, self.initialize_chain(network), hotkey_ss58, prompt
             )
         )
 

bittensor_cli/src/commands/wallets.py

@@ -54,6 +54,7 @@ async def associate_hotkey(
     wallet: Wallet,
     subtensor: SubtensorInterface,
     hotkey_ss58: str,
+    prompt: bool = False,
 ):
     """Associates a hotkey with a wallet"""
     owner_ss58 = await subtensor.get_hotkey_owner(hotkey_ss58)
@@ -78,7 +79,9 @@ async def associate_hotkey(
             f"Hotkey [{COLORS.GENERAL.HK}]{hotkey_ss58}[/{COLORS.GENERAL.HK}] is not associated with any wallet"
         )
 
-    if not Confirm.ask("Do you want to continue with the association?"):
+    if prompt and not Confirm.ask(
+        "Do you want to continue with the association?"
+    ):
         return False
 
     if not unlock_key(wallet).success: