Last commands finished.

Author: thewhaleking

bittensor_cli/cli.py

@@ -5341,6 +5341,7 @@ def weights_reveal(
             "-s",
             help="Corresponding salt for the hash function, e.g. -s 163,241,217 ...",
         ),
+        json_output: bool = Options.json_output,
         quiet: bool = Options.quiet,
         verbose: bool = Options.verbose,
         prompt: bool = Options.prompt,
@@ -5354,7 +5355,7 @@ def weights_reveal(
 
         [green]$[/green] btcli wt reveal --netuid 1 --uids 1,2,3,4 --weights 0.1,0.2,0.3,0.4 --salt 163,241,217,11,161,142,147,189
         """
-        self.verbosity_handler(quiet, verbose)
+        self.verbosity_handler(quiet, verbose, json_output)
         uids = list_prompt(uids, int, "UIDs of interest for the specified netuid")
         weights = list_prompt(
             weights, float, "Corresponding weights for the specified UIDs"
@@ -5387,7 +5388,7 @@ def weights_reveal(
             err_console.print(
                 "The number of UIDs you specify must match up with the specified number of weights"
             )
-            raise typer.Exit()
+            return
 
         if salt:
             salt = parse_to_list(
@@ -5416,6 +5417,7 @@ def weights_reveal(
                 salt,
                 __version_as_int__,
                 prompt=prompt,
+                json_output=json_output,
             )
         )
 
@@ -5439,6 +5441,7 @@ def weights_commit(
             "-s",
             help="Corresponding salt for the hash function, e.g. -s 163 -s 241 -s 217 ...",
         ),
+        json_output: bool = Options.json_output,
         quiet: bool = Options.quiet,
         verbose: bool = Options.verbose,
         prompt: bool = Options.prompt,
@@ -5456,7 +5459,7 @@ def weights_commit(
         [italic]Note[/italic]: This command is used to commit weights for a specific subnet and requires the user to have the necessary
         permissions.
         """
-        self.verbosity_handler(quiet, verbose)
+        self.verbosity_handler(quiet, verbose, json_output)
 
         if uids:
             uids = parse_to_list(
@@ -5485,7 +5488,7 @@ def weights_commit(
             err_console.print(
                 "The number of UIDs you specify must match up with the specified number of weights"
             )
-            raise typer.Exit()
+            return
 
         if salt:
             salt = parse_to_list(
@@ -5512,6 +5515,7 @@ def weights_commit(
                 weights,
                 salt,
                 __version_as_int__,
+                json_output=json_output,
                 prompt=prompt,
             )
         )

bittensor_cli/src/commands/weights.py

@@ -1,4 +1,5 @@
 import asyncio
+import json
 import os
 from datetime import datetime, timedelta
 from typing import TYPE_CHECKING
@@ -9,7 +10,12 @@
 from rich.prompt import Confirm
 from async_substrate_interface.errors import SubstrateRequestException
 
-from bittensor_cli.src.bittensor.utils import err_console, console, format_error_message
+from bittensor_cli.src.bittensor.utils import (
+    err_console,
+    console,
+    format_error_message,
+    json_console,
+)
 from bittensor_cli.src.bittensor.extrinsics.root import (
     convert_weights_and_uids_for_emit,
     generate_weight_hash,
@@ -372,6 +378,7 @@ async def reveal_weights(
     weights: list[float],
     salt: list[int],
     version: int,
+    json_output: bool = False,
     prompt: bool = True,
 ) -> None:
     """Reveal weights for a specific subnet."""
@@ -395,11 +402,13 @@ async def reveal_weights(
         subtensor, wallet, netuid, uids_, weights_, list(salt_), version, prompt=prompt
     )
     success, message = await extrinsic.reveal(weight_uids, weight_vals)
-
-    if success:
-        console.print("Weights revealed successfully")
+    if json_output:
+        json_console.print(json.dumps({"success": success, "message": message}))
     else:
-        err_console.print(f"Failed to reveal weights: {message}")
+        if success:
+            console.print("Weights revealed successfully")
+        else:
+            err_console.print(f"Failed to reveal weights: {message}")
 
 
 async def commit_weights(
@@ -410,6 +419,7 @@ async def commit_weights(
     weights: list[float],
     salt: list[int],
     version: int,
+    json_output: bool = False,
     prompt: bool = True,
 ):
     """Commits weights and then reveals them for a specific subnet"""
@@ -429,7 +439,10 @@ async def commit_weights(
         subtensor, wallet, netuid, uids_, weights_, list(salt_), version, prompt=prompt
     )
     success, message = await extrinsic.set_weights_extrinsic()
-    if success:
-        console.print("Weights set successfully")
+    if json_output:
+        json_console.print(json.dumps({"success": success, "message": message}))
     else:
-        err_console.print(f"Failed to commit weights: {message}")
+        if success:
+            console.print("Weights set successfully")
+        else:
+            err_console.print(f"Failed to commit weights: {message}")