Skip to content

Commit d63e404

Browse files
thewhalekingthewhaleking
authored
Merge pull request #386 from opentensor/feat/thewhaleking/color-pallete-refactor
COLOR_PALETTE refactor
2 parents 6abcd27 + 7d3ee18 commit d63e404

File tree

3 files changed

+127
-77
lines changed

3 files changed

+127
-77
lines changed

‎bittensor_cli/cli.py‎

Lines changed: 21 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
WalletOptions as WO,
2626
WalletValidationTypes as WV,
2727
Constants,
28-
COLOR_PALETTE,
28+
COLORS,
2929
HYPERPARAMS,
3030
)
3131
from bittensor_cli.version import __version__, __version_as_int__
@@ -341,8 +341,8 @@ def get_optional_netuid(netuid: Optional[int], all_netuids: bool) -> Optional[in
341341
return None
342342
elif netuid is None and all_netuids is False:
343343
answer = Prompt.ask(
344-
f"Enter the [{COLOR_PALETTE['GENERAL']['SUBHEADING_MAIN']}]netuid"
345-
f"[/{COLOR_PALETTE['GENERAL']['SUBHEADING_MAIN']}] to use. Leave blank for all netuids",
344+
f"Enter the [{COLORS.G.SUBHEAD_MAIN}]netuid"
345+
f"[/{COLORS.G.SUBHEAD_MAIN}] to use. Leave blank for all netuids",
346346
default=None,
347347
show_default=False,
348348
)
@@ -950,7 +950,8 @@ def initialize_chain(
950950
elif self.config["network"]:
951951
self.subtensor = SubtensorInterface(self.config["network"])
952952
console.print(
953-
f"Using the specified network [{COLOR_PALETTE['GENERAL']['LINKS']}]{self.config['network']}[/{COLOR_PALETTE['GENERAL']['LINKS']}] from config"
953+
f"Using the specified network [{COLORS.G.LINKS}]{self.config['network']}"
954+
f"[/{COLORS.G.LINKS}] from config"
954955
)
955956
else:
956957
self.subtensor = SubtensorInterface(defaults.subtensor.network)
@@ -1523,7 +1524,7 @@ def wallet_ask(
15231524
else:
15241525
wallet_name = Prompt.ask(
15251526
"Enter the [blue]wallet name[/blue]"
1526-
+ f" [{COLOR_PALETTE['GENERAL']['HINT']} italic](Hint: You can set this with `btcli config set --wallet-name`)",
1527+
+ f" [{COLORS.G.HINT} italic](Hint: You can set this with `btcli config set --wallet-name`)",
15271528
default=defaults.wallet.name,
15281529
)
15291530

@@ -2049,7 +2050,7 @@ def wallet_regen_coldkey(
20492050

20502051
if not wallet_name:
20512052
wallet_name = Prompt.ask(
2052-
f"Enter the name of the [{COLOR_PALETTE['GENERAL']['COLDKEY']}]new wallet (coldkey)",
2053+
f"Enter the name of the [{COLORS.G.CK}]new wallet (coldkey)",
20532054
default=defaults.wallet.name,
20542055
)
20552056

@@ -2106,7 +2107,7 @@ def wallet_regen_coldkey_pub(
21062107

21072108
if not wallet_name:
21082109
wallet_name = Prompt.ask(
2109-
f"Enter the name of the [{COLOR_PALETTE['GENERAL']['COLDKEY']}]new wallet (coldkey)",
2110+
f"Enter the name of the [{COLORS.G.CK}]new wallet (coldkey)",
21102111
default=defaults.wallet.name,
21112112
)
21122113
wallet = Wallet(wallet_name, wallet_hotkey, wallet_path)
@@ -2223,13 +2224,13 @@ def wallet_new_hotkey(
22232224

22242225
if not wallet_name:
22252226
wallet_name = Prompt.ask(
2226-
f"Enter the [{COLOR_PALETTE['GENERAL']['COLDKEY']}]wallet name",
2227+
f"Enter the [{COLORS.G.CK}]wallet name",
22272228
default=defaults.wallet.name,
22282229
)
22292230

22302231
if not wallet_hotkey:
22312232
wallet_hotkey = Prompt.ask(
2232-
f"Enter the name of the [{COLOR_PALETTE['GENERAL']['HOTKEY']}]new hotkey",
2233+
f"Enter the name of the [{COLORS.G.HK}]new hotkey",
22332234
default=defaults.wallet.hotkey,
22342235
)
22352236

@@ -2285,7 +2286,7 @@ def wallet_new_coldkey(
22852286

22862287
if not wallet_name:
22872288
wallet_name = Prompt.ask(
2288-
f"Enter the name of the [{COLOR_PALETTE['GENERAL']['COLDKEY']}]new wallet (coldkey)",
2289+
f"Enter the name of the [{COLORS.G.CK}]new wallet (coldkey)",
22892290
default=defaults.wallet.name,
22902291
)
22912292

@@ -2359,12 +2360,12 @@ def wallet_create_wallet(
23592360

23602361
if not wallet_name:
23612362
wallet_name = Prompt.ask(
2362-
f"Enter the name of the [{COLOR_PALETTE['GENERAL']['COLDKEY']}]new wallet (coldkey)",
2363+
f"Enter the name of the [{COLORS.G.CK}]new wallet (coldkey)",
23632364
default=defaults.wallet.name,
23642365
)
23652366
if not wallet_hotkey:
23662367
wallet_hotkey = Prompt.ask(
2367-
f"Enter the the name of the [{COLOR_PALETTE['GENERAL']['HOTKEY']}]new hotkey",
2368+
f"Enter the the name of the [{COLORS.G.HK}]new hotkey",
23682369
default=defaults.wallet.hotkey,
23692370
)
23702371

@@ -2742,9 +2743,10 @@ def wallet_sign(
27422743
self.verbosity_handler(quiet, verbose)
27432744
if use_hotkey is None:
27442745
use_hotkey = Confirm.ask(
2745-
f"Would you like to sign the transaction using your [{COLOR_PALETTE['GENERAL']['HOTKEY']}]hotkey[/{COLOR_PALETTE['GENERAL']['HOTKEY']}]?"
2746-
f"\n[Type [{COLOR_PALETTE['GENERAL']['HOTKEY']}]y[/{COLOR_PALETTE['GENERAL']['HOTKEY']}] for [{COLOR_PALETTE['GENERAL']['HOTKEY']}]hotkey[/{COLOR_PALETTE['GENERAL']['HOTKEY']}]"
2747-
f" and [{COLOR_PALETTE['GENERAL']['COLDKEY']}]n[/{COLOR_PALETTE['GENERAL']['COLDKEY']}] for [{COLOR_PALETTE['GENERAL']['COLDKEY']}]coldkey[/{COLOR_PALETTE['GENERAL']['COLDKEY']}]] (default is [{COLOR_PALETTE['GENERAL']['COLDKEY']}]coldkey[/{COLOR_PALETTE['GENERAL']['COLDKEY']}])",
2746+
f"Would you like to sign the transaction using your [{COLORS.G.HK}]hotkey[/{COLORS.G.HK}]?"
2747+
f"\n[Type [{COLORS.G.HK}]y[/{COLORS.G.HK}] for [{COLORS.G.HK}]hotkey[/{COLORS.G.HK}]"
2748+
f" and [{COLORS.G.CK}]n[/{COLORS.G.CK}] for [{COLORS.G.CK}]coldkey[/{COLORS.G.CK}]] "
2749+
f"(default is [{COLORS.G.CK}]coldkey[/{COLORS.G.CK}])",
27482750
default=False,
27492751
)
27502752

@@ -3035,11 +3037,11 @@ def stake_add(
30353037
raise typer.Exit()
30363038
if netuid is not None:
30373039
amount = FloatPrompt.ask(
3038-
f"Amount to [{COLOR_PALETTE['GENERAL']['SUBHEADING_MAIN']}]stake (TAO τ)"
3040+
f"Amount to [{COLORS.G.SUBHEAD_MAIN}]stake (TAO τ)"
30393041
)
30403042
else:
30413043
amount = FloatPrompt.ask(
3042-
f"Amount to [{COLOR_PALETTE['GENERAL']['SUBHEADING_MAIN']}]stake to each netuid (TAO τ)"
3044+
f"Amount to [{COLORS.G.SUBHEAD_MAIN}]stake to each netuid (TAO τ)"
30433045
)
30443046

30453047
if amount <= 0:
@@ -4112,7 +4114,8 @@ def sudo_set(
41124114
if not param_value:
41134115
if HYPERPARAMS.get(param_name):
41144116
param_value = Prompt.ask(
4115-
f"Enter the new value for [{COLOR_PALETTE['GENERAL']['SUBHEADING']}]{param_name}[/{COLOR_PALETTE['GENERAL']['SUBHEADING']}] in the VALUE column format"
4117+
f"Enter the new value for [{COLORS.G.SUBHEAD}]{param_name}[/{COLORS.G.SUBHEAD}] "
4118+
f"in the VALUE column format"
41164119
)
41174120
else:
41184121
param_value = None

‎bittensor_cli/src/__init__.py‎

Lines changed: 105 additions & 58 deletions
Original file line numberDiff line numberDiff line change
@@ -721,64 +721,111 @@ class WalletValidationTypes(Enum):
721721
},
722722
}
723723

724-
COLOR_PALETTE = {
725-
"GENERAL": {
726-
"HEADER": "#4196D6", # Light Blue
727-
"LINKS": "#8CB9E9", # Sky Blue
728-
"HINT": "#A2E5B8", # Mint Green
729-
"COLDKEY": "#9EF5E4", # Aqua
730-
"HOTKEY": "#ECC39D", # Light Orange/Peach
731-
"SUBHEADING_MAIN": "#7ECFEC", # Light Cyan
732-
"SUBHEADING": "#AFEFFF", # Pale Blue
733-
"SUBHEADING_EXTRA_1": "#96A3C5", # Grayish Blue
734-
"SUBHEADING_EXTRA_2": "#6D7BAF", # Slate Blue
735-
"CONFIRMATION_Y_N_Q": "#EE8DF8", # Light Purple/Pink
736-
"SYMBOL": "#E7CC51", # Gold
737-
"BALANCE": "#4F91C6", # Medium Blue
738-
"COST": "#53B5A0", # Teal
739-
"SUCCESS": "#53B5A0", # Teal
740-
"NETUID": "#CBA880", # Tan
741-
"NETUID_EXTRA": "#DDD5A9", # Light Khaki
742-
"TEMPO": "#67A3A5", # Grayish Teal
743-
},
744-
"STAKE": {
745-
"STAKE_AMOUNT": "#53B5A0", # Teal
746-
"STAKE_ALPHA": "#53B5A0", # Teal
747-
"STAKE_SWAP": "#67A3A5", # Grayish Teal
748-
"TAO": "#4F91C6", # Medium Blue
749-
"SLIPPAGE_TEXT": "#C25E7C", # Rose
750-
"SLIPPAGE_PERCENT": "#E7B195", # Light Coral
751-
"NOT_REGISTERED": "#EB6A6C", # Salmon Red
752-
"EXTRA_1": "#D781BB", # Pink
753-
},
754-
"POOLS": {
755-
"TAO": "#4F91C6", # Medium Blue
756-
"ALPHA_IN": "#D09FE9", # Light Purple
757-
"ALPHA_OUT": "#AB7CC8", # Medium Purple
758-
"RATE": "#F8D384", # Light Orange
759-
"TAO_EQUIV": "#8CB9E9", # Sky Blue
760-
"EMISSION": "#F8D384", # Light Orange
761-
"EXTRA_1": "#CAA8FB", # Lavender
762-
"EXTRA_2": "#806DAF", # Dark Purple
763-
},
764-
"GREY": {
765-
"GREY_100": "#F8F9FA", # Almost White
766-
"GREY_200": "#F1F3F4", # Very Light Grey
767-
"GREY_300": "#DBDDE1", # Light Grey
768-
"GREY_400": "#BDC1C6", # Medium Light Grey
769-
"GREY_500": "#5F6368", # Medium Grey
770-
"GREY_600": "#2E3134", # Medium Dark Grey
771-
"GREY_700": "#282A2D", # Dark Grey
772-
"GREY_800": "#17181B", # Very Dark Grey
773-
"GREY_900": "#0E1013", # Almost Black
774-
"BLACK": "#000000", # Pure Black
775-
},
776-
"SUDO": {
777-
"HYPERPARAMETER": "#4F91C6", # Medium Blue
778-
"VALUE": "#D09FE9", # Light Purple
779-
"NORMALIZED": "#AB7CC8", # Medium Purple
780-
},
781-
}
724+
725+
class Gettable:
726+
def __getitem__(self, item):
727+
return getattr(self, item)
728+
729+
730+
class ColorPalette(Gettable):
731+
def __init__(self):
732+
self.GENERAL = self.General()
733+
self.STAKE = self.Stake()
734+
self.POOLS = self.Pools()
735+
self.GREY = self.Grey()
736+
self.SUDO = self.Sudo()
737+
# aliases
738+
self.G = self.GENERAL
739+
self.S = self.STAKE
740+
self.P = self.POOLS
741+
self.GR = self.GREY
742+
self.SU = self.SUDO
743+
744+
class General(Gettable):
745+
HEADER = "#4196D6" # Light Blue
746+
LINKS = "#8CB9E9" # Sky Blue
747+
HINT = "#A2E5B8" # Mint Green
748+
COLDKEY = "#9EF5E4" # Aqua
749+
HOTKEY = "#ECC39D" # Light Orange/Peach
750+
SUBHEADING_MAIN = "#7ECFEC" # Light Cyan
751+
SUBHEADING = "#AFEFFF" # Pale Blue
752+
SUBHEADING_EXTRA_1 = "#96A3C5" # Grayish Blue
753+
SUBHEADING_EXTRA_2 = "#6D7BAF" # Slate Blue
754+
CONFIRMATION_Y_N_Q = "#EE8DF8" # Light Purple/Pink
755+
SYMBOL = "#E7CC51" # Gold
756+
BALANCE = "#4F91C6" # Medium Blue
757+
COST = "#53B5A0" # Teal
758+
SUCCESS = "#53B5A0" # Teal
759+
NETUID = "#CBA880" # Tan
760+
NETUID_EXTRA = "#DDD5A9" # Light Khaki
761+
TEMPO = "#67A3A5" # Grayish Teal
762+
# aliases
763+
CK = COLDKEY
764+
HK = HOTKEY
765+
SUBHEAD_MAIN = SUBHEADING_MAIN
766+
SUBHEAD = SUBHEADING
767+
SUBHEAD_EX_1 = SUBHEADING_EXTRA_1
768+
SUBHEAD_EX_2 = SUBHEADING_EXTRA_2
769+
SYM = SYMBOL
770+
BAL = BALANCE
771+
772+
class Stake(Gettable):
773+
STAKE_AMOUNT = "#53B5A0" # Teal
774+
STAKE_ALPHA = "#53B5A0" # Teal
775+
STAKE_SWAP = "#67A3A5" # Grayish Teal
776+
TAO = "#4F91C6" # Medium Blue
777+
SLIPPAGE_TEXT = "#C25E7C" # Rose
778+
SLIPPAGE_PERCENT = "#E7B195" # Light Coral
779+
NOT_REGISTERED = "#EB6A6C" # Salmon Red
780+
EXTRA_1 = "#D781BB" # Pink
781+
# aliases
782+
AMOUNT = STAKE_AMOUNT
783+
ALPHA = STAKE_ALPHA
784+
SWAP = STAKE_SWAP
785+
786+
class Pools(Gettable):
787+
TAO = "#4F91C6" # Medium Blue
788+
ALPHA_IN = "#D09FE9" # Light Purple
789+
ALPHA_OUT = "#AB7CC8" # Medium Purple
790+
RATE = "#F8D384" # Light Orange
791+
TAO_EQUIV = "#8CB9E9" # Sky Blue
792+
EMISSION = "#F8D384" # Light Orange
793+
EXTRA_1 = "#CAA8FB" # Lavender
794+
EXTRA_2 = "#806DAF" # Dark Purple
795+
796+
class Grey(Gettable):
797+
GREY_100 = "#F8F9FA" # Almost White
798+
GREY_200 = "#F1F3F4" # Very Light Grey
799+
GREY_300 = "#DBDDE1" # Light Grey
800+
GREY_400 = "#BDC1C6" # Medium Light Grey
801+
GREY_500 = "#5F6368" # Medium Grey
802+
GREY_600 = "#2E3134" # Medium Dark Grey
803+
GREY_700 = "#282A2D" # Dark Grey
804+
GREY_800 = "#17181B" # Very Dark Grey
805+
GREY_900 = "#0E1013" # Almost Black
806+
BLACK = "#000000" # Pure Black
807+
# aliases
808+
G_100 = GREY_100
809+
G_200 = GREY_200
810+
G_300 = GREY_300
811+
G_400 = GREY_400
812+
G_500 = GREY_500
813+
G_600 = GREY_600
814+
G_700 = GREY_700
815+
G_800 = GREY_800
816+
G_900 = GREY_900
817+
818+
class Sudo(Gettable):
819+
HYPERPARAMETER = "#4F91C6" # Medium Blue
820+
VALUE = "#D09FE9" # Light Purple
821+
NORMALIZED = "#AB7CC8" # Medium Purple
822+
# aliases
823+
HYPERPARAM = HYPERPARAMETER
824+
NORMAL = NORMALIZED
825+
826+
827+
COLOR_PALETTE = ColorPalette()
828+
COLORS = COLOR_PALETTE
782829

783830

784831
SUBNETS = {

‎bittensor_cli/src/commands/subnets/subnets.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1417,7 +1417,7 @@ async def pow_register(
14171417
use_cuda,
14181418
dev_id,
14191419
threads_per_block,
1420-
prompt: bool
1420+
prompt: bool,
14211421
):
14221422
"""Register neuron."""
14231423

0 commit comments

Comments
 (0)