Python实现CSV文件编码转换工具(转成UTF-8格式)

python main.py

pip install tkinterdnd2
python main.py

encodings = ['mbcs', 'ansi', 'utf-8', 'utf-8-sig']

self.listbox.drop_target_register(DND_FILES)
self.listbox.dnd_bind('<<Drop>>', self._on_drop)

import csv
import os
import sys
import tkinter as tk
from tkinter import filedialog, messagebox
from tkinter import ttk

# 尝试启用拖拽支持（需要安装: pip install tkinterdnd2）
try:
    from tkinterdnd2 import DND_FILES, TkinterDnD  # type: ignore
    DND_AVAILABLE = True
except Exception:
    DND_AVAILABLE = False
    TkinterDnD = None  # 占位，未启用时不使用


def is_csv_file(path: str) -> bool:
    return os.path.isfile(path) and path.lower().endswith('.csv')


def collect_csv_from_dir(folder: str):
    files = []
    if os.path.isdir(folder):
        for f in os.listdir(folder):
            full = os.path.join(folder, f)
            if is_csv_file(full):
                files.append(full)
    return files


def parse_dnd_event_data(root: tk.Tk, data: str):
    # 解析拖拽数据，兼容包含空格的路径
    try:
        parts = root.tk.splitlist(data)
    except Exception:
        parts = data.split()
    return [p.strip('{').strip('}') for p in parts]


class App:
    def __init__(self, root: tk.Tk):
        self.root = root
        self.root.title("CSV 文件转换UTF-8格式 工具（支持拖拽/选择文件/选择文件夹）")
        self.root.geometry('760x560')
        self.root.minsize(680, 480)

        self.files = []  # 去重使用
        self.files_set = set()

        self.output_folder_var = tk.StringVar()

        self._build_ui()

    def _build_ui(self):
        # ========== 输入区域（列表 + 拖拽） ==========
        input_frame = ttk.LabelFrame(self.root, text="待转换文件（将 CSV 文件或文件夹拖拽到此区域）")
        input_frame.pack(fill=tk.BOTH, expand=True, padx=12, pady=(12, 6))

        list_frame = ttk.Frame(input_frame)
        list_frame.pack(fill=tk.BOTH, expand=True, padx=8, pady=8)

        self.listbox = tk.Listbox(list_frame, selectmode=tk.EXTENDED)
        self.listbox.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)

        scrollbar = ttk.Scrollbar(list_frame, orient=tk.VERTICAL, command=self.listbox.yview)
        scrollbar.pack(side=tk.RIGHT, fill=tk.Y)
        self.listbox.config(yscrollcommand=scrollbar.set)

        hint_text = "提示：可点击下方按钮添加文件/文件夹，也可直接拖拽 CSV 文件或文件夹到列表中。"
        self.hint_label = ttk.Label(input_frame, text=hint_text, foreground="#666")
        self.hint_label.pack(anchor='w', padx=10, pady=(0, 8))

        # 拖拽支持
        if DND_AVAILABLE and isinstance(self.root, TkinterDnD.Tk):
            try:
                self.listbox.drop_target_register(DND_FILES)
                self.listbox.dnd_bind('<<Drop>>', self._on_drop)
            except Exception:
                pass
        else:
            if not DND_AVAILABLE:
                warn = "（未安装 tkinterdnd2，拖拽功能不可用。可执行: pip install tkinterdnd2）"
                self.hint_label.configure(text=f"{hint_text}\n{warn}")

        # ========== 操作按钮 ==========
        btns = ttk.Frame(self.root)
        btns.pack(fill=tk.X, padx=12, pady=(0, 6))

        ttk.Button(btns, text="添加文件…", command=self.add_files_dialog).pack(side=tk.LEFT, padx=(0, 6))
        ttk.Button(btns, text="添加文件夹…", command=self.add_folder_dialog).pack(side=tk.LEFT, padx=(0, 6))
        ttk.Button(btns, text="移除选中", command=self.remove_selected).pack(side=tk.LEFT, padx=(0, 6))
        ttk.Button(btns, text="清空列表", command=self.clear_list).pack(side=tk.LEFT)

        # ========== 输出设置 ==========
        out_frame = ttk.LabelFrame(self.root, text="输出设置")
        out_frame.pack(fill=tk.X, padx=12, pady=6)

        row = ttk.Frame(out_frame)
        row.pack(fill=tk.X, padx=10, pady=8)

        ttk.Label(row, text="输出文件夹：").pack(side=tk.LEFT)
        self.out_entry = ttk.Entry(row, textvariable=self.output_folder_var)
        self.out_entry.pack(side=tk.LEFT, fill=tk.X, expand=True, padx=6)
        ttk.Button(row, text="浏览…", command=self.select_output_folder).pack(side=tk.LEFT)

        # ========== 转换 & 进度 ==========
        bottom = ttk.Frame(self.root)
        bottom.pack(fill=tk.X, padx=12, pady=10)

        self.progress = ttk.Progressbar(bottom, mode='determinate')
        self.progress.pack(fill=tk.X, expand=True, side=tk.LEFT, padx=(0, 10))

        ttk.Button(bottom, text="开始转换", command=self.convert_files).pack(side=tk.RIGHT)

        # 状态栏
        self.status_var = tk.StringVar(value="准备就绪")
        status = ttk.Label(self.root, textvariable=self.status_var, relief=tk.SUNKEN, anchor='w')
        status.pack(fill=tk.X, side=tk.BOTTOM)

    # ---------- 文件源管理 ----------
    def _add_paths(self, paths):
        added = 0
        for p in paths:
            if os.path.isdir(p):
                for f in collect_csv_from_dir(p):
                    if f not in self.files_set:
                        self.files.append(f)
                        self.files_set.add(f)
                        self.listbox.insert(tk.END, f)
                        added += 1
            else:
                if is_csv_file(p) and p not in self.files_set:
                    self.files.append(p)
                    self.files_set.add(p)
                    self.listbox.insert(tk.END, p)
                    added += 1
        self.status_var.set(f"已添加 {added} 个文件，当前总数：{len(self.files)}")

    def _on_drop(self, event):
        paths = parse_dnd_event_data(self.root, event.data)
        self._add_paths(paths)

    def add_files_dialog(self):
        paths = filedialog.askopenfilenames(title="选择 CSV 文件", filetypes=[("CSV 文件", "*.csv"), ("所有文件", "*.*")])
        if paths:
            self._add_paths(paths)

    def add_folder_dialog(self):
        folder = filedialog.askdirectory(title="选择包含 CSV 的文件夹")
        if folder:
            self._add_paths([folder])

    def remove_selected(self):
        indices = list(self.listbox.curselection())
        if not indices:
            return
        # 从后向前删除，避免索引错乱
        for idx in reversed(indices):
            path = self.listbox.get(idx)
            self.listbox.delete(idx)
            if path in self.files_set:
                self.files_set.remove(path)
            if path in self.files:
                self.files.remove(path)
        self.status_var.set(f"已移除，当前总数：{len(self.files)}")

    def clear_list(self):
        self.listbox.delete(0, tk.END)
        self.files.clear()
        self.files_set.clear()
        self.status_var.set("列表已清空")

    def select_output_folder(self):
        folder = filedialog.askdirectory(title="请选择保存转换后的文件夹")
        if folder:
            self.output_folder_var.set(folder)

    # ---------- 转换逻辑 ----------
    def _read_rows_with_encodings(self, file_path):
        # 按顺序尝试不同编码读取
        encodings = ['mbcs', 'ansi', 'utf-8', 'utf-8-sig']
        last_err = None
        for enc in encodings:
            try:
                with open(file_path, newline='', encoding=enc) as csvfile:
                    reader = csv.reader(csvfile, delimiter=',', quotechar='"')
                    for row in reader:
                        yield row
                return
            except Exception as e:
                last_err = e
                continue
        # 如果都失败，再抛出最后的异常
        raise last_err if last_err else UnicodeDecodeError("codec", b"", 0, 1, "无法解码")

    def convert_files(self):
        if not self.files:
            messagebox.showwarning("提示", "请先添加需要转换的 CSV 文件或文件夹！")
            return

        output_folder = self.output_folder_var.get().strip()
        if not output_folder:
            messagebox.showerror("错误", "请选择保存转换后的文件夹！")
            return
        if not os.path.exists(output_folder):
            try:
                os.makedirs(output_folder, exist_ok=True)
            except Exception as e:
                messagebox.showerror("错误", f"无法创建输出文件夹：{e}")
                return

        total = len(self.files)
        self.progress.config(maximum=total, value=0)
        success, failed = 0, 0

        for i, src in enumerate(self.files, start=1):
            dst = os.path.join(output_folder, os.path.basename(src))
            try:
                with open(dst, 'w', newline='', encoding='utf-8') as f_w:
                    writer = csv.writer(f_w)
                    for row in self._read_rows_with_encodings(src):
                        writer.writerow(row)
                success += 1
                self.status_var.set(f"转换成功：{os.path.basename(src)}")
            except Exception as e:
                failed += 1
                self.status_var.set(f"转换失败：{os.path.basename(src)} -> {e}")
            finally:
                self.progress['value'] = i
                self.root.update_idletasks()

        msg = f"转换完成！成功：{success}，失败：{failed}。输出目录：{output_folder}"
        if failed:
            messagebox.showwarning("完成（部分失败）", msg)
        else:
            messagebox.showinfo("完成", msg)


def main():
    # 优先使用带 DnD 的 Tk
    if DND_AVAILABLE:
        root = TkinterDnD.Tk()
    else:
        root = tk.Tk()
    # Windows 上 ttk 默认样式更美观
    try:
        ttk.Style().theme_use('clam')
    except Exception:
        pass

    app = App(root)
    root.mainloop()


if __name__ == '__main__':
    main()