Encryptor-Decryptor-Files-Python/encrypt_decrypt_files_concurrent.py at main · mh97m/Encryptor-Decryptor-Files-Python · GitHub

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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
#!/usr/bin/env python
import os
import sys
import time
import argparse
# import smtplib
from cryptography.fernet import Fernet
from concurrent.futures import ThreadPoolExecutor


class EncryptorDecryptor:

    def __init__(self) -> None:
        args = self.getArgs()
        self.mode = args.mode
        self.num_encryption = args.num_enc
        self.directory = args.dir
        self.new_key = args.new_key

        self.keys = self.createKeys()

    def getArgs(self):
        parser = argparse.ArgumentParser(
            description='Encrypt or decrypt all files in a directory')
        parser.add_argument('--mode', choices=[
                            'encrypt', 'decrypt'], help='Mode to run the program in (encrypt or decrypt)')
        parser.add_argument('--dir', help='Directory to process')
        parser.add_argument('--num-enc', type=int, default=1,
                            help='Number of times each file should be encrypted or decrypted')
        parser.add_argument('--new-key', type=bool, default=False,
                            help='Make new key for encryption')
        return parser.parse_args()

    def createKeys(self):
        keys_file = os.path.join(os.getcwd(), 'key.txt')
        if self.new_key and self.mode == 'encrypt':
            keys = b''
            for i in range(0, self.num_encryption):
                key = Fernet.generate_key()
                keys += key + b' - '
            with open(keys_file, 'wb') as file:
                file.write(keys)
        elif os.path.exists(keys_file):
            with open(keys_file, 'rb') as file:
                keys = file.read()
        else:
            print('Key encryption not exists')
            exit()
        return keys.split(b' - ')[0:-1]

    def encryptData(self, file_name):
        with open(file_name, 'rb') as file:
            file_data = file.read()
        file_size = sys.getsizeof(file_data)
        for key in self.keys:
            fernet = Fernet(key)
            file_data = fernet.encrypt(file_data)
            time.sleep(file_size / (100  * 1024 * 1024))
        with open(file_name, 'wb') as new_file:
            new_file.write(file_data)

    def decryptData(self, file_name):
        with open(file_name, 'rb') as file:
            file_data = file.read()
        file_size = sys.getsizeof(file_data)
        for key in reversed(self.keys):
            fernet = Fernet(key)
            file_data = fernet.decrypt(file_data)
            time.sleep(file_size / (100  * 1024 * 1024))
        with open(file_name, 'wb') as new_file:
            new_file.write(file_data)

    # def sendKeysToEmail(self):

    #     # email content
    #     sender_email = "notification.3mail@gmail.com"
    #     receiver_email = "pejmanly@gmail.com"
    #     message = self.keys

    #     # SMTP server details
    #     smtp_server = "smtp.gmail.com"
    #     smtp_port = 587
    #     smtp_username = "notification.3mail@gmail.com"
    #     smtp_password = "(#)_(#)Notification.1209"

    #     # connect to SMTP server
    #     smtp = smtplib.SMTP(smtp_server, smtp_port)
    #     smtp.starttls() # start a secure connection
    #     smtp.login(smtp_username, smtp_password) # login

    #     # send email
    #     smtp.sendmail(sender_email, receiver_email, message)

    #     # close connection
    #     smtp.quit()

    def execute(self):
        with ThreadPoolExecutor() as executor:
            for root, _, files in os.walk(self.directory):
                for file in files:
                    file_name = os.path.join(root, file)
                    if self.mode == 'encrypt':
                        executor.submit(self.encryptData, file_name)
                    elif self.mode == 'decrypt':
                        executor.submit(self.decryptData, file_name)
        # self.sendKeysToEmail()


if __name__ == '__main__':
    start = time.monotonic()
    EncryptorDecryptor().execute()
    end = time.monotonic()
    print(f"Operation completed in: {end-start} seconds")