KRKR引擎汉化使用技巧之免转区的思路

import os
import codecs
import argparse

# 定义要转换的文件扩展名
extensions = ['.tjs', '.ks']

# 源编码和目标编码
source_encoding = 'shift-jis'
target_encoding = 'utf-16-le'

# 解析命令行参数
parser = argparse.ArgumentParser(description='批量转换文件编码从Shift-JIS到带BOM的UTF-16LE')
parser.add_argument('directory', nargs='?', default=os.getcwd(), help='要转换的文件夹路径（默认：当前工作目录）')
args = parser.parse_args()

def convert_file_encoding(file_path):
    """
    将单个文件从源编码转换为目标编码（不创建备份）
    """
    try:
        # 以源编码读取文件内容
        with codecs.open(file_path, 'r', encoding=source_encoding) as f:
            content = f.read()
        
        # 以目标编码写入文件（带BOM）
        # 对于UTF-16-LE，BOM是\xFF\xFE
        bom = codecs.BOM_UTF16_LE
        with open(file_path, 'wb') as f:
            f.write(bom)
            f.write(content.encode(target_encoding))
        
        print(f"成功转换: {file_path}")
        return True
    except Exception as e:
        print(f"转换失败: {file_path}")
        print(f"  错误信息: {str(e)}")
        return False

def main():
    # 获取要转换的根目录
    root_dir = args.directory
    print(f"开始转换文件编码，根目录: {root_dir}")
    print(f"源编码: {source_encoding}")
    print(f"目标编码: {target_encoding} (带BOM)")
    print(f"文件扩展名: {extensions}")
    print("=" * 50)
    
    success_count = 0
    fail_count = 0
    
    for dirpath, dirnames, filenames in os.walk(root_dir):
        for filename in filenames:
            # 检查文件扩展名
            if any(filename.lower().endswith(ext) for ext in extensions):
                file_path = os.path.join(dirpath, filename)
                if convert_file_encoding(file_path):
                    success_count += 1
                else:
                    fail_count += 1
    
    print("=" * 50)
    print(f"转换完成!")
    print(f"成功转换: {success_count} 个文件")
    print(f"转换失败: {fail_count} 个文件")

if __name__ == "__main__":
    main()