import os
from Config import Config

config = Config()

def get_img_lits(img_dir):
    imgs_List = [os.path.join(img_dir, name) for name in sorted(os.listdir(img_dir)[:10]) if
                 name.endswith(('.png', '.jpg', '.webp', '.tif', '.jpeg', '.JPG', '.PNG'))]
    return imgs_List


def preview_img_dir(train_dir_name):
    print('预览: ', train_dir_name)

    train_dir = os.path.join(config.data_dir, train_dir_name, 'train')
    img_dir = os.path.join(train_dir, os.listdir(train_dir)[0])

    img_paths_list = get_img_lits(img_dir)  # 注意传入自定义的web
    # 结果为 list,里面对象可以为
    dict_path = []
    for i in range(len(img_paths_list)):
        dict_path.append((img_paths_list[i], 'img_descrip' + str(i)))  # 图片路径，图片描述, 图片描述可以自定义字符串
    print(dict_path)
    return dict_path