summaryrefslogtreecommitdiff
path: root/scalaplugin/src/test/resource/jawn/randjson2.py
blob: eb6b9a3aecfaa4286a3ef3fe1c7de6039d39a996 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import json
import os
from random import *
import string
import sys

constants = [True, False, None]

def mkconstant():
    return choice(constants)

def mkinteger():
    return randint(-1e3, 1e3) * (10 ** normalvariate(0, 4)) + randint(-1e3, 1e3)

def mkdouble():
    return random() * (10 ** normalvariate(0, 30))

def mknum():
    if randint(0, 1):
        return mkdouble()
    else:
        return mkinteger()

def mkstring():
    n = int(min(abs(normalvariate(40, 20)), abs(normalvariate(30, 10))))
    return ''.join([choice(string.ascii_letters) for i in range(0, n)])

values = [mkconstant, mknum, mknum, mknum, mkstring]

def mkvalue():
    return choice(values)()

if __name__ == "__main__":
    args = sys.argv[1:]
    try:
        num = int(args[0])
        path = args[1]
        print "writing json (%d rows) into %s" % (num, path)
        f = open(path, 'w')
        f.write("[")
        for i in range(0, num):
            if i > 0: f.write(", ")
            c = {"foo": mkstring(),
                 "bar": mknum(),
                 "qux": mkvalue(),
                 "duh": {"a": mknum(), "b": mknum(), "c": mknum()},
                 "xyz": {"yy": mkstring(), "zz": mkvalue()},
                 "abc": [mkvalue() for i in range(0, 4)]}
            f.write(json.dumps(c))
        f.write("]")
        f.close()
    except Exception, e:
        print "usage: %s NUM PATH" % sys.argv[0]