Добавлено декодирование словаря

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.")