#-----------------------------------------------------------------#
# программа генерации пары ключей по шифру RSA
# !! требует установки библиотеки rsa !!!
# pip3 install rsa
# Функции:
#- выбор длины ключа
#- запись на диск пары ключей (открытый ключ и закрытый)
#- вывод на экран открытого ключа
#----------------------cyberbugz.ru-------------------------------#
import rsa                  # Внимание! Должна быть импортирована библиотека RSA
#from Crypto.PublicKey import RSA
#
# объявление переменных ------------------------------------------#
file_public="public.asc"     # имя открытого ключа
file_private="private.asc"   # имя закрытого ключа
key_size=512                 # длина блока ключа
# ТЕЛО ПРОГРАММЫ =================================================#
i=input("Укажите длину ключа: 1 - 1024 бит, 2-2048 бит, иначе: 512 бит")
if i==1:
    key_size=1024
elif i==2:
    key_size=2048
else:
    key_size=512
# генерация пары ключей по RSA
public_key, private_key = rsa.newkeys(key_size)
#
# запись ключей в файлы
with open(file_public, 'wb') as file:
    file.write(public_key.save_pkcs1())
with open(file_private, 'wb') as file:
    file.write(private_key.save_pkcs1())
#-----------------------------------------------------------------#
print("Ключи сгенерированы и записаны на диск")
print("Открытый ключ имеет вид: ")
print(public_key.save_pkcs1().decode())  #для наглядности декодируем из байтового вида
#------------------------------------------------------------------#
print("Пока!")