class Decrypt_page:
def __init__(self, translations, mainwindow):
self.translations = translations
self.enc_key = ""
self.salt = ""
self.filepath = ""
self.filepath_rsa = ""
self.chosen_algo = ""
self.parent_win = mainwindow
def button_dec_t(self):
#self.bottom_widget_dec.setCurrentIndex(0)
if check_dark_mode(db_location) == "False":
pass
#self.btn_dec_t.setStyleSheet(DEC_TEXT_PRESSED_QSS)
#self.btn_dec_f.setStyleSheet(DEC_FILE_DEPRESSED_QSS)
else:
pass
#self.btn_dec_t.setStyleSheet(DARK_DEC_TEXT_PRESSED_QSS)
#self.btn_dec_f.setStyleSheet(DARK_DEC_FILE_DEPRESSED_QSS)
def button_dec_f(self):
#self.bottom_widget_dec.setCurrentIndex(1)
if check_dark_mode(db_location) == "False":
pass
#self.btn_dec_t.setStyleSheet(DEC_TEXT_DEPRESSED_QSS)
#self.btn_dec_f.setStyleSheet(DEC_FILE_PRESSED_QSS)
else:
pass
#self.btn_dec_t.setStyleSheet(DARK_DEC_TEXT_DEPRESSED_QSS)
#self.btn_dec_f.setStyleSheet(DARK_DEC_FILE_PRESSED_QSS)
def decryption(self):
"""
This method handles the frame for the entire decrypt tab
#final_layout = QVBoxLayout()
# DEFINE TOP WIDGET (TABS AND SWITCHING BETWEEN THEM)
#dec_button_text = self.translations["buttons"]["decrypt_text"]
#dec_button_files = self.translations["buttons"]["decrypt_files"]
#self.btn_dec_t = QPushButton(f"{dec_button_text}")
#self.btn_dec_t.setObjectName("btn_dec_t")
#self.btn_dec_t.clicked.connect(self.button_dec_t)
#self.btn_dec_f = QPushButton(f"{dec_button_files}")
#self.btn_dec_f.setObjectName("btn_dec_f")
#self.btn_dec_f.clicked.connect(self.button_dec_f)
#if check_dark_mode(db_location) == "False":
#self.btn_dec_t.setStyleSheet(DEC_TEXT_PRESSED_QSS)
#else:
#self.btn_dec_t.setStyleSheet(DARK_DEC_TEXT_PRESSED_QSS)
top_actions = QHBoxLayout()
top_actions.setSpacing(0)
top_actions.setContentsMargins(0, 16, 0, 0)
#top_actions.addWidget(self.btn_dec_t)
#top_actions.addWidget(self.btn_dec_f)
self.top_widget_dec = QWidget()
self.top_widget_dec.setLayout(top_actions)
# DEFINE BOTTOM WIDGET (TAB CONTENTS)
#self.tab_dec_t = self.tab_dec_text()
#self.tab_dec_f = self.tab_dec_files()
self.bottom_widget_dec = QTabWidget()
self.bottom_widget_dec.tabBar().setObjectName("DecryptionTab")
#self.bottom_widget_dec.addTab(self.tab_dec_t, "")
#self.bottom_widget_dec.addTab(self.tab_dec_f, "")
self.bottom_widget_dec.setCurrentIndex(1) # default to text decryption tab
# add top and bottom widgets to layout
final_layout.addWidget(self.top_widget_dec)
final_layout.addWidget(self.bottom_widget_dec)
# Finish layout
main = QWidget()
main.setLayout(final_layout)
return main
"""
# init layout
self.layout = QGridLayout()
title_label = QLabel(self.translations["labels"]["decryption_file"])
title_label.setAlignment(Qt.AlignCenter)
title_label.setObjectName("title_label")
title_label.setSizePolicy(
QSizePolicy(QSizePolicy.Policy.Ignored, QSizePolicy.Policy.Fixed))
title_label.setAlignment(Qt.AlignmentFlag.AlignHCenter)
self.layout.addWidget(title_label, 0, 1, 1, 8)
pad = QLabel(" ")
self.layout.addWidget(pad, 1, 0, 1, 2)
self.layout.addWidget(pad, 1, 8, 1, 2)
# FILE BROWSE LABEL
open_file_label = QLabel(
self.translations["labels"]["insert_file_dec"])
open_file_label.setObjectName("large_label")
open_file_label.setAlignment(Qt.AlignCenter)
self.layout.addWidget(open_file_label, 1, 2, 1, 3)
# FILE BROWSE
open_file_btn = QPushButton(
self.translations["buttons"]["browse_files"])
open_file_btn.clicked.connect(self.filedialogopen)
self.layout.addWidget(open_file_btn, 1, 5, 1, 3)
# ALGORITHM LABEL
algo_label = QLabel(self.translations["labels"]["set_dec_algorithm"])
self.layout.addWidget(algo_label, 2, 2, 1, 3)
# ALGORITHM DROPDOWN MENU
self.algo_button = QPushButton(
self.translations["buttons"]["algorithm"])
self.algo_dropdown = QMenu()
self.algo_dropdown.setObjectName("algo_menu_dec")
for algo in ENC_ALGORITHMS_FILES:
self.algo_dropdown.addAction(algo)
self.algo_dropdown.addSeparator()
self.algo_button.setMenu(self.algo_dropdown)
self.algo_dropdown.triggered.connect(self.algorithms)
self.layout.addWidget(self.algo_button, 2, 5, 1, 3)
# CUSTOM RSA KEY SELECTION LABEL
self.rsa_key_selection_label = QLabel(
self.translations["labels"]["encryption_rsa_key_label"])
self.layout.addWidget(self.rsa_key_selection_label, 3, 3, 1, 1)
self.rsa_key_selection_label.setHidden(True)
# CUSTOM RSA KEY FILEOPEN PROMPT
self.rsa_selection_btn = QPushButton(
self.translations["buttons"]["browse_files"])
self.rsa_selection_btn.setText("private.pem")
self.rsa_selection_btn.clicked.connect(self.filedialogopen_rsa)
self.layout.addWidget(self.rsa_selection_btn, 3, 5, 1, 3)
self.rsa_selection_btn.setHidden(True)
# ENCRYPTION KEY LABEL
self.enc_key_label = QLabel(
self.translations["labels"]["encryption_key_label"])
self.enc_key_label.setAlignment(Qt.AlignCenter)
self.layout.addWidget(self.enc_key_label, 3, 2, 1, 2)
# ENCRYPTION KEY INPUT
self.text_box_dec_text = PasswordEdit()
self.layout.addWidget(self.text_box_dec_text, 3, 4, 1, 5)
# ENCRYPTION SALT LABEL
self.enc_salt_label = QLabel(self.translations["labels"]["salt_label"])
self.layout.addWidget(self.enc_salt_label, 4, 2, 1, 3)
# ENCRYPTION SALT INPUT
self.text_box_salt_text = PasswordEdit()
self.layout.addWidget(self.text_box_salt_text, 4, 4, 1, 5)
# DECRYPT BUTTON
decrypt_button = QPushButton(
self.translations["buttons"]["final_decrypt"])
decrypt_button.clicked.connect(self.decrypt_file)
self.layout.addWidget(decrypt_button, 5, 2, 1, 6)
# finish layout
main = QWidget()
main.setLayout(self.layout)
return main
def tab_dec_text(self):
"""
This method handles the text decryption tab
"""
# init layout and set suitable column widths
layout = QGridLayout()
pad = QLabel()
layout.addWidget(pad, 0, 0, 1, 1)
layout.addWidget(pad, 0, 9, 1, 1)
# INSERT TEXT LABEL
text_ins_label = QLabel(self.translations["labels"]["insert_text_dec"])
text_ins_label.setObjectName(
"large_label") # set object name for qss tag effects
text_ins_label.setAlignment(Qt.AlignCenter)
layout.addWidget(text_ins_label, 0, 1, 1, 3)
# INSERT TEXT BOX
text_insert = QLineEdit()
layout.addWidget(text_insert, 0, 4, 1, 5)
# ALGORITHM LABEL
algo_label = QLabel(self.translations["labels"]["set_dec_algorithm"])
algo_label.setAlignment(Qt.AlignCenter)
layout.addWidget(algo_label, 1, 1, 1, 3)
# ALGORITHM DROPDOWN MENU
self.algo_button_ttab = QPushButton(
self.translations["buttons"]["algorithm"])
self.algo_dropdown = QMenu()
self.algo_dropdown.setObjectName("algo_menu_dec")
for algo in ENC_ALGORITHMS:
self.algo_dropdown.addAction(algo)
self.algo_dropdown.addSeparator()
self.algo_button_ttab.setMenu(self.algo_dropdown)
self.algo_dropdown.triggered.connect(self.algorithms_text_tab)
layout.addWidget(self.algo_button_ttab, 1, 4, 1, 3)
# ENCRYPTION SALT LABEL
self.enc_salt_label = QLabel(self.translations["labels"]["salt_label"])
self.enc_salt_label.setAlignment(Qt.AlignCenter)
layout.addWidget(self.enc_salt_label, 2, 1, 1, 3)
# ENCRYPTION SALT INPUT
self.text_box_salt_text = PasswordEdit()
layout.addWidget(self.text_box_salt_text, 2, 4, 1, 5)
# DECRYPT BUTTON
decrypt_button = QPushButton(
self.translations["buttons"]["final_decrypt"])
layout.addWidget(decrypt_button, 3, 2, 1, 6)
main = QWidget()
main.setLayout(layout)
return main
def filedialogopen(self):
self._files = FileDialog().fileOpen()
self.filepath = self._files
def filedialogopen_rsa(self):
"""
File dialog opening method
"""
self._files_rsa = FileDialog().fileOpen()
self.filepath_rsa = self._files_rsa
fileout = ""
try:
fileout = os.path.basename(self.filepath_rsa)
self.rsa_selection_btn.setText(fileout)
except TypeError:
self.rsa_selection_btn.setText("private.pem")
def filedialogsave(self):
self._save = FileDialog().fileSave()
# Decrypt parameters set and function call
def decrypt_file(self):
self.enc_key = self.text_box_dec_text.text()
self.salt = self.text_box_salt_text.text()
salt = self.salt
print("salt:", salt)
filepath = self.filepath
try:
fileout = os.path.basename(self.filepath)
except TypeError:
return
enc_key = self.enc_key
print(enc_key)
# File out gets the name of the file for saving the file
if self.chosen_algo == "AES":
decryptor = decrypt.Decryption(password=enc_key, salt=salt)
result = decryptor.decrypt_with_aes(filepath, fileout)
if result == -1:
print("failed decrypting")
failed_decrypt = self.translations["prompts"]["failed_decrypt"]
msg = QMessageBox()
msg.setIcon(QMessageBox.Warning)
msg.setText(failed_decrypt)
msg.exec_()
return
inprogresslist.append(f"Decrypted: {fileout}")
progress = self.translations["prompts"]["ready"]
self.parent_win.right_layout.clear()
self.parent_win.right_layout.addItems(
[f"{progress} ({len(inprogresslist)})"])
self.parent_win.right_layout.addItems(inprogresslist)
self.parent_win.right_layout.setHidden(False)
return
if self.chosen_algo == "RSA":
if self.filepath_rsa == "":
decryptor = decrypt.Decryption(password=enc_key, salt=salt)
result = decryptor.decrypt_with_rsa(filename=filepath,
priv_key="private.pem",
fileout=fileout)
if result == -2:
no_RSA_keys = self.translations["prompts"]["no_rsa_keys"]
print("Cant open key file")
msg = QMessageBox()
msg.setIcon(QMessageBox.Warning)
msg.setText(no_RSA_keys)
msg.exec_()
return
if result == -1:
print("failed decrypting")
failed_decrypt = self.translations["prompts"][
"failed_decrypt"]
msg = QMessageBox()
msg.setIcon(QMessageBox.Warning)
msg.setText(failed_decrypt)
msg.exec_()
return
else:
decryptor = decrypt.Decryption(password=enc_key, salt=salt)
result = decryptor.decrypt_with_rsa(filename=filepath,
priv_key=self.filepath_rsa,
fileout=fileout)
if result == -2:
no_RSA_keys = self.translations["prompts"]["no_rsa_keys"]
print("Cant open key file")
msg = QMessageBox()
msg.setIcon(QMessageBox.Warning)
msg.setText(no_RSA_keys)
msg.exec_()
return
if result == -1:
print("failed decrypting")
failed_decrypt = self.translations["prompts"][
"failed_decrypt"]
msg = QMessageBox()
msg.setIcon(QMessageBox.Warning)
msg.setText(failed_decrypt)
msg.exec_()
return
inprogresslist.append(f"Decrypted: {fileout}")
progress = self.translations["prompts"]["ready"]
self.parent_win.right_layout.clear()
self.parent_win.right_layout.addItems(
[f"{progress} ({len(inprogresslist)})"])
self.parent_win.right_layout.addItems(inprogresslist)
self.parent_win.right_layout.setHidden(False)
return
if self.chosen_algo == "Chacha":
decryptor = decrypt.Decryption(password=enc_key, salt=salt)
result = decryptor.decrypt_with_chacha(filepath, fileout)
if result == -1:
print("failed decrypting")
failed_decrypt = self.translations["prompts"]["failed_decrypt"]
msg = QMessageBox()
msg.setIcon(QMessageBox.Warning)
msg.setText(failed_decrypt)
msg.exec_()
return
inprogresslist.append(f"Decrypted: {fileout}")
progress = self.translations["prompts"]["ready"]
self.parent_win.right_layout.clear()
self.parent_win.right_layout.addItems(
[f"{progress} ({len(inprogresslist)})"])
self.parent_win.right_layout.addItems(inprogresslist)
self.parent_win.right_layout.setHidden(False)
return
# Filepath is the path for the file
# Fileout is the name of the file, comes out with added
# .dec prefix after decryption
return
def algorithms(self, algorithm):
"""
Change the encryption button text to chosen algorithm
"""
disabled_password = self.translations["prompts"]["encryption_disabled"]
disabled_salt = self.translations["prompts"]["salt_disabled"]
self.chosen_algo = algorithm.text()
self.algo_button.setText(self.chosen_algo)
if self.chosen_algo == "RSA":
self.text_box_dec_text.setHidden(True)
# self.text_box_dec_text.setToolTip(disabled_password)
self.text_box_salt_text.setHidden(True)
# self.text_box_salt_text.setToolTip(disabled_salt)
self.rsa_key_selection_label.setHidden(False)
self.rsa_selection_btn.setHidden(False)
self.enc_key_label.setHidden(True)
self.enc_salt_label.setHidden(True)
else:
self.text_box_dec_text.setHidden(False)
# self.text_box_dec_text.setToolTip("")
# self.text_box_dec_text.setToolTip("")
self.text_box_salt_text.setHidden(False)
# self.text_box_salt_text.setToolTip("")
self.rsa_key_selection_label.setHidden(True)
self.rsa_selection_btn.setHidden(True)
self.enc_key_label.setHidden(False)
self.enc_salt_label.setHidden(False)
self.layout.update()
return algorithm
def algorithms_text_tab(self, algorithm):
"""
Change the encryption button text to chosen algorithm
"""
self.chosen_algo = algorithm.text()
self.algo_button_ttab.setText(self.chosen_algo)
self.layout.update()
return algorithm
def tab_dec_files(self):
"""
This method handles the file decryption tab
"""
# init layout
self.layout = QGridLayout()
pad = QLabel(" ")
self.layout.addWidget(pad, 0, 0, 1, 2)
self.layout.addWidget(pad, 0, 8, 1, 2)
# FILE BROWSE LABEL
open_file_label = QLabel(
self.translations["labels"]["insert_file_dec"])
open_file_label.setObjectName("large_label")
open_file_label.setAlignment(Qt.AlignCenter)
self.layout.addWidget(open_file_label, 0, 2, 1, 3)
# FILE BROWSE
open_file_btn = QPushButton(
self.translations["buttons"]["browse_files"])
open_file_btn.clicked.connect(self.filedialogopen)
self.layout.addWidget(open_file_btn, 0, 5, 1, 3)
# ALGORITHM LABEL
algo_label = QLabel(self.translations["labels"]["set_dec_algorithm"])
self.layout.addWidget(algo_label, 1, 2, 1, 3)
# ALGORITHM DROPDOWN MENU
self.algo_button = QPushButton(
self.translations["buttons"]["algorithm"])
self.algo_dropdown = QMenu()
self.algo_dropdown.setObjectName("algo_menu_dec")
for algo in ENC_ALGORITHMS_FILES:
self.algo_dropdown.addAction(algo)
self.algo_dropdown.addSeparator()
self.algo_button.setMenu(self.algo_dropdown)
self.algo_dropdown.triggered.connect(self.algorithms)
self.layout.addWidget(self.algo_button, 1, 5, 1, 3)
# CUSTOM RSA KEY SELECTION LABEL
self.rsa_key_selection_label = QLabel(
self.translations["labels"]["encryption_rsa_key_label"])
self.layout.addWidget(self.rsa_key_selection_label, 2, 3, 1, 1)
self.rsa_key_selection_label.setHidden(True)
# CUSTOM RSA KEY FILEOPEN PROMPT
self.rsa_selection_btn = QPushButton(
self.translations["buttons"]["browse_files"])
self.rsa_selection_btn.setText("private.pem")
self.rsa_selection_btn.clicked.connect(self.filedialogopen_rsa)
self.layout.addWidget(self.rsa_selection_btn, 2, 5, 1, 3)
self.rsa_selection_btn.setHidden(True)
# ENCRYPTION KEY LABEL
self.enc_key_label = QLabel(
self.translations["labels"]["encryption_key_label"])
self.enc_key_label.setAlignment(Qt.AlignCenter)
self.layout.addWidget(self.enc_key_label, 2, 2, 1, 2)
# ENCRYPTION KEY INPUT
self.text_box_dec_text = PasswordEdit()
self.layout.addWidget(self.text_box_dec_text, 2, 4, 1, 5)
# ENCRYPTION SALT LABEL
self.enc_salt_label = QLabel(self.translations["labels"]["salt_label"])
self.layout.addWidget(self.enc_salt_label, 3, 2, 1, 3)
# ENCRYPTION SALT INPUT
self.text_box_salt_text = PasswordEdit()
self.layout.addWidget(self.text_box_salt_text, 3, 4, 1, 5)
# DECRYPT BUTTON
decrypt_button = QPushButton(
self.translations["buttons"]["final_decrypt"])
decrypt_button.clicked.connect(self.decrypt_file)
self.layout.addWidget(decrypt_button, 4, 2, 1, 6)
# finish layout
main = QWidget()
main.setLayout(self.layout)
return main
class Encrypt_page:
def __init__(self, translations, mainwindow):
# Define used class parameters to be set in the selections
self.translations = translations
self.defaults = check_encryption_defaults(db_location)
self.filepath = ""
self.filepath_rsa = ""
self.salt = ""
self.enc_key = ""
self.parent_win = mainwindow
if self.defaults["default_hash"] != "":
self.chosen_algo = self.defaults["default_hash"]
else:
self.chosen_algo = ""
if self.defaults["default_algo"] != "":
self.chosen_algorithm = self.defaults["default_algo"]
else:
self.chosen_algorithm = ""
def button_enc_t(self):
self.bottom_widget.setCurrentIndex(0)
if check_dark_mode(db_location) == "False":
self.btn_enc_t.setStyleSheet(ENC_TEXT_PRESSED_QSS)
self.btn_enc_f.setStyleSheet(ENC_FILE_DEPRESSED_QSS)
else:
self.btn_enc_t.setStyleSheet(DARK_ENC_TEXT_PRESSED_QSS)
self.btn_enc_f.setStyleSheet(DARK_ENC_FILE_DEPRESSED_QSS)
def button_enc_f(self):
self.bottom_widget.setCurrentIndex(1)
if check_dark_mode(db_location) == "False":
self.btn_enc_t.setStyleSheet(ENC_TEXT_DEPRESSED_QSS)
self.btn_enc_f.setStyleSheet(ENC_FILE_PRESSED_QSS)
else:
self.btn_enc_t.setStyleSheet(DARK_ENC_TEXT_DEPRESSED_QSS)
self.btn_enc_f.setStyleSheet(DARK_ENC_FILE_PRESSED_QSS)
def encryption(self):
"""
This method handles frame for the entire encrypt tab
"""
final_layout = QVBoxLayout()
# DEFINE TOP WIDGET (TABS AND SWITCHING BETWEEN THEM)
enc_button_text = self.translations["buttons"]["encrypt_text"]
enc_button_files = self.translations["buttons"]["encrypt_files"]
self.btn_enc_t = QPushButton(f"{enc_button_text}")
self.btn_enc_t.setObjectName("btn_enc_t")
self.btn_enc_t.clicked.connect(self.button_enc_t)
self.btn_enc_f = QPushButton(f"{enc_button_files}")
self.btn_enc_f.setObjectName("btn_enc_f")
self.btn_enc_f.clicked.connect(self.button_enc_f)
if check_dark_mode(db_location) == "False":
self.btn_enc_t.setStyleSheet(ENC_TEXT_PRESSED_QSS)
else:
self.btn_enc_t.setStyleSheet(DARK_ENC_TEXT_PRESSED_QSS)
top_actions = QHBoxLayout()
top_actions.setSpacing(0)
top_actions.setContentsMargins(0, 16, 0, 0)
top_actions.addWidget(self.btn_enc_t)
top_actions.addWidget(self.btn_enc_f)
self.top_widget = QWidget()
self.top_widget.setLayout(top_actions)
# DEFINE BOTTOM WIDGET (TAB CONTENTS)
self.tab_enc_t = self.tab_enc_text()
self.tab_enc_f = self.tab_enc_files()
self.bottom_widget = QTabWidget()
self.bottom_widget.tabBar().setObjectName("EncryptionTab")
self.bottom_widget.addTab(self.tab_enc_t, "")
self.bottom_widget.addTab(self.tab_enc_f, "")
self.bottom_widget.setCurrentIndex(0) # default to the text tab
# Add top and bottom parts to the layout
final_layout.addWidget(self.top_widget)
final_layout.addWidget(self.bottom_widget)
# Finish layout
main = QWidget()
main.setLayout(final_layout)
return main
def tab_enc_text(self):
"""
This method handles the text encryption tab
"""
# init layout
layout = QGridLayout()
# INSERT TEXT LABEL
text_to_enc_label = QLabel(self.translations["labels"]["insert_text_enc"])
text_to_enc_label.setAlignment(Qt.AlignCenter)
text_to_enc_label.setObjectName("large_label")
layout.addWidget(text_to_enc_label, 0, 1, 1, 3)
# INSERT TEXT BOX
self.text_insert = QLineEdit()
layout.addWidget(self.text_insert, 0, 4, 1, 5)
# ALGORITHM SET LABEL
algo_text_label = QLabel(self.translations["labels"]["set_enc_algorithm"])
algo_text_label.setAlignment(Qt.AlignCenter)
layout.addWidget(algo_text_label, 1, 1, 1, 3)
# ALGORITHM DROPDOWN MENU
algo_trans = self.translations["buttons"]["algorithm"]
self.algo_button_ttab = QPushButton(algo_trans)
self.algo_dropdown = QMenu()
self.algo_dropdown.setObjectName("algo_menu_enc_text")
for algo in ENC_ALGORITHMS:
self.algo_dropdown.addAction(algo)
self.algo_dropdown.addSeparator()
self.algo_button_ttab.setMenu(self.algo_dropdown)
self.algo_dropdown.triggered.connect(self.algorithms_text)
if self.defaults["default_hash"] != "":
self.algo_button_ttab.setText(self.defaults["default_hash"])
layout.addWidget(self.algo_button_ttab, 1, 4, 1, 3)
# ENCRYPTION KEY INPUT AND CONFIRM LABELS
enc_text_label = QLabel(self.translations["labels"]["encryption_key_label"])
enc_text_label.setAlignment(Qt.AlignCenter)
enc_conf_label = QLabel(
self.translations["labels"]["encryption_key_confirm_label"]
)
enc_conf_label.setAlignment(Qt.AlignCenter)
enc_text_label.setHidden(True)
enc_conf_label.setHidden(True)
layout.addWidget(enc_text_label, 2, 3, 1, 1)
layout.addWidget(enc_conf_label, 3, 2, 1, 2)
# ENCRYPTION KEY INPUT AND CONFIRM
self.text_box_enc_text_ttab = PasswordEdit()
self.text_box_enc_text_ttab.setHidden(True)
if self.defaults["default_key"] != "":
self.text_box_enc_text_ttab.setText(self.defaults["default_key"])
self.text_box_enc_text_confirm_ttab = PasswordEdit()
self.text_box_enc_text_confirm_ttab.setHidden(True)
if self.defaults["default_key"] != "":
self.text_box_enc_text_confirm_ttab.setText(self.defaults["default_key"])
layout.addWidget(self.text_box_enc_text_ttab, 2, 4, 1, 3)
layout.addWidget(self.text_box_enc_text_confirm_ttab, 3, 4, 1, 3)
# SALT INPUT LABEL
salt_label = QLabel(self.translations["labels"]["salt_label"])
salt_label.setAlignment(Qt.AlignCenter)
salt_label.setObjectName("large_label")
layout.addWidget(salt_label, 4, 1, 1, 3)
# SALT INPUT
self.salt_insert_box_ttab = PasswordEdit()
if self.defaults["default_salt"] != "":
self.salt_insert_box_ttab.setText(self.defaults["default_salt"])
layout.addWidget(self.salt_insert_box_ttab, 4, 4, 1, 5)
# ENCRYPT BUTTON
enc_trans = self.translations["buttons"]["final_encrypt"]
encrypt_button = QPushButton(enc_trans)
encrypt_button.clicked.connect(self.encrypt_text)
self.encrypt_result = QLineEdit()
self.encrypt_result.setHidden(True)
layout.addWidget(encrypt_button, 5, 2, 1, 6)
layout.addWidget(self.encrypt_result, 6, 0, 1, 10)
# finish and set layout
main = QWidget()
main.setLayout(layout)
return main
def filedialogopen(self):
"""
File dialog opening method
"""
self._files = FileDialog().fileOpen()
self.filepath = self._files
def filedialogopen_rsa(self):
"""
File dialog opening method
"""
self._files_rsa = FileDialog().fileOpen()
self.filepath_rsa = self._files_rsa
if self.filepath_rsa != None:
fileout = os.path.basename(self.filepath_rsa)
self.rsa_selection_btn.setText(fileout)
return
fileout = "public.pem"
self.rsa_selection_btn.setText(fileout)
def filedialogsave(self):
"""
File save method
"""
self._save = FileDialog().fileSave()
# Encrypt parameters set and function call
def encrypt_file(self):
self.enc_key = self.text_box_enc_text.text()
self.enc_key_confirm = self.text_box_enc_text_confirm.text()
self.salt = self.salt_insert_box.text()
filepath = self.filepath
try:
fileout = os.path.basename(self.filepath)
except TypeError:
return
salt = self.salt
enc_key = self.enc_key
print(self.chosen_algorithm)
if str(self.enc_key) != str(self.enc_key_confirm):
pwd_mismatch = self.translations["prompts"]["password_mismatch"]
msg = QMessageBox()
msg.setIcon(QMessageBox.Warning)
msg.setText(pwd_mismatch)
display = msg.exec_()
return
# File out gets the name of the file for saving the file
if self.chosen_algorithm == "AES":
encryptor = encrypt.Encryption(password=enc_key, salt=salt)
encryptor.encrypt_with_aes(filepath, fileout)
if self.chosen_algorithm == "RSA":
if self.filepath_rsa == "":
encryptor = encrypt.Encryption(password=enc_key, salt=salt)
encryptor.encrypt_with_rsa(
filename=filepath, fileout=fileout, pub_key=None
)
else:
encryptor = encrypt.Encryption(password=enc_key, salt=salt)
encryptor.encrypt_with_rsa(
filename=filepath, fileout=fileout, pub_key=self.filepath_rsa
)
if self.chosen_algorithm == "Chacha":
encryptor = encrypt.Encryption(password=enc_key, salt=salt)
encryptor.encrypt_with_chacha(filepath, fileout)
# Filepath is the path for the file
# Fileout is the name of the file, comes out with added
# _encryted prefix after ecnryption
inprogresslist.append(f"Encrypted: {fileout}")
progress = self.translations["prompts"]["ready"]
self.parent_win.right_layout.clear()
self.parent_win.right_layout.addItems([f"{progress} ({len(inprogresslist)})"])
self.parent_win.right_layout.addItems(inprogresslist)
self.parent_win.right_layout.setHidden(False)
return
def encrypt_text(self):
result = ""
self.encrypt_result.setHidden(False)
text_hasher = encrypt.Encryption(salt=self.salt_insert_box_ttab.text())
if self.chosen_algo == "MD5":
result = text_hasher.hash_with_md5(self.text_insert.text())
if self.chosen_algo == "SHA-256":
result = text_hasher.hash_with_sha256(self.text_insert.text())
if self.chosen_algo == "SHA-512":
result = text_hasher.hash_with_sha512(self.text_insert.text())
if self.chosen_algo == "SHA3-512":
result = text_hasher.hash_with_sha3_512(self.text_insert.text())
self.encrypt_result.setText(result)
def algorithms(self, algorithm):
disabled_password = self.translations["prompts"]["encryption_disabled"]
disabled_salt = self.translations["prompts"]["salt_disabled"]
self.chosen_algorithm = algorithm.text()
self.algo_button.setText(self.chosen_algorithm)
if self.chosen_algorithm == "RSA":
self.enc_text_label.setHidden(True)
self.enc_conf_label.setHidden(True)
self.salt_label.setHidden(True)
self.text_box_enc_text.setHidden(True)
self.text_box_enc_text.setDisabled(True)
self.text_box_enc_text.setToolTip(disabled_password)
self.text_box_enc_text_confirm.setHidden(True)
self.text_box_enc_text_confirm.setDisabled(True)
self.text_box_enc_text_confirm.setToolTip(disabled_password)
self.salt_insert_box.setHidden(True)
self.salt_insert_box.setDisabled(True)
self.salt_insert_box.setToolTip(disabled_salt)
self.rsa_key_selection_label.setHidden(False)
self.rsa_selection_btn.setHidden(False)
else:
self.enc_text_label.setHidden(False)
self.enc_conf_label.setHidden(False)
self.salt_label.setHidden(False)
self.text_box_enc_text.setHidden(False)
self.text_box_enc_text.setDisabled(False)
self.text_box_enc_text.setToolTip("")
self.text_box_enc_text_confirm.setHidden(False)
self.text_box_enc_text_confirm.setDisabled(False)
self.text_box_enc_text.setToolTip("")
self.salt_insert_box.setHidden(False)
self.salt_insert_box.setDisabled(False)
self.text_box_enc_text.setToolTip("")
self.rsa_key_selection_label.setHidden(True)
self.rsa_selection_btn.setHidden(True)
self.layout.update()
return algorithm
def algorithms_text(self, algorithm):
"""
Change the encryption button text to chosen algorithm
"""
self.chosen_algo = algorithm.text()
self.algo_button_ttab.setText(self.chosen_algo)
self.layout.update()
return algorithm
def tab_enc_files(self):
"""
This method handles the file encryption tab
"""
# init layout
self.layout = QGridLayout()
self.layout.setContentsMargins(0, 0, 0, 0)
pad = QLabel(" ")
self.layout.addWidget(pad, 0, 8, 1, 2)
self.layout.addWidget(pad, 0, 0, 1, 1)
# FILE BROWSER LABEL
file_browse_label = QLabel(self.translations["labels"]["browse_file_enc"])
file_browse_label.setObjectName("large_label")
file_browse_label.setAlignment(Qt.AlignCenter)
self.layout.addWidget(file_browse_label, 0, 2, 1, 3)
# INSERT FILE BROWSER
file_browse_btn = QPushButton(self.translations["buttons"]["browse_files"])
file_browse_btn.clicked.connect(self.filedialogopen)
self.layout.addWidget(file_browse_btn, 0, 5, 1, 3)
# ALGORITHM SET LABEL
self.algo_text_label = QLabel(self.translations["labels"]["set_enc_algorithm"])
self.algo_text_label.setAlignment(Qt.AlignCenter)
self.layout.addWidget(self.algo_text_label, 1, 2, 1, 3)
# ALGORITHM DROPDOWN MENU
self.algo_button = QPushButton(self.translations["buttons"]["algorithm"])
self.algo_dropdown = QMenu()
self.algo_dropdown.setObjectName("algo_menu_enc_files")
for algo in ENC_ALGORITHMS_FILES:
self.algo_dropdown.addAction(algo)
self.algo_dropdown.addSeparator()
self.algo_button.setMenu(self.algo_dropdown)
self.algo_dropdown.triggered.connect(self.algorithms)
if self.defaults["default_algo"] != "":
self.algo_button.setText(self.defaults["default_algo"])
# if self.algo_dropdown.triggered:
# self.algo_button.setText(self.chosen_algo)
# self.layout.update()
self.layout.addWidget(self.algo_button, 1, 5, 1, 3)
# CUSTOM RSA KEY SELECTION LABEL
self.rsa_key_selection_label = QLabel(
self.translations["labels"]["encryption_rsa_key_label"]
)
self.layout.addWidget(self.rsa_key_selection_label, 2, 3, 1, 1)
self.rsa_key_selection_label.setHidden(True)
# CUSTOM RSA KEY FILEOPEN PROMPT
self.rsa_selection_btn = QPushButton(
self.translations["buttons"]["browse_files"]
)
self.rsa_selection_btn.setText("public.pem")
self.rsa_selection_btn.clicked.connect(self.filedialogopen_rsa)
self.layout.addWidget(self.rsa_selection_btn, 2, 5, 1, 3)
self.rsa_selection_btn.setHidden(True)
# ENCRYPTION KEY INPUT AND CONFIRM LABELS
self.enc_text_label = QLabel(
self.translations["labels"]["encryption_key_label"]
)
self.enc_conf_label = QLabel(
self.translations["labels"]["encryption_key_confirm_label"]
)
self.enc_text_label.setAlignment(Qt.AlignCenter)
self.enc_conf_label.setAlignment(Qt.AlignCenter)
self.layout.addWidget(self.enc_text_label, 2, 2, 1, 2)
self.layout.addWidget(self.enc_conf_label, 3, 2, 1, 2)
# ENCRYPTION KEY INPUT AND CONFIRM
self.text_box_enc_text = PasswordEdit()
if self.defaults["default_key"] != "":
self.text_box_enc_text.setText(self.defaults["default_key"])
self.text_box_enc_text_confirm = PasswordEdit()
if self.defaults["default_key"] != "":
self.text_box_enc_text_confirm.setText(self.defaults["default_key"])
self.layout.addWidget(self.text_box_enc_text, 2, 4, 1, 5)
self.layout.addWidget(self.text_box_enc_text_confirm, 3, 4, 1, 5)
# SALT INPUT LABEL
self.salt_label = QLabel(self.translations["labels"]["salt_label"])
self.salt_label.setObjectName("large_label")
self.salt_label.setAlignment(Qt.AlignCenter)
self.layout.addWidget(self.salt_label, 4, 2, 1, 2)
# SALT INPUT
self.salt_insert_box = PasswordEdit()
if self.defaults["default_salt"] != "":
self.salt_insert_box.setText(self.defaults["default_salt"])
self.layout.addWidget(self.salt_insert_box, 4, 4, 1, 5)
# ENCRYPT BUTTON
self.encrypt_button = QPushButton(self.translations["buttons"]["final_encrypt"])
self.layout.addWidget(self.encrypt_button, 5, 3, 1, 5)
self.encrypt_button.clicked.connect(self.encrypt_file)
# finish and set layout
self.main = QWidget()
self.main.setLayout(self.layout)
return self.main
