Produits Entr'ouvert » w.c.s.

def test_data_sources_export(pub):

    create_superuser(pub)

    create_role()

    NamedDataSource.wipe()

    data_source = NamedDataSource(name='foobar')

    data_source.data_source = {'type': 'formula', 'value': '[]'}

    data_source.store()

    app = login(get_app(pub))

    resp = app.get('/backoffice/settings/data-sources/1/')

    resp = resp.click(href='export')

    xml_export = resp.text

    ds = StringIO(xml_export)

    data_source2 = NamedDataSource.import_from_xml(ds)

    assert data_source2.name == 'foobar'

def test_data_sources_import(pub):

    create_superuser(pub)

    create_role()

    NamedDataSource.wipe()

    data_source = NamedDataSource(name='foobar')

    data_source.data_source = {'type': 'formula', 'value': '[]'}

    data_source.store()

    data_source_xml = ET.tostring(data_source.export_to_xml(include_id=True))

    NamedDataSource.wipe()

    assert NamedDataSource.count() == 0

    app = login(get_app(pub))

    resp = app.get('/backoffice/settings/data-sources/')

    resp = resp.click(href='import')

    resp.forms[0]['file'] = Upload('datasource.wcs', data_source_xml)

    resp = resp.forms[0].submit()

    assert NamedDataSource.count() == 1

    # import the same datasource a second time, make sure slug is not reused

    resp = app.get('/backoffice/settings/data-sources/')

    resp = resp.click(href='import')

    resp.forms[0]['file'] = Upload('datasource.wcs', data_source_xml)

    resp = resp.forms[0].submit()

    assert NamedDataSource.count() == 2

    assert NamedDataSource.get(1).slug == 'foobar'

    assert NamedDataSource.get(2).slug == 'foobar-1'

    # import an invalid file

    resp = app.get('/backoffice/settings/data-sources/')

    resp = resp.click(href='import')

    resp.form['file'] = Upload('datasource.wcs', b'garbage')

    resp = resp.form.submit()

    assert 'Invalid File' in resp.text

    NamedDataSource.wipe()

import xml.etree.ElementTree as ET

from wcs.qommon import _, force_str

from wcs.qommon.form import FileWidget

from wcs.qommon.form import Form

from wcs.qommon.form import UrlWidget

from wcs.qommon.form import get_response

from wcs.qommon.form import get_session

from wcs.qommon import misc

    _q_exports = ['', 'edit', 'delete', 'export']

    def export(self):

        x = self.datasource.export_to_xml(include_id=True)

        misc.indent_xml(x)

        response = get_response()

        response.set_content_type('application/x-wcs-datasource')

        response.set_header(

            'content-disposition',

            'attachment; filename=datasource-%s.wcs' % self.datasource.slug)

        return '<?xml version="1.0"?>\n' + force_str(ET.tostring(x))

    _q_exports = ['', 'new', ('import', 'p_import')]

        r += htmltext('<a href="import" rel="popup">%s</a>') % _('Import')

    def p_import(self):

        form = Form(enctype='multipart/form-data')

        import_title = _('Import Data Source')

        form.add(FileWidget, 'file', title=_('File'), required=True)

        form.add_submit('submit',  import_title)

        form.add_submit('cancel', _('Cancel'))

        if form.get_submit() == 'cancel':

            return redirect('.')

        if form.is_submitted() and not form.has_errors():

            try:

                return self.import_submit(form)

            except ValueError:

                pass

        get_response().breadcrumb.append(('import', _('Import')))

        html_top('datasources', title=import_title)

        r = TemplateIO(html=True)

        r += htmltext('<h2>%s</h2>') % import_title

        r += htmltext('<p>%s</p>') % _(

            'You can install a new data source by uploading a file '

            'or by pointing to the data source URL.')

        r += form.render()

        return r.getvalue()

    def import_submit(self, form):

        self.imported_datasource = None

        fp = form.get_widget('file').parse().fp

        error = False

        try:

            datasource = NamedDataSource.import_from_xml(fp)

            get_session().message = (

                'info', _('This datasource has been successfully imported.'))

        except ValueError:

            error = True

        if error:

            form.set_error('file', _('Invalid File'))

            raise ValueError()

        self.imported_datasource = datasource

        datasource.slug = None  # a new one will be set in .store()

        datasource.store()

        return redirect('%s/' % datasource.id)

    _indexes = ['slug']

  <a href="export">{% trans "Export" %}</a>