test_script/HR_OCR/tools/suffix.py

'''
Author: zeke-chin zeke-chin@icloud.com
Date: 2022-10-11 16:38:18
LastEditors: zeke-chin zeke-chin@icloud.com
LastEditTime: 2022-10-12 17:53:30
FilePath: /test_script/HR_OCR/tools/suffix.py
Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
'''
import os
from pathlib import Path
from itertools import chain
import sys

# conf
target_path = '/Users/zeke/work/sx/OCR/image_data/营业执照90'
#suffix = sys.argv[2]
suffix = 'jpg' if len(sys.argv) != 3 else sys.argv[2]

def get_range(n):
    len_n = len(str(n))
    for num in range(1, n + 1):
        output = str(num)
        while len(output) < len_n:
            output = f'0{output}'
        yield output


file_paths = list(chain(*[Path(target_path).rglob(f'*.{ext}') for ext in ['jpeg', 'jpg', 'png', 'JPG', 'PNG']]))
print(len(file_paths))
num = len(file_paths)
file_name_list = list(get_range(num))
for i in range(num):
    file = file_paths[i]
    print(file)
    new = file.parent / f'{file_name_list[i]}_img.{suffix}'
    file.rename(new)