Produits Entr'ouvert » Hobo

class ImportForm(forms.Form):

    parameters_json = forms.FileField(label=_('Parameters Export File'))

{% extends "hobo/base.html" %}

{% load i18n %}

{% block appbar %}

<h2>{% trans "Parameters Import" %}</h2>

{% endblock %}

{% block breadcrumb %}

{{ block.super }}

<a href="{% url 'environment-import' %}">{% trans 'Parameters Import' %}</a>

{% endblock %}

{% block content %}

<form method="post" enctype="multipart/form-data">

  {% csrf_token %}

  {{ form.as_p }}

  <div class="buttons">

    <button class="submit-button">{% trans "Import" %}</button>

    <a class="cancel" href="{% url 'home' %}">{% trans 'Cancel' %}</a>

  </div>

</form>

{% endblock %}

    url(r'^import/$', views.ImportView.as_view(), name='environment-import'),

    url(r'^export/$', views.ExportView.as_view(), name='environment-export'),

from hobo.profile.utils import get_profile_dict

def export_parameters():

    from .models import Variable

    variables = []

    for var in Variable.objects.filter(service_pk__isnull=True):

        variables.append({

            'name': var.name,

            'label': var.label,

            'value': var.value,

            'auto': var.auto})

    parameters = {'variables': variables}

    parameters.update(get_profile_dict())

    return parameters

def import_parameters(parameters):

    from .models import Variable

    from hobo.profile.models import AttributeDefinition

    objects = []

    try:

        for variables in parameters.get('variables'):

            obj, created = Variable.objects.get_or_create(name=variables['name'])

            for key, value in variables.items():

                if key == 'name':

                    continue

                setattr(obj, key, value)

            objects.append(obj)

        for fields in parameters.get('profile', []).get('fields'):

            obj, created = AttributeDefinition.objects.get_or_create(

                label=fields['label'], name=fields['name'])

            for key, value in fields.items():

                if key == 'label' or key == 'name':

                    continue

                setattr(obj, key, value)

            objects.append(obj)

    except Exception as exc:

        return {'err': True, 'message': '%s, %s' % (exc.__class__.__name__, exc)}

    for obj in objects:

        obj.save()

    return {'err': False}

from django.utils.encoding import force_text

from django.utils.translation import ugettext_lazy as _

from django.views.generic import View

from django.views.generic.edit import CreateView, UpdateView, DeleteView, FormView

class ImportView(FormView):

    form_class = forms.ImportForm

    template_name = 'environment/import.html'

    success_url = reverse_lazy('home')

    def form_valid(self, form):

        try:

            parameters_json = json.loads(

                force_text(self.request.FILES['parameters_json'].read()))

        except ValueError:

            form.add_error('parameters_json', _('File is not in the expected JSON format.'))

            return self.form_invalid(form)

        result = utils.import_parameters(parameters_json)

        if result.get('err'):

            form.add_error('parameters_json',

                           _('Parameters import fails: %s') % force_text(result.get('message')))

            return self.form_invalid(form)

        return super(ImportView, self).form_valid(form)

class ExportView(View):

    def get(self, request, *args, **kwargs):

        response = HttpResponse(content_type='application/json')

        response['Content-Disposition'] = 'attachment; filename="hobo-export.json"'

        json.dump(utils.export_parameters(), response, indent=2)

        return response

    <li><a rel="popup" href="{% url 'environment-import' %}">{% trans 'Import' %}</a></li>

    <li><a href="{% url 'environment-export' %}">{% trans 'Export' %}</a></li>

import json

from webtest import Upload

from hobo.profile.models import AttributeDefinition

def test_export_import_view(app, admin_user):

    combo = Combo.objects.create(base_url='https://combo.agglo.love',

                                 template_name='...portal-user...',

                                 slug='portal')

    Variable.objects.create(name='foo', value='bar').save()

    Variable.objects.create(name='foo2', value='bar2', service=combo).save()

    app = login(app, 'admin', 'password')

    resp = app.get('/sites/export/', status=200)

    assert sorted(resp.json.keys()) == ['profile', 'variables']

    assert resp.json['variables'] == [

        {'name': 'foo', 'label': '', 'value': 'bar', 'auto': False}]

    assert resp.json['profile']['fields'][0]['name'] == 'title'

    assert resp.json['profile']['fields'][0]['required'] is False

    assert resp.json['profile']['fields'][1]['name'] == 'first_name'

    assert resp.json['profile']['fields'][1]['required'] is True

    # modify exported file

    export = resp.json

    export['variables'][0]['label'] = 'bar'

    fields = export['profile']['fields']

    assert fields[0]['name'] == 'title'

    assert fields[1]['name'] == 'first_name'

    fields[0]['description'] = 'genre'

    fields[0], fields[1] = fields[1], fields[0]

    export_json = json.dumps(export)

    # add new content

    Variable.objects.create(name='foo3', value='bar3').save()

    AttributeDefinition.objects.create(name='prefered_color').save()

    assert Variable.objects.count() == 3

    assert AttributeDefinition.objects.count() == 12

    assert Variable.objects.get(name='foo').label == ''

    assert AttributeDefinition.objects.get(name='title').description == ''

    assert AttributeDefinition.objects.get(name='title').order == 1

    assert AttributeDefinition.objects.get(name='first_name').order == 2

    assert AttributeDefinition.objects.get(name='prefered_color').order == 12

    # import valid content

    resp = app.get('/', status=200)

    resp = resp.click('Import')

    resp.form['parameters_json'] = Upload(

        'export.json', export_json.encode('utf-8'), 'application/json')

    resp = resp.form.submit()

    assert Variable.objects.count() == 3

    assert AttributeDefinition.objects.count() == 12

    assert Variable.objects.get(name='foo').label == 'bar'

    assert AttributeDefinition.objects.get(name='title').description == 'genre'

    assert AttributeDefinition.objects.get(name='title').order == 1

    assert AttributeDefinition.objects.get(name='first_name').order == 2

    assert AttributeDefinition.objects.get(name='prefered_color').order == 12

    # import empty json

    resp = app.get('/', status=200)

    resp = resp.click('Import')

    resp.form['parameters_json'] = Upload(

        'export.json', b'{}', 'application/json')

    resp = resp.form.submit()

    assert Variable.objects.count() == 3

    assert AttributeDefinition.objects.count() == 12

    assert Variable.objects.get(name='foo').label == 'bar'

    assert AttributeDefinition.objects.get(name='title').description == 'genre'

    assert AttributeDefinition.objects.get(name='title').order == 1

    assert AttributeDefinition.objects.get(name='first_name').order == 2

    assert AttributeDefinition.objects.get(name='prefered_color').order == 12

    # import from scratch

    Variable.objects.all().delete()

    AttributeDefinition.objects.all().delete()

    Variable.objects.create(name='foo2', value='bar2', service=combo).save()

    AttributeDefinition.objects.create(name='prefered_color').save()

    assert Variable.objects.count() == 1

    assert AttributeDefinition.objects.count() == 1

    resp = app.get('/', status=200)

    resp = resp.click('Import')

    resp.form['parameters_json'] = Upload(

        'export.json', export_json.encode('utf-8'), 'application/json')

    resp = resp.form.submit()

    assert Variable.objects.count() == 2

    assert AttributeDefinition.objects.count() == 12

    assert Variable.objects.get(name='foo').label == 'bar'

    assert AttributeDefinition.objects.get(name='title').order == 3

    assert AttributeDefinition.objects.get(name='first_name').order == 2

    assert AttributeDefinition.objects.get(name='prefered_color').order == 1

    # import invalid json

    resp = app.get('/', status=200)

    resp = resp.click('Import')

    resp.form['parameters_json'] = Upload(

        'export.json', b'garbage', 'application/json')

    resp = resp.form.submit()

    assert resp.html.find('ul', {'class': 'errorlist'}).li.text == \

        'File is not in the expected JSON format.'

    assert Variable.objects.count() == 2

    assert AttributeDefinition.objects.count() == 12

    # import corrupted json

    export['variables'][0]['label'] = 'foofoo'

    fields = export['profile']['fields']

    assert fields[1]['name'] == 'title'

    fields[1]['label'] = 'Genre'

    export_json = json.dumps(export)

    resp = app.get('/', status=200)

    resp = resp.click('Import')

    resp.form['parameters_json'] = Upload(

        'export.json', export_json.encode('utf-8'), 'application/json')

    resp = resp.form.submit()

    assert 'Parameters import fails: IntegrityError' in resp.html.find(

        'ul', {'class': 'errorlist'}).li.text

    assert Variable.objects.count() == 2

    assert AttributeDefinition.objects.count() == 12

    assert Variable.objects.get(name='foo').label == 'bar'