Esempio n. 1
0
def load_users(dir, user_count, records, size, audience='example.com'):
    now = int(time.time())
    for i in xrange(user_count):
        user = '******' % i
        if isinstance(records, tuple):
            rec = random.randint(*records)
        else:
            rec = records
        fn = os.path.join(dir, urllib.quote(audience, ''), urllib.quote(user, '') + '.db')
        db = Database(fn)
        items = []
        for count in xrange(rec):
            if isinstance(size, tuple):
                s = random.randint(*size)
            else:
                s = size
            items.append('{"id": "item-%s-%s", "data": "%s"}' % (now, count, 'a'*s))
        db.extend(items)
        sys.stdout.write('.')
        sys.stdout.flush()
    print 'done.'