【Linux】35. python脚本重命名各子目录下的图片

此python脚本用来重命名当前目录各子目录下的图像文件。
命名格式：类别名_当前时间的微秒_随机数(0~100之间)_文件后缀。
运行命令 python3 rename.py

假设/home/user/pictures/目录下有两个文件夹car和person下都装有几百张jpg格式图片，把rename.py也放在/home/user/pictures/目录下，运行 python3 rename.py后，car和person的每张图片分别被重命名为car_xx_xx.jpg和person_xx_xx.jpg。原来是什么格式图片，脚本会根据原来的后缀名，把图片重命名后依然是什么格式的图片。

rename.py的内容如下

import os
import random
import argparse
from datetime import datetime

IMG_SUFFIX = ['.jpg', '.JPG', '.png', '.jpeg', '.bmp']


def get_microsecond():
    t = datetime.utcnow()
    return t.microsecond


def get_random_name(name, suffix):
    return name + '_' + str(get_microsecond()) + '_' + str(random.randint(0, 100)) + suffix


# def parse_args():
#     """Parse input arguments."""
#     parser = argparse.ArgumentParser(description='Rename')
#     parser.add_argument('--name', dest='name', help='name', default='others')
#     parser.add_argument('--input', dest='input',
#                         help='input', default='others')

#     args = parser.parse_args()
#     return args


def main():
    # args = parse_args()
    # input_path = args.input
    wd = os.getcwd()
    random.seed(123456)
    for _,dirs,_ in os.walk(wd):
            for d in dirs:
                print(d)
                file_list = os.listdir(d)
                image_list = [l for l in file_list if os.path.splitext(l)[-1] in IMG_SUFFIX]
    
                for l in image_list:
                        name, suffix = os.path.splitext(l)
                        img_name = get_random_name(d, suffix)
                        print(img_name)
                        os.rename(os.path.join(wd, d, l), os.path.join(wd, d, img_name))


if __name__ == '__main__':
    main()