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()获取相应的参数是一个嵌套列表的字典。之后就可以很方便进行操作
网友评论