train / gen.py
ar08's picture
Create gen.py
408c595 verified
raw
history blame
503 Bytes
import paramiko
# Define key file names
private_key_file = 'test_rsa.key'
public_key_file = 'test_rsa.key.pub'
# Generate RSA key pair
key = paramiko.RSAKey.generate(bits=2048)
# Save private key
with open(private_key_file, 'wb') as f:
key.write_private_key_file(private_key_file)
# Save public key
with open(public_key_file, 'wb') as f:
f.write(key.get_base64())
print(f'RSA key pair generated and saved:')
print(f'Private key: {private_key_file}')
print(f'Public key: {public_key_file}')