From e16fc55ce2dac35083e8ded2c81053792f921bcd Mon Sep 17 00:00:00 2001 From: stirelshka8_BigARM Date: Sun, 2 Mar 2025 17:52:42 +0300 Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5?= =?UTF-8?q?=D0=BD=D0=BE=20=D0=B4=D0=B5=D0=BA=D0=BE=D0=B4=D0=B8=D1=80=D0=BE?= =?UTF-8?q?=D0=B2=D0=B0=D0=BD=D0=B8=D0=B5=20=D1=81=D0=BB=D0=BE=D0=B2=D0=B0?= =?UTF-8?q?=D1=80=D1=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 28 +++++++++------------------- 1 file changed, 9 insertions(+), 19 deletions(-) diff --git a/main.py b/main.py index be5ae86..2055cea 100644 --- a/main.py +++ b/main.py @@ -1,45 +1,35 @@ -""" - This is a simple RSA encryptor/decryptor. - It uses a key pair which is stored in a 'keys' directory. - The keys are generated when you run the script. - The script will encrypt a message for you if you provide it. - The script will decrypt a message for you if you provide it. - The script will generate a key pair and save it in the 'keys' directory. -""" - import os +import json from RSA_Crypto.encrypt import RSAEncryptor from RSA_Crypto.decrypt import RSADecryptor from RSA_Crypto.generate_save import RSAKeyPair -# Переменные для генерации и сохранения ключей KEYS_DIR = 'keys' PRIVATE_KEY_FILE = f'{KEYS_DIR}/private-key.pem' PUBLIC_KEY_FILE = f'{KEYS_DIR}/public-key.pem' -# Создание директории для ключей if not os.path.isdir(KEYS_DIR): os.makedirs(KEYS_DIR, exist_ok=True) -# Генерация и сохранение ключей -key_pair = RSAKeyPair(PRIVATE_KEY_FILE, PUBLIC_KEY_FILE,) +key_pair = RSAKeyPair(PRIVATE_KEY_FILE, PUBLIC_KEY_FILE) key_pair.generate_and_save_keys() -# Шифрование сообщения -message = "{organization: 'ООО Рога, Копыта & Ко', license_expiration_date: '2024-01-01', license_number: '1234567890', count_licenses: 100}" +message = '{"organization": "ООО Рога, Копыта & Ко", "license_expiration_date": "2024-01-01", "license_number": "1234567890", "count_licenses": 100}' rsa_encryptor = RSAEncryptor(PUBLIC_KEY_FILE) encrypted_message = rsa_encryptor.encrypt_message(message) -# Запись зашифрованного сообщения в файл with open('output.txt', 'w') as f: f.write(encrypted_message) -# Дешифрование сообщения из файла rsa_decryptor = RSADecryptor(PRIVATE_KEY_FILE) -# Чтение зашифрованного сообщения из файла with open('output.txt', 'r') as f: encrypted_message_from_file = f.read() decrypted_message = rsa_decryptor.decrypt_message(encrypted_message_from_file) -print("Расшифрованное сообщение:", decrypted_message) + +try: + decrypted_message_dict = json.loads(decrypted_message) + print(decrypted_message_dict['organization']) +except json.JSONDecodeError: + print("Ошибка: Не удалось декодировать сообщение как JSON.")