重新指定目录下所有文件名后缀

python_demo_1

重新指定目录下所有文件名后缀

import argparse
import os

def batch_rename(work_dir, old_text, new_text):
    # 传递了当前和新扩展名，他将批量重命名给定目录中的一组文件
    #files = os.listdir(work_dir)
    for filename in os.listdir(work_dir):
        split_file = os.path.splittext(filename)
        file_ext = split_file[1]
        # 开始检查文件扩展名的逻辑
        if old_ext == file_ext: 
        # 返回具有新扩展名的已更改文件名
            newfile =  split_file[0] + new_ext
            # 写文件
            os.rename(
                os.path.join(work_dir, filename),
                os.path.join(work_dir, newfile)
            )
    print('rename is done')
    print('os.listdir(work_dir)')

def get_parser():
    parser = argparse.ArgumentParser(description='change extension of files in a working directory')
    parser.add_argument('work_dir', metavar='WORK_DIR', type=str, nargs=1, help='the directory where to change extension')
    parser.add_argument('old_ext', metavar='OLD_EXT', type=str, nargs=1, help='old extension')
    parser.add_argument('new_ext', metavar='NEW_EXT', type=str, nargs=1, help='new extension')
    return parser

def main():
    parser = get_parser()  # 添加命令行参数
    args = vars(parser.parser_args())
    # 使用传递的第一个参数设置变量work_dir
    work_dir = args['work_dir'][0]
     # 使用传递的第二个参数设置变量old_ext
    old_ext = args['old_ext'][0]
    if old_ext[0] != '.':
        old_ext = '.' + old_ext
     # 使用传递的第三个参数设置变量new_ext
    new_ext = args['new_ext'][0]
    if new_ext[0] != '.':
        new_ext = '.' + new_ext
    
    batch_rename(work_dir, old_ext, new_ext)

if __name__ == '__main__':
    main()