Use unlock_key fn globally

bittensor_cli/src/commands/stake/add.py

@@ -17,6 +17,7 @@
     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_extrinsic(
     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/move.py

@@ -17,6 +17,7 @@
     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 "
@@ -777,10 +775,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 ..."):
@@ -933,10 +928,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 @@
     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

@@ -147,10 +147,7 @@ async def _find_event_attributes_in_extrinsic_receipt(
                 )
                 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"):