Produits Entr'ouvert » w.c.s.

import collections

def test_delete_formdata(pub):

    from wcs.ctl.delete_formdata import DeleteFormdata

    formdef = FormDef()

    formdef.name = 'example'

    formdef.fields = [fields.StringField(id='0', label='Your Name'),

                      fields.EmailField(id='1', label='Email')]

    formdef.store()

    formdef.data_class().wipe()

    formdata = formdef.data_class()()

    formdata.data = {'0': 'John Doe', '1': 'john@example.net'}

    formdata.store()

    formdata2 = formdef.data_class()()

    formdata2.data = {'0': 'Foo Bar', '1': 'foot@bar.net'}

    formdata2.store()

    assert formdef.data_class().count() == 2

    delete_cmd = DeleteFormdata()

    base_options = {}

    sub_options_class = collections.namedtuple('Options', ['hostname'])

    sub_options = sub_options_class(False)

    delete_cmd.delete_formdata(pub, 'example.net')

    assert formdef.data_class().count() == 0

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

# Copyright (C) 2005-2010  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 os

from qommon.ctl import Command, make_option

class DeleteFormdata(Command):

    name = 'delete_formdata'

    def __init__(self):

        Command.__init__(self, [

                make_option('--hostname', metavar='HOSTNAME', action='store',

                            dest='hostname'),

                ])

    def execute(self, base_options, sub_options, args):

        import publisher

        publisher.WcsPublisher.configure(self.config)

        publisher = publisher.WcsPublisher.create_publisher(

                register_cron=False, register_tld_names=False)

        publisher.app_dir = os.path.join(publisher.app_dir, )

        publisher.set_config()

        self.delete_formdata(publisher, sub_options.hostname)

    def delete_formdata(self, pub, hostname):

        from wcs.formdef import FormDef

        app_dir = pub.app_dir

        for formdef in FormDef.select():

            formdata = formdef.data_class()

            formdata.wipe()

DeleteFormdata.register()