Astroport.ONE/venv/lib/python3.11/site-packages/examples/save_encrypted_ascii_armor_message.py
2024-03-01 16:15:45 +01:00

65 lines
2.1 KiB
Python

# Copyright 2014-2024 Vincent Texier <vit@free.fr>
#
# DuniterPy is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# DuniterPy is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import getpass
from duniterpy import __version__
from duniterpy.key import AsciiArmor, SigningKey
# CONFIG #######################################
ENCRYPTED_AA_MESSAGE_PATH = "/tmp/duniter_aa_encrypted_message.txt"
################################################
def save_encrypted_ascii_armor_message():
# Ask public key of the recipient
pubkeyBase58 = input("Enter public key of the message recipient: ")
# prompt hidden user entry
salt = getpass.getpass("Enter your passphrase (salt): ")
# prompt hidden user entry
password = getpass.getpass("Enter your password: ")
# init SigningKey instance
signing_key = SigningKey.from_credentials(salt, password)
# Enter the message
message = input("Enter your message: ")
print(f"Message signed by puplic key: {signing_key.pubkey}")
comment = f"Generated by DuniterPy {__version__}"
# Encrypt the message, only the recipient secret key will be able to decrypt the message
encrypted_message = AsciiArmor.create(
message,
pubkeyBase58,
[signing_key],
message_comment=comment,
signatures_comment=comment,
)
# Save encrypted message in a file
with open(ENCRYPTED_AA_MESSAGE_PATH, "w", encoding="utf-8") as file_handler:
file_handler.write(encrypted_message)
print(f"Encrypted Ascii Armor Message saved in file {ENCRYPTED_AA_MESSAGE_PATH}")
if __name__ == "__main__":
save_encrypted_ascii_armor_message()