gen_dirs_and_files.py

import random
import string
import os
import os.path
import binascii

MAX_DEPTH=6
MIN_DIRS=1
MAX_DIRS=5
MIN_FILES=1
MAX_FILES=10
MIN_NAME_LEN=3
MAX_NAME_LEN=15
MIN_FILE_SIZE=1024
MAX_FILE_SIZE=100*1024
CREATE_INTERMEDIATE_FILES=True

def gen_dir(current, depth):
    if depth < MAX_DEPTH:
        ndirs = random.randint(MIN_DIRS, MAX_DIRS)
        for i in range(ndirs):
            dirname = ''.join([random.choice(string.letters) for j in range(random.randint(MIN_NAME_LEN, MAX_NAME_LEN))])
            dirpath = os.path.join(current, dirname)
            print 'Creating %s/' % dirpath
            os.mkdir(dirpath)
            gen_dir(dirpath, depth + 1)
    
    if CREATE_INTERMEDIATE_FILES or depth >= MAX_DEPTH:
        nfiles = random.randint(MIN_FILES, MAX_FILES)
        for i in range(nfiles):
            filename = ''.join([random.choice(string.letters) for j in range(random.randint(MIN_NAME_LEN, MAX_NAME_LEN))])
            filepath = os.path.join(current, filename)
            print "Creating %s" % filepath
            with file(filepath, 'w') as f:
                for _ in range(random.randint(MIN_FILE_SIZE / 256, MAX_FILE_SIZE / 256)):
                    lnum = random.getrandbits(256 * 8)
                    s = hex(lnum)[2:]
                    s = s.rstrip('L')
                    if len(s) & 1:
                        s = '0' + s
                    s = binascii.unhexlify(s)
                    f.write(s)



def main():
    gen_dir('.', 0)

if __name__ == '__main__':
    main()