Produits Entr'ouvert » Chrono

        return created

class AgendasImportForm(forms.Form):

    agendas_json = forms.FileField(label=_('Agendas Export File'))

{% extends "chrono/manager_base.html" %}

{% load i18n %}

{% block appbar %}

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

{% 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 'chrono-manager-homepage' %}">{% trans 'Cancel' %}</a>

  </div>

</form>

{% endblock %}

  <a href="{% url 'chrono-manager-agenda-export' pk=object.id %}">{% trans 'Export' %}</a>

<a rel="popup" href="{% url 'chrono-manager-agendas-import' %}">{% trans 'Import' %}</a>

        url(r'^agendas/import/$', views.agendas_import,

            name='chrono-manager-agendas-import'),

        url(r'^agendas/(?P<pk>\w+)/export$', views.agenda_export,

            name='chrono-manager-agenda-export'),

    results = {'created': 0, 'updated': 0}

            created = Agenda.import_json(data, overwrite=overwrite)

            if created:

                results['created'] += 1

            else:

                results['updated'] += 1

    return results

                    ExceptionsImportForm, AgendasImportForm)

from .utils import import_site

class AgendasImportView(FormView):

    form_class = AgendasImportForm

    template_name = 'chrono/agendas_import.html'

    success_url = reverse_lazy('chrono-manager-homepage')

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

        if not request.user.is_staff:

            raise PermissionDenied()

        return super(AgendasImportView, self).dispatch(request, *args, **kwargs)

    def form_valid(self, form):

        try:

            agendas_json = json.load(self.request.FILES['agendas_json'])

        except ValueError:

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

            return self.form_invalid(form)

        results = import_site(agendas_json, overwrite=True)

        if results.get('created') == 0 and results.get('updated') == 0:

            messages.info(self.request, _('No agendas were found.'))

        else:

            if results.get('created') == 0:

                message1 = _('No agenda created.')

            else:

                message1 = ungettext('An agenda has been created.',

                                '%(count)d agendas have been created.', results['created'])

            if results.get('updated') == 0:

                message2 = _('No agenda updated.')

            else:

                message2 = ungettext('An agenda has been updated.',

                                '%(count)d agendas have been updated.', results['updated'])

            messages.info(self.request, '%s %s' % (message1, message2))

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

agendas_import = AgendasImportView.as_view()

class AgendaExport(ManagedAgendaMixin, DetailView):

    model = Agenda

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

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

        json.dump({'agendas': [self.get_object().export_json()]}, response, indent=2)

        return response

agenda_export = AgendaExport.as_view()

import json

from chrono.manager.utils import export_site

def test_import_agenda_as_manager(app, manager_user):

    # open /manage/ access to manager_user, and check agenda import is not

    # allowed.

    agenda = Agenda(label='Foo bar')

    agenda.view_role = manager_user.groups.all()[0]

    agenda.save()

    app = login(app, username='manager', password='manager')

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

    resp = app.get('/manage/agendas/import/', status=403)

def test_import_agenda(app, admin_user):

    agenda = Agenda(label=u'Foo bar')

    agenda.save()

    app = login(app)

    resp = app.get('/manage/agendas/%s/settings' % agenda.id)

    resp = resp.click('Export')

    assert resp.headers['content-type'] == 'application/json'

    agenda_export = resp.body

    # invalid json

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

    resp = resp.click('Import')

    resp.form['agendas_json'] = Upload('export.json', b'garbage', 'application/json')

    resp = resp.form.submit()

    assert 'File is not in the expected JSON format.' in resp.text

    # empty json

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

    resp = resp.click('Import')

    resp.form['agendas_json'] = Upload('export.json', b'{}', 'application/json')

    resp = resp.form.submit().follow()

    assert 'No agendas were found.' in resp.text

    # existing agenda

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

    resp = resp.click('Import')

    resp.form['agendas_json'] = Upload('export.json', agenda_export, 'application/json')

    resp = resp.form.submit().follow()

    assert 'No agenda created. An agenda has been updated.' in resp.text

    assert Agenda.objects.count() == 1

    # new agenda

    Agenda.objects.all().delete()

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

    resp = resp.click('Import')

    resp.form['agendas_json'] = Upload('export.json', agenda_export, 'application/json')

    resp = resp.form.submit().follow()

    assert 'An agenda has been created. No agenda updated.' in resp.text

    assert Agenda.objects.count() == 1