def create_new_window1 def encrypt_decrypt_fileinput_file_path output_file_path key # 根据输入文件的后缀选择合适的加密解密函数 if input_file_pathendswithdocx # 如果是docx文件使用Python库docx对文本进行加密解密

def create_new_window1(): def encrypt_file(input_file_path, output_file_path, key): # 根据输入文件的后缀选择合适的加密函数 if input_file_path.endswith('.docx'): # 如果是.docx文件，使用Python库docx对文本进行加密 document = Document(input_file_path) for paragraph in document.paragraphs: text = paragraph.text encrypted_text = encrypt_text(text, key) #调用encrypt_text函数进行加密操作 paragraph.text = encrypted_text document.save(output_file_path)#保存加密后的文件 elif input_file_path.endswith('.xlsx'): #如果是.xlsx文件，使用Python库openpyxl对单元格数据进行加密 workbook = load_workbook(input_file_path) for sheet in workbook.sheetnames: worksheet = workbook[sheet] for row in worksheet.iter_rows(): for cell in row: if cell.value is None: continue if isinstance(cell.value, str): encrypted_value = encrypt_text(cell.value, key)#调用encrypt_text函数进行加密操作 cell.value = encrypted_value workbook.save(output_file_path)#保存加密后的文件 else: #如果是其他类型的文件，则按照转子密码进行加密 #定义三个转子，用于进行加密操作 rotor1 = list(string.ascii_lowercase) rotor2 = list(string.ascii_lowercase) rotor3 = list(string.ascii_lowercase) #根据提供的密钥对转子进行初始化 for i in range(key[0]): rotor1.append(rotor1.pop(0)) for i in range(key[1]): rotor2.append(rotor2.pop(0)) for i in range(key[2]): rotor3.append(rotor3.pop(0)) #存储加密后的结果数据 result_data = [] #打开输入文件进行读取数据 with open(input_file_path, 'r') as file: data = file.read() #遍历输入文件的每个字符 for character in data: if character.isalpha():#如果字符是字母 index = rotor1.index(character.lower())#在第一个转子中查找字母在转子中的位置 index = rotor2[index]#将结果作为索引在第二个转子中查找 index = rotor3[index]#将结果作为索引在第三个转子中查找 result_data.append(rotor1[index])#将最终的结果字母添加到结果数据中 else:#如果字符不是字母，则直接添加到结果数据中 result_data.append(character) #将加密后的结果数据写入输出文件 with open(output_file_path, 'w') as file: file.write(''.join(result_data)) return True

def decrypt_file(input_file_path, output_file_path, key):
    # 根据输入文件的后缀选择合适的解密函数
    if input_file_path.endswith('.docx'):
        # 如果是.docx文件，使用Python库docx对文本进行解密
        document = Document(input_file_path)
        for paragraph in document.paragraphs:
            text = paragraph.text
            decrypted_text = decrypt_text(text, key) #调用decrypt_text函数进行解密操作
            paragraph.text = decrypted_text
        document.save(output_file_path)#保存解密后的文件
    elif input_file_path.endswith('.xlsx'):
        #如果是.xlsx文件，使用Python库openpyxl对单元格数据进行解密
        workbook = load_workbook(input_file_path)
        for sheet in workbook.sheetnames:
            worksheet = workbook[sheet]
            for row in worksheet.iter_rows():
                for cell in row:
                    if cell.value is None:
                        continue
                    if isinstance(cell.value, str):
                        decrypted_value = decrypt_text(cell.value, key)#调用decrypt_text函数进行解密操作
                        cell.value = decrypted_value
        workbook.save(output_file_path)#保存解密后的文件
    else:
        #如果是其他类型的文件，则按照转子密码进行解密
        #定义三个转子，用于进行解密操作
        rotor1 = list(string.ascii_lowercase)
        rotor2 = list(string.ascii_lowercase)
        rotor3 = list(string.ascii_lowercase)
        #根据提供的密钥对转子进行初始化
        for i in range(key[0]):
            rotor1.append(rotor1.pop(0))
        for i in range(key[1]):
            rotor2.append(rotor2.pop(0))
        for i in range(key[2]):
            rotor3.append(rotor3.pop(0))
        #存储解密后的结果数据
        result_data = []
        #打开输入文件进行读取数据
        with open(input_file_path, 'r') as file:
            data = file.read()
            #遍历输入文件的每个字符
            for character in data:
                if character.isalpha():#如果字符是字母
                    index = rotor1.index(character.lower())#在第一个转子中查找字母在转子中的位置
                    index = rotor2.index(index)#将结果作为索引在第二个转子中查找
                    index = rotor3.index(index)#将结果作为索引在第三个转子中查找
                    result_data.append(rotor1[index])#将最终的结果字母添加到结果数据中
                else:#如果字符不是字母，则直接添加到结果数据中
                    result_data.append(character)
        #将解密后的结果数据写入输出文件
        with open(output_file_path, 'w') as file:
            file.write(''.join(result_data))
    return True

#获取输入的文件路径、输出的文件路径和密钥
def on_encrypt():
    input_file_path = e_input.get()
    output_file_path = e_output.get()
    key = e_key.get()
    #检查是否填写完整的输入、输出路径和密钥
    if not input_file_path or not output_file_path or not key:
        messagebox.showwarning("Warning", "请填写完整的输入、输出路径和密钥！")
        return
    #调用encrypt_file函数进行文件加密操作
    if encrypt_file(input_file_path, output_file_path, key):
        messagebox.showinfo("Success", "文件加密成功！")

def on_decrypt():
    input_file_path = e_input.get()
    output_file_path = e_output.get()
    key = e_key.get()
    #检查是否填写完整的输入、输出路径和密钥
    if not input_file_path or not output_file_path or not key:
        messagebox.showwarning("Warning", "请填写完整的输入、输出路径和密钥！")
        return
    #调用decrypt_file函数进行文件解密操作
    if decrypt_file(input_file_path, output_file_path, key):
        messagebox.showinfo("Success", "文件解密成功！")

def on_select_input():
    #弹出选择文件对话框并获取选择的文件路径
    file_path = filedialog.askopenfilename()
    if file_path:
        #将选择的文件路径设置到输入路径文本框中
        e_input.delete(0, tk.END)
        e_input.insert(0, file_path)

def on_select_output():
    #获取输入的文件路径
    input_file_path = e_input.get()
    if not input_file_path:
        return
    #根据输入路径的文件扩展名生成默认的输出文件路径，并弹出保存文件对话框获取输出文件路径
    file_extension = os.path.splitext(input_file_path)[1]
    file_path = filedialog.asksaveasfilename(defaultextension=file_extension)
    if file_path:
        #将选择的文件路径设置到输出路径文本框中
        e_output.delete(0, tk.END)
        e_output.insert(0, file_path)

new_window = tk.Toplevel(root_new)
new_window.title('转轮机密码加密\解密窗口')
new_window.geometry('2000x1000')

lbl_input = tk.Label(new_window, text="输入文件路径:")
lbl_input.pack()
e_input = tk.Entry(new_window, width=50)
e_input.pack()
btn_select_input = tk.Button(new_window, text="选择文件", command=on_select_input)
btn_select_input.pack()

lbl_output = tk.Label(new_window, text="输出文件路径:")
lbl_output.pack()
e_output = tk.Entry(new_window, width=50)
e_output.pack()
btn_select_output = tk.Button(new_window, text="选择路径", command=on_select_output)
btn_select_output.pack()

lbl_key = tk.Label(new_window, text="密钥:")
lbl_key.pack()
e_key = tk.Entry(new_window)
e_key.pack()

btn_encrypt = tk.Button(new_window, text="加密", command=on_encrypt)
btn_encrypt.pack()

btn_decrypt = tk.Button(new_window, text="解密", command=on_decrypt)
btn_decrypt.pack(