Python更改指定目录文件扩展名实例

Python更改指定目录文件扩展名实例，代码简洁适合初学者

import os
import argparse

def batch_rename(work_dir, old_ext, new_ext):

    for filename in os.listdir(work_dir):
        split_file = os.path.splitext(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)
            )

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.parse_args())

    work_dir = args['work_dir'][0]

    old_ext = args['old_ext'][0]
    if old_ext[0] != '.':
        old_ext = '.' + old_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()

这里用到了argparse，其作用就是在命令行提示用户输入相应的参数，通过get_parse().parse_args()获取相应的参数是一个嵌套列表的字典。之后就可以很方便进行操作