Produits Entr'ouvert » Chrono

                                   TimePeriodException, TimePeriodExceptionSource, WEEKDAYS_LIST)

class TimePeriodExceptionSourceRefreshForm(forms.ModelForm):

    ics_file = forms.FileField(

        label=_('ICS File'), required=False,

        help_text=_('ICS file containing events which will be considered as exceptions.'))

    class Meta:

        model = TimePeriodExceptionSource

        fields = []

    def __init__(self, *args, **kwargs):

        super().__init__(*args, **kwargs)

        if self.instance.ics_filename is None:

            del self.fields['ics_file']

          <a rel="popup" href="{% url 'chrono-manager-time-period-exception-source-refresh' object.pk %}">

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

{% load i18n %}

{% block appbar %}

<h2>{% if form.instance.ics_filename %}{% trans "Replace exceptions" %}{% else %}{% trans "Refresh exceptions" %}{% endif %}</h2>

{% endblock %}

{% block content %}

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

  {% if form.instance.ics_filename %}

    <p class="notice">{% trans "To replace existing exceptions, please upload a new file." %}</p>

  {% else %}

  <p class="notice">

    {% trans 'Press the button "Refresh" to refresh existing exceptions from:' %}

    <br />

    <a href="{{ form.instance.ics_url }}">{{ form.instance.ics_url }}</a>

  </p>

  {% endif %}

  {% csrf_token %}

  {{ form.as_p }}

  <p>

  </p>

  <div class="buttons">

    <button>{% if form.instance.ics_filename %}{% trans "Replace" %}{% else %}{% trans "Refresh" %}{% endif %}</button>

    <a class="cancel" href="{% url 'chrono-manager-agenda-settings' pk=agenda.id %}">{% trans 'Cancel' %}</a>

  </div>

</form>

{% endblock %}

        url(r'^time-period-exceptions-source/(?P<pk>\d+)/refresh$', views.time_period_exception_source_refresh,

            name='chrono-manager-time-period-exception-source-refresh'),

                    ExceptionsImportForm, TimePeriodExceptionSourceRefreshForm, AgendasImportForm, TimePeriodAddForm)

class TimePeriodExceptionSourceRefreshView(ManagedDeskSubobjectMixin, UpdateView):

    model = TimePeriodExceptionSource

    form_class = TimePeriodExceptionSourceRefreshForm

    template_name = 'chrono/manager_refresh_exceptions.html'

    def form_valid(self, form):

        exceptions = None

        try:

            if form.instance.ics_filename is not None:

                exceptions = form.instance.desk.import_timeperiod_exceptions_from_ics_file(

                    form.cleaned_data['ics_file'],

                    source=form.instance)

            else:

                exceptions = form.instance.desk.import_timeperiod_exceptions_from_remote_ics(

                    form.instance.ics_url,

                    source=form.instance)

        except ICSError as e:

            form.add_error(None, force_text(e))

            return self.form_invalid(form)

        if exceptions is not None:

            message = ungettext('An exception has been imported.',

                                '%(count)d exceptions have been imported.', exceptions)

            message = message % {'count': exceptions}

            messages.info(self.request, message)

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

time_period_exception_source_refresh = TimePeriodExceptionSourceRefreshView.as_view()

@mock.patch('chrono.agendas.models.requests.get')

def test_meetings_agenda_refresh_time_period_exception_source(mocked_get, app, admin_user):

    agenda = Agenda.objects.create(label='Foo bar', kind='meetings')

    desk = Desk.objects.create(agenda=agenda, label='Desk A')

    MeetingType(agenda=agenda, label='Blah').save()

    TimePeriod.objects.create(

        weekday=1, desk=desk, start_time=datetime.time(10, 0), end_time=datetime.time(12, 0)

    )

    ics_file_content = b"""BEGIN:VCALENDAR

VERSION:2.0

PRODID:-//foo.bar//EN

BEGIN:VEVENT

DTSTART:20180101

DTEND:20180101

SUMMARY:New Year's Eve

RRULE:FREQ=YEARLY

END:VEVENT

END:VCALENDAR"""

    login(app)

    # import a source from a file

    resp = app.get('/manage/agendas/%d/' % agenda.pk).follow()

    resp = resp.click('Settings')

    resp = resp.click('upload')

    resp.form['ics_file'] = Upload('exceptions.ics', ics_file_content, 'text/calendar')

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

    assert TimePeriodException.objects.filter(desk=desk).count() == 2

    source1 = TimePeriodExceptionSource.objects.latest('pk')

    assert source1.timeperiodexception_set.count() == 2

    exceptions1 = list(source1.timeperiodexception_set.order_by('pk'))

    # refresh the source

    resp = app.get('/manage/agendas/%d/' % agenda.pk).follow()

    resp = resp.click('Settings')

    resp = resp.click('upload')

    resp = resp.click(href='/manage/time-period-exceptions-source/%d/refresh' % source1.pk)

    resp.form['ics_file'] = Upload('exceptions.ics', ics_file_content, 'text/calendar')

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

    assert TimePeriodException.objects.count() == 2

    assert source1.timeperiodexception_set.count() == 2

    new_exceptions1 = list(source1.timeperiodexception_set.order_by('pk'))

    assert exceptions1[0].pk != new_exceptions1[0].pk

    assert exceptions1[1].pk != new_exceptions1[1].pk

    ics_url_content = """BEGIN:VCALENDAR

VERSION:2.0

PRODID:-//foo.bar//EN

BEGIN:VEVENT

DTSTART:20180101

DTEND:20180101

SUMMARY:New Year's Eve

END:VEVENT

END:VCALENDAR"""

    # import a source from an url

    resp = app.get('/manage/agendas/%d/' % agenda.pk).follow()

    resp = resp.click('Settings')

    resp = resp.click('upload')

    resp.form['ics_url'] = 'http://example.com/foo.ics'

    mocked_response = mock.Mock()

    mocked_response.text = ics_url_content

    mocked_get.return_value = mocked_response

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

    assert TimePeriodException.objects.filter(desk=desk).count() == 3

    source2 = TimePeriodExceptionSource.objects.latest('pk')

    assert source2.timeperiodexception_set.count() == 1

    exceptions2 = list(source2.timeperiodexception_set.order_by('pk'))

    # refresh the source

    resp = app.get('/manage/agendas/%d/' % agenda.pk).follow()

    resp = resp.click('Settings')

    resp = resp.click('upload')

    resp = resp.click(href='/manage/time-period-exceptions-source/%d/refresh' % source2.pk)

    mocked_response = mock.Mock()

    mocked_response.text = ics_url_content

    mocked_get.return_value = mocked_response

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

    assert TimePeriodException.objects.count() == 3

    assert source2.timeperiodexception_set.count() == 1

    new_exceptions2 = list(source2.timeperiodexception_set.order_by('pk'))

    assert exceptions2[0].pk != new_exceptions2[0].pk