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

    # TODO : export too : label, auto?

    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 var in parameters['variables']:

            objects.append(Variable(**var))

        for field in parameters['profile']['fields']:

            objects.append(AttributeDefinition(**field))

    except Exception as exc:

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

    Variable.objects.filter(service_pk__isnull=True).delete()

    AttributeDefinition.objects.all().delete()

    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') % 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')

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

        return response

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

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

import json

from webtest import Upload

def test_export_import_view(app, admin_user):

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

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

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

    assert [x for x in 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'] = 'foo'

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

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

    export_json = json.dumps(export)

    # 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.'

    # 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 resp.html.find('ul', {'class': 'errorlist'}).li.text == \

        "Parameters import fails: KeyError, 'variables'"

    # 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().follow()

    # check imported parameters

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

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

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

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

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

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

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

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

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