gen-json-blob.py

#!/usr/bin/env python3.7

import argparse
import json
import random
import string


def get_params():
    parser = argparse.ArgumentParser()

    parser.add_argument(
        'num_keys',
        type=int,
        help='number of keys in the dict'
    )
    parser.add_argument(
        '--key_size',
        type=int,
        default=10
    )
    parser.add_argument(
        '--value_size',
        type=int,
        default=10
    )
    parser.add_argument(
        '--json_indent',
        type=int,
        default=4
    )

    return parser.parse_args()


def randstr(size):
    return ''.join(random.choice(string.ascii_lowercase) for _ in range(size))


def gen_blob(params):
    d = {}

    for _ in range(params.num_keys):
        key = randstr(params.key_size)
        val = randstr(params.value_size)
        d[key] = val

    return json.dumps(d, indent=params.json_indent)


if __name__ == '__main__':
    print(gen_blob(get_params()))