import os
import shutil

from loguru import logger


def get_filename() -> str:
    current_file_name = os.path.basename(__file__)
    log_name = "{}.log"
    return log_name.format(current_file_name.split('.')[0])


def set_log() -> None:
    filename = get_filename()
    logger.add(f'../log/{filename}')


set_log()


# 输入文件夹路径和输出文件夹路径
input_folder = '/mnt/nfs/data/geneformer/output/v2/looms'
output_folder = '/mnt/nfs/data/geneformer/output/v2/looms_sub'
per_group_num = 100


if __name__ == '__main__':
    # 创建输出文件夹（如果不存在）
    os.makedirs(output_folder, exist_ok=True)

    # 遍历输入文件夹中的文件
    files = os.listdir(input_folder)
    group_count = 0

    for i, file in enumerate(files, start=1):
        # 构造输出文件夹路径
        group_folder = os.path.join(output_folder, f'group_{group_count + 1}')

        # 创建输出文件夹（如果不存在）
        os.makedirs(group_folder, exist_ok=True)

        # 构造输入文件路径和输出文件路径
        input_file = os.path.join(input_folder, file)
        output_file = os.path.join(group_folder, file)

        # 移动文件到输出文件夹
        shutil.copy(input_file, output_file)

        if i % per_group_num == 0:
            group_count += 1