Produits Entr'ouvert » w.c.s. » Au quotidien

# w.c.s. - web application for online forms

# Copyright (C) 2005-2016  Entr'ouvert

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, see <http://www.gnu.org/licenses/>.

import time

import random

from wcs.formdef import FormDef

import sys

def get_form_fields(formdef):

    data = {}

    for field in formdef.fields:

        if hasattr(field, 'items'):

            # pick a random value from choices

            data[field.id] = random.choice(field.items)

        else:

            value = field.varname or field.label

            data[field.id] = '%s %s' % (value, random.randint(1, 10))

    return data

def generate_formdata(form_slug, number=10):

    try:

        formdef = FormDef.get_by_urlname(form_slug)

    except KeyError:

        print >> sys.stderr, 'formdef does not exist'

        sys.exit(1)

    for i in xrange(int(number)):

        formdata = formdef.data_class()()

        formdata.data = get_form_fields(formdef)

        formdata.status = 'wf-new'

        formdata.receipt_time = time.localtime(time.time()-random.randint(0, 3600)*i)

        formdata.store()

if len(sys.argv) < 2:

    print >> sys.stderr, 'form slug must be provided in arguments'

    sys.exit(1)

generate_formdata(*sys.argv[1:])