Merge pull request #3202 from opentensor/bug/roman/add_stake

[v10] Avoid negative balance to `add_stake`

Author: basfroman

bittensor/core/extrinsics/asyncex/staking.py

@@ -89,6 +89,13 @@ async def add_stake_extrinsic(
             subtensor.get_existential_deposit(block_hash=block_hash),
         )
 
+        # Leave existential balance to keep key alive.
+        if old_balance <= existential_deposit:
+            return ExtrinsicResponse(
+                False,
+                f"Balance ({old_balance}) is not enough to cover existential deposit `{existential_deposit}`.",
+            ).with_log()
+
         # Leave existential balance to keep key alive.
         if amount > old_balance - existential_deposit:
             # If we are staking all, we need to leave at least the existential deposit.

bittensor/core/extrinsics/staking.py

@@ -86,6 +86,13 @@ def add_stake_extrinsic(
         )
         existential_deposit = subtensor.get_existential_deposit(block=block)
 
+        # Leave existential balance to keep key alive.
+        if old_balance <= existential_deposit:
+            return ExtrinsicResponse(
+                False,
+                f"Balance ({old_balance}) is not enough to cover existential deposit `{existential_deposit}`.",
+            ).with_log()
+
         # Leave existential balance to keep key alive.
         if amount > old_balance - existential_deposit:
             # If we are staking all, we need to leave at least the existential deposit.