import os
import glob

# get all target csv files
file_list = glob.glob("./datasets/all/chempile_code_complete_*.csv")

for file_path in file_list:
    # extract filename and number part
    dir_name, file_name = os.path.split(file_path)
    prefix, num_suffix = file_name.split("_")[-2], file_name.split("_")[-1].split(".")[0]
    # convert number to 3-digit format
    new_num = num_suffix.zfill(3)
    # construct new file path
    new_file_name = file_name.replace(f"{prefix}_{num_suffix}", f"{prefix}_{new_num}")
    new_file_path = os.path.join(dir_name, new_file_name)
    # rename file
    os.rename(file_path, new_file_path)