import os
import re
import shutil
import sys
import time


# 进度条
def process_bar(value, speed, total):
    sys.stdout.write(value)
    for i in range(total):
        if i != 10:
            sys.stdout.write("==")
        else:
            sys.stdout.write("== " + str(i * 10) + "%/100%")
        sys.stdout.flush()
        time.sleep(speed)


path = input('请输入文件夹路径（以\结尾）：')
dirlist = ['行程码', '健康码']
imgs = ['.bmp', '.png', '.jpg', '.jpeg', '.pbm', '.pgm', '.ppm', '.tif', '.tiff', '.dib']

# 创建文件夹
process_bar('创建文件夹中... ', 0.01, 11)
for dil in dirlist:
    if not os.path.exists(path + dil):
        os.makedirs(path + dil)
    else:
        print("文件夹创建失败")
print("\n文件夹创建成功")

# 遍历处理
prefix = input('重命名格式：前缀+编号-姓名\n请输入图片命名前缀：')
for file in os.listdir(path):
    if file.endswith('.jpg') or file.endswith('.png'):
        newname = file[re.search(r"(\d{4}-\d{1,2}-\d{1,2}-)", file).span()[1]:]
        if not file.startswith(prefix): newname = prefix + newname
        if "行程码" in file:
            newname = newname.replace('-行程码截图', '')
            os.rename(os.path.join(path, file), os.path.join(path, newname))
            shutil.move(os.path.join(path, newname), os.path.join(path, dirlist[0]))
        elif "健康码" in file:
            newname = newname.replace('-健康码截图', '')
            os.rename(os.path.join(path, file), os.path.join(path, newname))
            shutil.move(os.path.join(path, newname), os.path.join(path, dirlist[1]))
process_bar('自动化分类中... ', 0.1, 11)
print("\n文件重命名移动成功！")
os.system('pause')