cipher/window.py at master · TAnd-dev/cipher · 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
import tkinter as tk
from tkinter import filedialog

from cipher import encrypt, decrypt


# function for selecting file
def open_file():
    file_path = filedialog.askopenfilename()
    label_path['text'] = file_path if file_path else 'Выберите файл'
    label_path['fg'] = 'black'


# function for saving file
def save_file():
    # Check if a file is selected
    if label_path.cget('text') == 'Выберите файл':
        label_path['fg'] = 'red'
    # Check if a password isn't set
    elif not input_pass.get():
        label_pass['fg'] = 'red'

    else:
        # Check method (decrypt or encrypt)
        if method.get():
            result, status = encrypt(label_path.cget('text'), input_pass.get(), del_file.get())
        else:
            ext = input_ext.get() if input_ext.get() else '.txt'
            result, status = decrypt(label_path.cget('text'), input_pass.get(), ext, del_file.get())

        show_label_info(result, status)


# show fields for extension
def show_input_ext():
    label_ext.place(x=32, y=150)
    input_ext.grid(row=6, column=1, padx=10, pady=27)


# hide fields for extension
def hide_input_ext():
    label_ext.place_forget()
    input_ext.grid_forget()


# show result
def show_label_info(text, status):
    label_info.place(x=200, y=2)
    label_info['text'] = text
    label_info['fg'] = 'green' if status else 'red'


# Initialization of program
window = tk.Tk()

# Settings of window
window.title('Cipher')
window.geometry('550x250')
window.columnconfigure([0, 1, 2], minsize=50, weight=1)

# The variable in which the method is written
method = tk.BooleanVar()
method.set(True)

# The variable in which the parameter of deletion is written
del_file = tk.BooleanVar()
del_file.set(False)

# Buttons and labels
open_btn = tk.Button(text='Выбрать файл', width=20, command=open_file)
open_btn.grid(row=0, column=0, padx=30, pady=15)

btn_delete = tk.Checkbutton(text='Удалить исходный файл?', width=25, variable=del_file)
btn_delete.grid(row=0, column=1)

label_path = tk.Label(text='Выберите файл')
label_path.place(x=32, y=45)

label_method = tk.Label(text='Выберите метод')
label_method.grid(row=2, column=0, pady=15)

radio_dec_btn = tk.Radiobutton(text='Зашифровать файл', variable=method, value=True, command=hide_input_ext)
radio_dec_btn.grid(row=2, column=1, pady=15)

radio_enc_btn = tk.Radiobutton(text='Расшифровать файл', variable=method, value=False, command=show_input_ext)
radio_enc_btn.grid(row=2, column=3, pady=15, padx=10)

label_pass = tk.Label(text='Укажите пароль')
label_pass.grid(row=3, column=0, pady=5)

input_pass = tk.Entry()
input_pass.grid(row=3, column=1, padx=10, pady=5)

label_ext = tk.Label(text='Укажите расширение для файла (если не указать, файл сохраниться с расширением .txt)')
input_ext = tk.Entry()
input_ext.insert(0, '.txt')

save_btn = tk.Button(text='Сохранить файл', command=save_file)
save_btn.grid(row=8, column=1, padx=10)

label_info = tk.Label()

window.mainloop()