From cd976c9f49d03f8b9366981882c0e396d8f74347 Mon Sep 17 00:00:00 2001 From: Cameron Fairchild Date: Tue, 15 Nov 2022 15:56:12 -0500 Subject: [PATCH] mention balance if not no prompt --- bittensor/_cli/__init__.py | 24 ++++++++++++++++++------ 1 file changed, 18 insertions(+), 6 deletions(-) diff --git a/bittensor/_cli/__init__.py b/bittensor/_cli/__init__.py index d4549d05ae..cc14d09641 100644 --- a/bittensor/_cli/__init__.py +++ b/bittensor/_cli/__init__.py @@ -742,15 +742,27 @@ def check_transfer_config( config: 'bittensor.Config'): sys.exit() else: config.dest = str(dest) + + # Get current balance and print to user. + if not config.no_prompt: + wallet = bittensor.wallet( config ) + subtensor = bittensor.subtensor( config ) + with bittensor.__console__.status(":satellite: Checking Balance..."): + account_balance = subtensor.get_balance( wallet.coldkeypub.ss58_address ) + bittensor.__console__.print("Balance: [green]{}[/green]".format(account_balance)) # Get amount. if not config.get('amount'): - amount = Prompt.ask("Enter Tao amount to transfer") - try: - config.amount = float(amount) - except ValueError: - console.print(":cross_mark:[red] Invalid Tao amount[/red] [bold white]{}[/bold white]".format(amount)) - sys.exit() + if not config.no_prompt: + amount = Prompt.ask("Enter TAO amount to transfer") + try: + config.amount = float(amount) + except ValueError: + console.print(":cross_mark:[red] Invalid TAO amount[/red] [bold white]{}[/bold white]".format(amount)) + sys.exit() + else: + console.print(":cross_mark:[red] Invalid TAO amount[/red] [bold white]{}[/bold white]".format(amount)) + sys.exit(1) def check_unstake_config( config: 'bittensor.Config' ): if config.subtensor.get('network') == bittensor.defaults.subtensor.network and not config.no_prompt: