feat: refactor move.py and remove.py with create_table

Author: leonace924

bittensor_cli/src/commands/stake/move.py

@@ -16,6 +16,7 @@
 from bittensor_cli.src.bittensor.utils import (
     confirm_action,
     console,
+    create_table,
     print_error,
     group_subnets,
     get_subnet_name,
@@ -166,7 +167,7 @@ async def display_stake_movement_cross_subnets(
         )
 
     # Create and display table
-    table = Table(
+    table = create_table(
         title=(
             f"\n[{COLOR_PALETTE.G.HEADER}]"
             f"Moving stake from: "
@@ -177,14 +178,6 @@ async def display_stake_movement_cross_subnets(
             f"[/{COLOR_PALETTE.G.SUBHEAD}]\nNetwork: {subtensor.network}\n"
             f"[/{COLOR_PALETTE.G.HEADER}]"
         ),
-        show_footer=True,
-        show_edge=False,
-        header_style="bold white",
-        border_style="bright_black",
-        style="bold",
-        title_justify="center",
-        show_lines=False,
-        pad_edge=True,
     )
 
     table.add_column(
@@ -351,16 +344,8 @@ async def stake_move_transfer_selection(
         raise ValueError
 
     # Display hotkeys with stakes
-    table = Table(
+    table = create_table(
         title=f"\n[{COLOR_PALETTE['GENERAL']['HEADER']}]Hotkeys with Stakes\n",
-        show_footer=True,
-        show_edge=False,
-        header_style="bold white",
-        border_style="bright_black",
-        style="bold",
-        title_justify="center",
-        show_lines=False,
-        pad_edge=True,
     )
     table.add_column("Index", justify="right")
     table.add_column("Identity", style=COLOR_PALETTE["GENERAL"]["SUBHEADING"])

bittensor_cli/src/commands/stake/remove.py

@@ -18,6 +18,7 @@
 from bittensor_cli.src.bittensor.utils import (
     confirm_action,
     console,
+    create_table,
     print_verbose,
     print_error,
     get_hotkey_wallets_for_wallet,
@@ -449,21 +450,13 @@ async def unstake_all(
             if not unstake_all_alpha
             else "Unstaking Summary - All Alpha Stakes"
         )
-        table = Table(
+        table = create_table(
             title=(
                 f"\n[{COLOR_PALETTE.G.HEADER}]{table_title}[/{COLOR_PALETTE.G.HEADER}]\n"
                 f"Wallet: [{COLOR_PALETTE.G.COLDKEY}]{wallet.name}[/{COLOR_PALETTE.G.COLDKEY}], "
                 f"Coldkey ss58: [{COLOR_PALETTE.G.CK}]{coldkey_ss58}[/{COLOR_PALETTE.G.CK}]\n"
                 f"Network: [{COLOR_PALETTE.G.HEADER}]{subtensor.network}[/{COLOR_PALETTE.G.HEADER}]\n"
             ),
-            show_footer=True,
-            show_edge=False,
-            header_style="bold white",
-            border_style="bright_black",
-            style="bold",
-            title_justify="center",
-            show_lines=False,
-            pad_edge=True,
         )
         table.add_column("Netuid", justify="center", style="grey89")
         table.add_column(
@@ -1054,16 +1047,8 @@ async def _unstake_selection(
 
     # Display existing hotkeys, id, and staked netuids.
     subnet_filter = f" for Subnet {netuid}" if netuid is not None else ""
-    table = Table(
+    table = create_table(
         title=f"\n[{COLOR_PALETTE.G.HEADER}]Hotkeys with Stakes{subnet_filter}\n",
-        show_footer=True,
-        show_edge=False,
-        header_style="bold white",
-        border_style="bright_black",
-        style="bold",
-        title_justify="center",
-        show_lines=False,
-        pad_edge=True,
     )
     table.add_column("Index", justify="right")
     table.add_column("Identity", style=COLOR_PALETTE.G.SUBHEAD)
@@ -1091,18 +1076,10 @@ async def _unstake_selection(
     netuid_stakes = hotkey_stakes[selected_hotkey_ss58]
 
     # Display hotkey's staked netuids with amount.
-    table = Table(
+    table = create_table(
         title=f"\n[{COLOR_PALETTE['GENERAL']['HEADER']}]Stakes for hotkey \n"
         f"[{COLOR_PALETTE['GENERAL']['SUBHEADING']}]{selected_hotkey_name}\n"
         f"{selected_hotkey_ss58}\n",
-        show_footer=True,
-        show_edge=False,
-        header_style="bold white",
-        border_style="bright_black",
-        style="bold",
-        title_justify="center",
-        show_lines=False,
-        pad_edge=True,
     )
     table.add_column("Subnet", justify="right")
     table.add_column("Symbol", style=COLOR_PALETTE["GENERAL"]["SYMBOL"])
@@ -1340,16 +1317,8 @@ def _create_unstake_table(
         f"Coldkey ss58: [{COLOR_PALETTE.G.CK}]{wallet_coldkey_ss58}[/{COLOR_PALETTE.G.CK}]\n"
         f"Network: {network}[/{COLOR_PALETTE.G.HEADER}]\n"
     )
-    table = Table(
+    table = create_table(
         title=title,
-        show_footer=True,
-        show_edge=False,
-        header_style="bold white",
-        border_style="bright_black",
-        style="bold",
-        title_justify="center",
-        show_lines=False,
-        pad_edge=True,
     )
 
     table.add_column("Netuid", justify="center", style="grey89")