Creating Skeleton Files in Batches Suitable for Testing

make-skeleton-tzx.py

# -*- coding: utf-8 -*-

"""Create Skeleton sample files for testing."""

from __future__ import print_function

import os
import random
import sys

# Hex and extension belong to Spectrum .TZX data version 1. FMT/1000.
BIN="\x5A\x58\x54\x61\x70\x65\x21\x1A\x01"
EXT="tzx"
NUMBER=10
SIZE=10
OBJECTS="objects"

def bytes(n):
    for i in range(n):
        yield chr(random.randint(0, 255))

def makeloc():
    try:
        os.makedirs(OBJECTS)
    except OSError:
        pass

def main():
    makeloc()
    no=NUMBER
    size=SIZE
    try:
        no=int(sys.argv[1])
    except (IndexError, ValueError):
        pass
    try:
        size=int(sys.argv[2])
    except (IndexError, ValueError):
        pass
    for i in range(no):
        name = os.path.join(OBJECTS, "{:09d}-skeleton.{}".format(i, EXT))
        size_ = (1024 * 1024 * size) - len(BIN)
        with open(name, "wb") as skeleton_file:
            skeleton_file.write(BIN)
            for byte in bytes(4):
                skeleton_file.write(byte)
            skeleton_file.write('0' * size_)

if __name__ == '__main__':
    main()