decrypt_code.py

import os

from cryptography.fernet import Fernet
from PyInquirer import Separator, prompt
from termcolor import colored
from stegano import lsb
import encryption.aes
import steganography.lsb

from util.key_gen import key_gen

# Defining the decryption function


def decrypt_func() -> None:
    # Asking the user for a prompt
    enc_info = prompt([
        {
            'type': 'list',
            'qmark': '>',
            'name': 'type_of_data',
            'message': 'What do you want to decrypt?',
            'choices': [
                Separator(),
                {
                    'name': 'Text',
                },
                {
                    'name': 'File',
                },
                {
                    'name': 'Image',
                },
            ],
        },


    ])

    if 'type_of_data' not in enc_info:
        # user hit Ctrl+C
        return

    # Storing the type of data in a variable
    type_of_data: str = enc_info['type_of_data']

    # Calling the appropriate function as per data
    if type_of_data == 'File':
        handle_file_dec()
    elif type_of_data == 'Image':
        handle_image_dec()
    else:
        handle_text_dec()


def handle_text_dec() -> None:
    # Using decryption information
    decrypt_info = prompt([
        {
            'type': 'input',
            'qmark': '>',
            'name': 'data',
            'message': 'Enter the text to decrypt.',
        },
        {
            'type': 'password',
            'qmark': '>',
            'name': 'password',
            'message': 'Enter password:',
        },
    ])

    if 'data' not in decrypt_info:
        # user hit Ctrl+C
        return

    # Storing data in variables
    encrypted_secret = decrypt_info['data']
    password = decrypt_info['password']

    decrypted_text = encryption.aes.decrypt_text(encrypted_secret, password)

    # Printing the text on the console
    print(colored('The decrypted text is: ', 'white') +
          colored(decrypted_text, 'green'))


def handle_file_dec() -> None:
    # Getting the file info
    file_info = prompt([
        {
            'type': 'input',
            'qmark': '>',
            'name': 'file_path',
            'message': 'Enter the path to the file.',
        },
        {
            'type': 'password',
            'qmark': '>',
            'name': 'password',
            'message': 'Enter the password: ',
        },
    ])

    if 'file_path' not in file_info:
        # user hit Ctrl+C
        return

    # Storing the password in a variable
    password = file_info['password']
    file_path = file_info['file_path']

    encryption.aes.decrypt_file(file_path, password)
    print(colored('File decrypted succesfully.', 'green'))

    # print(colored("Ran into an issue.", "red"))


def handle_image_dec():
    # Using decryption information
    decrypt_info = prompt([
        {
            'type': 'input',
            'qmark': '>',
            'name': 'image_path',
            'message': 'Enter the path of the image to decrypt.',
        },
        {
            'type': 'password',
            'qmark': '>',
            'name': 'password',
            'message': 'Enter password:',
        },
    ])

    image_path = decrypt_info['image_path']
    password = decrypt_info['password']


    # Trying to decrypt text
    data = steganography.lsb.decrypt_image(image_path)
    decrypted_text = encryption.aes.decrypt_text(data, password)


    # Printing the text on the console
    print(colored('The decrypted text is: ', 'white') +
          colored(decrypted_text, 'green'))