import pandas as pd
import os
import shutil
import csv

def convert_excel_to_txt(folder_path='.'):
    # 计数器，用于最后汇总
    count = 0
    target_dir = os.path.join(os.path.dirname(__file__), '../Assets/GameMain/DataTables')
    target_dir = os.path.abspath(target_dir)
    
    # 确保目标目录存在
    os.makedirs(target_dir, exist_ok=True)
    
    for file_name in os.listdir(folder_path):
        if file_name.endswith(('.xlsx', '.xls')):
            file_path = os.path.join(folder_path, file_name)
            base_name = os.path.splitext(file_path)[0]
            output_file = base_name.replace(os.path.basename(base_name), os.path.basename(base_name)) + '.txt'
            
            print(f"正在处理: {file_name}...")
            
            try:
                # 读取 Excel
                df = pd.read_excel(file_path, header=None)
                
                # 预处理：将 NaN 替换为空字符串，否则导出会变成 "nan"
                df = df.fillna('')

                # 导出设置：
                # 1. sep='\t' : 使用制表符分隔
                # 2. quoting=csv.QUOTE_NONE : 不使用引号包裹字段，也不会把 " 变成 ""
                # 3. escapechar='\\' : 如果单元格内恰好有 Tab 键，会用反斜杠转义，防止数据列错位
                df.to_csv(
                    output_file, 
                    sep='\t', 
                    index=False, 
                    header=False, 
                    encoding='utf-8', 
                    quoting=csv.QUOTE_NONE, 
                    escapechar='\\' 
                )
                
                # 复制文件到目标目录
                target_file = os.path.join(target_dir, os.path.basename(output_file))
                shutil.copy2(output_file, target_file)
                
                print(f"成功转换 -> {output_file}")
                print(f"已复制到 -> {target_file}")
                count += 1
            
            except Exception as e:
                print(f"处理 {file_name} 时出错: {e}")

    print(f"\n任务完成！共转换了 {count} 个文件。")

if __name__ == "__main__":
    convert_excel_to_txt('.')
    
    # --- 关键修改：在这里添加暂停 ---
    print("\n" + "="*30)
    input("按回车键(Enter)退出程序...")