File size: 1,305 Bytes
e5ec528
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
# encryption_tool/encryption_tool.py
from cryptography.fernet import Fernet


def generate_key():
    """
    Generates a secret key for encryption and decryption.
    
    Returns:
        str: The generated key.
    """
    key = Fernet.generate_key()
    return key.decode()


def encrypt_message(message, key):
    """
    Encrypts a message using the provided key.
    
    Args:
        message (str): The message to encrypt.
        key (str): The encryption key.
    
    Returns:
        str: The encrypted message.
    """
    try:
        fernet = Fernet(key.encode())
        encrypted_message = fernet.encrypt(message.encode())
        return encrypted_message.decode()
    except Exception as e:
        return f"Encryption failed: {str(e)}"


def decrypt_message(encrypted_message, key):
    """
    Decrypts an encrypted message using the provided key.
    
    Args:
        encrypted_message (str): The encrypted message to decrypt.
        key (str): The encryption key.
    
    Returns:
        str: The decrypted message or an error message.
    """
    try:
        fernet = Fernet(key.encode())
        decrypted_message = fernet.decrypt(encrypted_message.encode()).decode()
        return decrypted_message
    except Exception as e:
        return f"Decryption failed: {str(e)}"