Python实现Excel表格保存到不同文件夹

"""
点击“上传”按钮，从本地上传待处理的Excel表格
点击“处理”按钮，对Excel表格进行处理
点击“保存A”按钮，保存处理后的Excel表格到A文件夹下
点击“保存B”按钮，保存处理后的Excel表格到B文件夹下
"""

# 作 者:Leo
# 时 间:2024/9/26 21:52
import tkinter as tk
from tkinter import filedialog, messagebox
import pandas as pd
import os
"""
点击“上传”按钮，从本地上传待处理的Excel表格
点击“处理”按钮，对Excel表格进行处理
点击“保存A”按钮，保存处理后的Excel表格到A文件夹下
点击“保存B”按钮，保存处理后的Excel表格到B文件夹下
"""
# 创建主窗口
root = tk.Tk()
root.title("Excel 文件处理器")
root.geometry("400x250")

# 定义全局变量以存储上传和处理的文件路径及处理后的数据
file_path = ""
processed_data = None

# 上传Excel文件的函数
def upload_file():
    global file_path
    file_path = filedialog.askopenfilename(filetypes=[("Excel files", "*.xlsx;*.xls")])
    if file_path:
        messagebox.showinfo("文件上传", f"已上传文件: {file_path}")
    else:
        messagebox.showwarning("文件上传", "没有选择任何文件")

# 处理Excel文件的函数
def process_file():
    global processed_data
    if not file_path:
        messagebox.showwarning("处理文件", "请先上传一个文件")
        return
    try:
        # 读取Excel文件
        df = pd.read_excel(file_path)

        # 数据处理，比如删除空值
        processed_data = df.dropna()

        messagebox.showinfo("处理文件", "文件处理完成")
    except Exception as e:
        messagebox.showerror("处理文件", f"文件处理时发生错误: {e}")

# 保存文件的通用函数
def save_file(folder_name):
    global processed_data
    if processed_data is None:
        messagebox.showwarning("保存文件", "请先处理文件")
        return
    try:
        # 获取原始文件的目录和文件名
        file_dir = os.path.dirname(file_path)
        base_name = os.path.basename(file_path).split('.')[0]  # 不带扩展名的文件名

        # 创建文件夹（如果不存在）
        folder_path = os.path.join(file_dir, folder_name)
        os.makedirs(folder_path, exist_ok=True)

        # 初始化后缀i
        i = 1
        # 构建文件名
        save_path = os.path.join(folder_path, f"{base_name}_processed_{i}.xlsx")

        # 检查文件是否存在，若存在则递增i
        while os.path.exists(save_path):
            i += 1
            save_path = os.path.join(folder_path, f"{base_name}_processed_{i}.xlsx")

        # 保存处理后的数据
        processed_data.to_excel(save_path, index=False)
        messagebox.showinfo("保存文件", f"文件已保存到: {save_path}")
    except Exception as e:
        messagebox.showerror("保存文件", f"文件保存时发生错误: {e}")

# 保存A按钮
def save_a():
    save_file("A")

# 保存B按钮
def save_b():
    save_file("B")

# 创建按钮并绑定功能
upload_button = tk.Button(root, text="上传", command=upload_file)
process_button = tk.Button(root, text="处理", command=process_file)
save_a_button = tk.Button(root, text="保存A", command=save_a)
save_b_button = tk.Button(root, text="保存B", command=save_b)

# 布局按钮
upload_button.pack(pady=5)
process_button.pack(pady=5)
save_a_button.pack(pady=5)
save_b_button.pack(pady=5)

# 进入主事件循环
root.mainloop()