Tkinter Button命令与Entry二进制数据处理：常见陷阱与最佳实践

def my_function():
    print("Button clicked!")

# 错误做法：函数会在程序启动时立即执行，而不是在按钮点击时
my_button = tk.Button(root, text="Click Me", command=my_function())

# 原始代码中的错误
save_button = tk.Button(root, command=save_key_to_file(), text="Save Key", ...)

def save_key_to_file():
    # ... 函数体 ...
    print("保存密钥功能被调用")

save_button = tk.Button(root, command=save_key_to_file, text="Save Key", ...)

def save_key_to_file(file_name):
    # ... 函数体，可以使用 file_name 参数 ...
    print(f"保存密钥到文件: {file_name}")

# 使用 lambda 表达式传递参数
save_button = tk.Button(root, command=lambda: save_key_to_file("custom_file.key"), text="Save Key", ...)

import tkinter as tk
from tkinter import filedialog
from cryptography.fernet import Fernet
import os # 用于文件操作，虽然在这个例子中没有直接使用os.path，但通常会用到

# 初始化主窗口
root = tk.Tk()
root.title("密钥管理工具")
root.geometry("500x250")
root.resizable(False, False)
root.config(bg="#333333") # 设置背景色

# 密钥输入框
key_entry = tk.Entry(root, bg="grey", fg="green", width=50, font=("Arial", 12))
key_entry.place(x=35, y=100)

def generate_key():
    """生成Fernet密钥并显示在Entry框中。"""
    key = Fernet.generate_key()
    key_entry.delete(0, tk.END) # 清空Entry内容
    key_entry.insert(0, key.decode('utf-8')) # 将二进制密钥解码为字符串显示

def save_key_to_file():
    """从Entry获取密钥文本，编码为二进制并保存到文件。"""
    key_text = key_entry.get()
    if not key_text:
        print("Entry框中没有密钥可保存。")
        return

    # 将字符串密钥编码为二进制数据
    # Fernet密钥本身是base64编码的二进制数据，从Entry获取后需要再次编码回原始字节
    # 如果Entry中是直接粘贴的base64字符串，它本身就是可直接用于Fernet的字节表示
    # 这里假设Entry中可能包含用户输入的文本，因此需要编码。
    # 对于Fernet生成的密钥，它已经是bytes类型，decode('utf-8')后是字符串，
    # 再encode('utf-8')会得到原始bytes。
    try:
        key_bytes = key_text.encode('utf-8')
        # 实际Fernet密钥是base64编码的bytes，可以直接使用。
        # 如果从Entry获取的是用户可能修改过的文本，确保它是有效的base64编码字符串
        # 并且是Fernet兼容的长度。
        # 这里为了简化，直接对Entry内容进行UTF-8编码。
        # 实际应用中，可能需要更严格的校验或直接使用Fernet生成的bytes。
    except UnicodeEncodeError:
        print("密钥包含无法编码的字符。")
        return

    file_path = "file.key" # 默认文件名
    try:
        with open(file_path, "wb") as file: # 以二进制写入模式打开文件
            file.write(key_bytes)
        print(f"密钥已成功保存到 {file_path}")
    except IOError as e:
        print(f"保存文件时发生错误: {e}")

def load_key_from_file():
    """从文件加载二进制密钥并显示在Entry框中。"""
    file_path = filedialog.askopenfilename(
        title="选择密钥文件",
        filetypes=[("Key files", "*.key"), ("All files", "*.*")]
    )
    if not file_path:
        return

    try:
        with open(file_path, "rb") as file: # 以二进制读取模式打开文件
            key_bytes = file.read()
        key_entry.delete(0, tk.END)
        key_entry.insert(0, key_bytes.decode('utf-8')) # 将二进制密钥解码为字符串显示
        print(f"密钥已从 {file_path} 加载。")
    except FileNotFoundError:
        print(f"文件未找到: {file_path}")
    except Exception as e:
        print(f"加载文件时发生错误: {e}")

# 按钮样式配置
button_style = {
    "borderwidth": 0,
    "bg": "black",
    "fg": "green",
    "activebackground": '#2e2e2e',
    "activeforeground": "green",
    "font": ("Arial", 10, "bold")
}

# 生成密钥按钮
generate_button = tk.Button(root, text="生成密钥", command=generate_key, **button_style)
generate_button.place(x=35, y=130)

# 保存密钥按钮 (修正了 command 参数)
save_button = tk.Button(root, text="保存密钥", command=save_key_to_file, **button_style)
save_button.place(x=150, y=130)

# 加载密钥按钮 (修正了 command 参数)
load_button = tk.Button(root, text="加载密钥", command=load_key_from_file, **button_style)
load_button.place(x=265, y=130)

root.mainloop()