file_tran/ui.py

import tkinter as tk
from tkinter import filedialog, messagebox
import os
import subprocess
from sender import send_file
from receiver import receive_file
from utils import load_key

class FileTransferApp:
    def __init__(self, root):
        self.root = root
        self.root.title("文件传输系统")
        self.root.geometry("800x900")  # 增大窗口尺寸
        self.root.resizable(False, False)  # 禁止调整窗口大小

        # 设置窗口背景色
        self.root.configure(bg="#f5f5f5")

        # 加载密钥
        self.sender_private_key = load_key('sender_private.pem')
        self.receiver_public_key = load_key('receiver_public.pem')
        self.receiver_private_key = load_key('receiver_private.pem')
        self.sender_public_key = load_key('sender_public.pem')

        # 创建UI
        self.create_main_frame()

    def create_main_frame(self):
        # 主界面，选择发送方或接收方
        self.main_frame = tk.Frame(self.root, bg="#f5f5f5")
        self.main_frame.pack(fill="both", expand=True)

        self.selection_var = tk.StringVar(value="sender")  # 默认选择发送方

        # 选择发送方或接收方
        tk.Radiobutton(self.main_frame, text="我是发送方", variable=self.selection_var, value="sender", font=("Arial", 16),
                       bg="#f5f5f5", command=self.switch_mode).pack(pady=20)
        tk.Radiobutton(self.main_frame, text="我是接收方", variable=self.selection_var, value="receiver", font=("Arial", 16),
                       bg="#f5f5f5", command=self.switch_mode).pack(pady=20)

        # 按钮：生成密钥
        tk.Button(self.main_frame, text="生成公钥/私钥对", font=("Arial", 16), bg="#8BC34A", fg="white", relief="raised", bd=4,
                  width=20, height=2, command=self.generate_keys).pack(pady=20)

        # 初始显示发送方界面
        self.sender_frame = None
        self.receiver_frame = None
        self.switch_mode()

    def switch_mode(self):
        """切换模式（发送方或接收方）"""
        # 清空当前界面
        if self.sender_frame:
            self.sender_frame.destroy()
        if self.receiver_frame:
            self.receiver_frame.destroy()

        mode = self.selection_var.get()

        if mode == "sender":
            self.create_sender_frame()
        elif mode == "receiver":
            self.create_receiver_frame()

    def create_sender_frame(self):
        """创建发送方界面"""
        self.sender_frame = tk.LabelFrame(self.main_frame, text="发送文件", padx=30, pady=30, bg="#ffffff", font=("Arial", 14, "bold"))
        self.sender_frame.pack(padx=30, pady=30, fill="both", expand=True)

        self.filename_var = tk.StringVar()

        # 选择文件按钮
        tk.Button(self.sender_frame, text="选择文件", command=self.select_file,
                  font=("Arial", 16), bg="#4CAF50", fg="white", relief="raised", bd=4,
                  width=20, height=3, activebackground="#45a049").pack(padx=10, pady=15)

        # 显示文件路径
        self.filename_entry = tk.Entry(self.sender_frame, textvariable=self.filename_var, width=60, font=("Arial", 14), state='readonly', bd=3)
        self.filename_entry.pack(padx=10, pady=15)

        # 输入接收方 IP 和端口
        tk.Label(self.sender_frame, text="接收方 IP:", font=("Arial", 14), bg="#ffffff").pack(pady=10)
        self.receiver_ip_entry = tk.Entry(self.sender_frame, width=40, font=("Arial", 14), bd=3)
        self.receiver_ip_entry.pack(pady=10)
        self.receiver_ip_entry.insert(0, "192.168.1.1")

        tk.Label(self.sender_frame, text="接收方端口:", font=("Arial", 14), bg="#ffffff").pack(pady=10)
        self.receiver_port_entry = tk.Entry(self.sender_frame, width=40, font=("Arial", 14), bd=3)
        self.receiver_port_entry.pack(pady=10)
        self.receiver_port_entry.insert(0, "12345")

        # 增大按钮大小
        send_button = tk.Button(self.sender_frame, text="发送文件", command=self.send_file,
                                font=("Arial", 16), bg="#007BFF", fg="white", relief="raised", bd=4,
                                width=25, height=3, activebackground="#0056b3")
        send_button.pack(padx=10, pady=30, fill='x')

    def create_receiver_frame(self):
        """创建接收方界面"""
        self.receiver_frame = tk.LabelFrame(self.main_frame, text="接收文件", padx=30, pady=30, bg="#ffffff", font=("Arial", 14, "bold"))
        self.receiver_frame.pack(padx=30, pady=30, fill="both", expand=True)

        # 接收文件按钮
        receive_button = tk.Button(self.receiver_frame, text="接收文件", command=self.receive_file,
                                   font=("Arial", 16), bg="#FF5722", fg="white", relief="raised", bd=4,
                                   width=25, height=3, activebackground="#e64a19")
        receive_button.pack(padx=10, pady=30, fill='x')

    def generate_keys(self):
        """调用 generate_key.py 生成公钥和私钥"""
        try:
            # 使用 subprocess 调用 generate_key.py 生成密钥对
            subprocess.run(["python", "generate_key.py"], check=True)
            messagebox.showinfo("成功", "公钥和私钥已生成！")
        except subprocess.CalledProcessError as e:
            messagebox.showerror("错误", f"密钥生成失败：{e}")

    def select_file(self):
        """选择文件"""
        file_path = filedialog.askopenfilename()
        if file_path:
            self.filename_var.set(file_path)

    def send_file(self):
        """发送文件"""
        filename = self.filename_var.get()
        if not filename or not os.path.exists(filename):
            messagebox.showerror("错误", "请选择有效的文件！")
            return

        receiver_ip = self.receiver_ip_entry.get()
        receiver_port = self.receiver_port_entry.get()

        if not receiver_ip or not receiver_port:
            messagebox.showerror("错误", "请填写接收方的 IP 和端口！")
            return

        try:
            # 调用 sender.py 中的 send_file 函数
            send_file(filename, receiver_ip, int(receiver_port), self.receiver_public_key, self.sender_private_key)
            messagebox.showinfo("成功", "文件发送成功！")
        except Exception as e:
            messagebox.showerror("错误", f"文件发送失败：{e}")

    def receive_file(self):
        """接收文件"""
        try:
            port = 12345
            # 调用 receiver.py 中的 receive_file 函数
            receive_file(port, self.receiver_private_key, self.sender_public_key)
            messagebox.showinfo("成功", "文件接收成功！")
        except Exception as e:
            messagebox.showerror("错误", f"文件接收失败：{e}")

if __name__ == "__main__":
    root = tk.Tk()
    app = FileTransferApp(root)
    root.mainloop()