generate key for wallet

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__":