import os
import random
import string
from concurrent.futures import ThreadPoolExecutor
def generate_random_filename(length=10):
    letters = string.ascii_lowercase
    return ''.join(random.choice(letters) for i in range(length))
def write_random_to_file(file_path, size):
    with open(file_path, 'wb') as file:
        file.write(os.urandom(size))
def create_files(file_count, file_size):
    os.makedirs('test', exist_ok=True)
    params = [
        ('test/' + generate_random_filename(20) + '.txt', file_size)
        for _ in range(file_count)
    ]
    with ThreadPoolExecutor(max_workers=None) as executor:
        list(executor.map(lambda param: write_random_to_file(*param), params))
if __name__ == "__main__":
    # 生成文件数量
    file_count = 2000
    # 单个文件大小，单位字节 1024=1kb
    file_size = 1024 * 1024
    create_files(file_count, file_size)