模式匹配-修改文件名称

import os
import re

def rename_files(directory):
    """
    重命名目录中符合模式('...', '...').txt的文件为第一个参数+.txt格式

    参数:
    directory (str): 要遍历的目录路径
    """
    # 匹配形如('内容1', '内容2').txt的文件名模式
    pattern = re.compile(r"^\('([^']*)',\s*'([^']*)'\)\.txt$")

    for root, _, files in os.walk(directory):
        for filename in files:
            # 检查文件名是否匹配模式
            match = pattern.match(filename)
            if match:
                # 构造新文件名（使用第一个捕获组内容 + .txt）
                new_name = f"{match.group(1)}.txt"
                old_path = os.path.join(root, filename)
                new_path = os.path.join(root, new_name)

                try:
                    os.rename(old_path, new_path)
                    print(f"成功重命名: {filename} -> {new_name}")
                except FileExistsError:
                    print(f"跳过重命名: {new_name} 已存在")
                except Exception as e:
                    print(f"处理 {filename} 时出错: {str(e)}")


if __name__ == "__main__":
    import sys

    target_dir = "/data/Competitions/labels/train"
    if not os.path.isdir(target_dir):
        print(f"错误: {target_dir} 不是有效目录")
        sys.exit(1)

    rename_files(target_dir)