Merge pull request #585 from opentensor/fix/thewhaleking/wallet-fixes

Author: thewhaleking

bittensor_cli/src/bittensor/utils.py

@@ -1443,5 +1443,5 @@ def get_hotkey_pub_ss58(wallet: Wallet) -> str:
     """
     try:
         return wallet.hotkeypub.ss58_address
-    except KeyFileError:
+    except (KeyFileError, AttributeError):
         return wallet.hotkey.ss58_address

bittensor_cli/src/commands/wallets.py

@@ -452,13 +452,15 @@ async def wallet_create(
         "error": "",
         "data": None,
     }
+
     if uri:
         try:
             keypair = Keypair.create_from_uri(uri)
-            wallet.set_coldkey(keypair=keypair, encrypt=False, overwrite=False)
-            wallet.set_coldkeypub(keypair=keypair, encrypt=False, overwrite=False)
-            wallet.set_hotkey(keypair=keypair, encrypt=False, overwrite=False)
-            wallet.set_coldkeypub(keypair=keypair, encrypt=False, overwrite=False)
+            wallet.set_coldkey(keypair=keypair, encrypt=False, overwrite=overwrite)
+            wallet.set_coldkeypub(keypair=keypair, encrypt=False, overwrite=overwrite)
+            wallet.set_hotkey(keypair=keypair, encrypt=False, overwrite=overwrite)
+            wallet.set_hotkeypub(keypair=keypair, encrypt=False, overwrite=overwrite)
+            wallet.set_coldkeypub(keypair=keypair, encrypt=False, overwrite=overwrite)
             output_dict["success"] = True
             output_dict["data"] = {
                 "name": wallet.name,