Skip to content

Instantly share code, notes, and snippets.

@caffeinism
Last active October 26, 2022 06:17
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save caffeinism/dbfd974374d620911a6c0c3dd1daadfb to your computer and use it in GitHub Desktop.
Save caffeinism/dbfd974374d620911a6c0c3dd1daadfb to your computer and use it in GitHub Desktop.
test write files
import argparse
import pathlib
import os
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='create directories and write files')
parser.add_argument('target', type=pathlib.Path)
parser.add_argument('--num-directories', type=int, default=1000)
parser.add_argument('--num-total-files', type=int, default=1281167)
parser.add_argument('--file-size', type=int, default=100)
args = parser.parse_args()
os.makedirs(args.target, exist_ok=True)
for i in range(args.num_directories):
directory_path = args.target / f'{i:04d}'
os.mkdir(directory_path)
num_files = args.num_total_files // args.num_directories + int(i < args.num_total_files % args.num_directories)
for j in range(num_files):
file_path = directory_path / f'{j:08d}'
with open(file_path, 'wb') as f:
f.write(b'0123456789abcdef' * 64 * args.file_size)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment