diff --git a/scripts/regenerate_keys.py b/scripts/regenerate_keys.py index b3c505c45b..22bc3ca99b 100755 --- a/scripts/regenerate_keys.py +++ b/scripts/regenerate_keys.py @@ -55,6 +55,15 @@ def main(): default=True, help="Regenerate pool keys", ) + parser.add_argument( + "-w", + "--wallet", + type=str2bool, + nargs="?", + const=True, + default=True, + help="Regenerate wallet keys", + ) args = parser.parse_args() if key_config_filename.exists(): @@ -101,6 +110,11 @@ def main(): key_config["pool_target"] = pool_target.hex() with open(key_config_filename, "w") as f: safe_dump(key_config, f) + if args.wallet: + wallet_sk = PrivateKey.from_seed(token_bytes(32)) + key_config["wallet_sk"] = bytes(wallet_sk).hex() + with open(key_config_filename, "w") as f: + safe_dump(key_config, f) if __name__ == "__main__":