Skip to content

Utils App with convert command #174

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
ibraheem-abe merged 2 commits into from
Oct 10, 2024
Merged

Utils App with convert command #174

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
5c89e8c
Adds a utils app with a single subcommand (convert) to easily convert…
thewhaleking Oct 10, 2024
3e16772
Better UX
thewhaleking Oct 10, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 36 additions & 0 deletions bittensor_cli/cli.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@
Constants,
)
from bittensor_cli.src.bittensor import utils
from bittensor_cli.src.bittensor.balances import Balance
from bittensor_cli.src.bittensor.async_substrate_interface import (
SubstrateRequestException,
)
Expand Down Expand Up @@ -400,6 +401,7 @@ class CLIManager:
root_app: typer.Typer
subnets_app: typer.Typer
weights_app: typer.Typer
utils_app = typer.Typer(epilog=_epilog)

def __init__(self):
self.config = {
Expand Down Expand Up @@ -524,6 +526,9 @@ def __init__(self):
self.weights_app, name="weight", hidden=True, no_args_is_help=True
)

# utils app
self.app.add_typer(self.utils_app, name="utils", no_args_is_help=True)

# config commands
self.config_app.command("set")(self.set_config)
self.config_app.command("get")(self.get_config)
Expand Down Expand Up @@ -4333,6 +4338,37 @@ def weights_commit(
)
)

@staticmethod
@utils_app.command("convert")
def convert(
from_rao: Optional[str] = typer.Option(
None, "--rao", help="Convert amount from Rao"
),
from_tao: Optional[float] = typer.Option(
None, "--tao", help="Convert amount from Tao"
),
):
"""
Allows for converting between tao and rao using the specified flags
"""
if from_tao is None and from_rao is None:
err_console.print("Specify `--rao` and/or `--tao`.")
raise typer.Exit()
if from_rao is not None:
rao = int(float(from_rao))
console.print(
f"{rao}{Balance.rao_unit}",
"=",
Balance.from_rao(rao),
)
if from_tao is not None:
tao = float(from_tao)
console.print(
f"{Balance.unit}{tao}",
"=",
f"{Balance.from_tao(tao).rao}{Balance.rao_unit}",
)

def run(self):
self.app()

Expand Down