import os
from shutil import copyfile

src_pth = "pb_input/"
dst_pth = "pb_input_new/"

for file_name in os.listdir(src_pth):
    if not file_name.endswith(".bin") or ".ms_" not in file_name:
        print(file_name)
        continue
    source = os.path.join(src_pth, file_name)
    model_name = file_name[:str.find(file_name, ".ms_")]
    index = file_name[str.find(file_name,".ms_") + 4 : str.find(file_name,".bin")]
    input_name = model_name + ".ms.bin"
    if index != "1":
        input_name = input_name + "_" + index
    target = os.path.join(dst_pth, input_name)
    try:
        copyfile(source, target)
        print("copy {} done.".format(file_name))
    except:
        print("copy {} failed.".format(file_name))