Produits Entr'ouvert » Chrono

# -*- coding: utf-8 -*-

# Generated by Django 1.11.18 on 2021-01-27 16:46

from __future__ import unicode_literals

from django.db import migrations

def create_exceptions_desk(apps, schema_editor):

    Agenda = apps.get_model('agendas', 'Agenda')

    Desk = apps.get_model('agendas', 'Desk')

    desks = []

    for agenda in Agenda.objects.filter(kind='events'):

        desk = Desk.objects.create(agenda=agenda, slug='_exceptions_holder')

        desk.import_timeperiod_exceptions_from_settings()

class Migration(migrations.Migration):

    dependencies = [

        ('agendas', '0075_event_recurrence_end_date'),

    ]

    operations = [

        migrations.RunPython(create_exceptions_desk, migrations.RunPython.noop),

    ]

        exceptions = self.get_recurrence_exceptions(min_start, max_start)

            events.extend(event.get_recurrences(min_start, max_start, excluded_datetimes, exceptions))

    def get_recurrence_exceptions(self, min_start, max_start):

        desk, _ = Desk.objects.get_or_create(agenda=self, slug='_exceptions_holder')

        return TimePeriodException.objects.filter(

            Q(desk=desk) | Q(unavailability_calendar__desks=desk),

            start_datetime__lt=max_start,

            end_datetime__gt=min_start,

        )

    def get_recurrences(self, min_datetime, max_datetime, excluded_datetimes=None, exceptions=None):

        exceptions = exceptions or self.agenda.get_recurrence_exceptions(min_datetime, max_datetime)

        for exception in exceptions:

            dtstart = localtime(exception.start_datetime)

            start_datetime = localtime(self.start_datetime)

            if start_datetime.time() < dtstart.time():

                dtstart += datetime.timedelta(days=1)

            dtstart = dtstart.replace(

                hour=start_datetime.hour, minute=start_datetime.minute, second=0, microsecond=0

            )

            rrule_set.exrule(

                rrule(

                    freq=DAILY,

                    dtstart=make_naive(dtstart),

                    until=make_naive(exception.end_datetime),

                )

            )

{% if has_recurring_events %}

<div class="section">

<h3>{% trans "Recurrence exceptions" %}

<a rel="popup" class="button" href="{% url 'chrono-manager-desk-add-import-time-period-exceptions' pk=desk.pk %}">{% trans 'Configure' %}</a>

</h3>

<div>

<ul class="objects-list single-links">

{% for exception in exceptions|slice:":5" %}

   <li><a rel="popup" {% if not exception.read_only %}href="{% url 'chrono-manager-time-period-exception-edit' pk=exception.pk %}"{% endif %}>

  {{ exception }}

  {% if not exception.read_only %}

  <a rel="popup" class="delete" href="{% url 'chrono-manager-time-period-exception-delete' pk=exception.id %}">{% trans "remove" %}</a>

  {% endif %}

{% endfor %}

{% if exceptions|length > 5 %}

<li><a class="timeperiod-exception-all desk-{{ desk.pk }}" rel="popup" data-selector="div.timeperiod" href="{% url 'chrono-manager-time-period-exception-extract-list' pk=desk.id %}">({% trans 'see all exceptions' %})</a></li>

{% endif %}

<li><a class="add" rel="popup" href="{% url 'chrono-manager-agenda-add-time-period-exception' agenda_pk=object.pk pk=desk.pk %}">{% trans 'Add a time period exception' %}</a></li>

</ul>

</div>

</div>

{% endif %}

        elif self.object.kind == 'events':

            desk = Desk.objects.create(agenda=self.object, slug='_exceptions_holder')

            desk.import_timeperiod_exceptions_from_settings()

        if self.agenda.kind == 'events':

            context['has_recurring_events'] = self.agenda.event_set.filter(

                recurrence_rule__isnull=False

            ).exists()

            desk, _ = Desk.objects.get_or_create(agenda=self.agenda, slug='_exceptions_holder')

            context['exceptions'] = TimePeriodException.objects.filter(

                Q(desk=desk) | Q(unavailability_calendar__desks=desk),

                end_datetime__gt=now(),

            ).select_related('source')

            context['desk'] = desk

@override_settings(

    EXCEPTIONS_SOURCES={

        'holidays': {'class': 'workalendar.europe.France', 'label': 'Holidays'},

    }

)

def test_recurring_events_exceptions(freezer):

    freezer.move_to('2021-05-01 12:00')

    agenda = Agenda.objects.create(label='Agenda', kind='events')

    desk = Desk.objects.create(slug='_exceptions_holder', agenda=agenda)

    event = Event.objects.create(

        agenda=agenda,

        start_datetime=now(),

        repeat='daily',

        places=5,

    )

    event.refresh_from_db()

    start_datetime = localtime(event.start_datetime)

    recurrences = event.get_recurrences(now(), now() + datetime.timedelta(days=7))

    assert recurrences[0].start_datetime.strftime('%m-%d') == '05-01'

    first_of_may = recurrences[0]

    recurrence = event.get_or_create_event_recurrence(first_of_may.start_datetime)

    recurrence.delete()

    desk.import_timeperiod_exceptions_from_settings(enable=True)

    recurrences = event.get_recurrences(now(), now() + datetime.timedelta(days=7))

    # 05-01 is a holiday

    assert recurrences[0].start_datetime.strftime('%m-%d') == '05-02'

    with pytest.raises(ValueError):

        recurrence = event.get_or_create_event_recurrence(first_of_may.start_datetime)

    first_event = recurrences[0]

    # exception before first_event start_datetime

    time_period_exception = TimePeriodException.objects.create(

        desk=desk,

        start_datetime=first_event.start_datetime - datetime.timedelta(hours=1),

        end_datetime=first_event.start_datetime - datetime.timedelta(minutes=30),

    )

    recurrences = event.get_recurrences(now(), now() + datetime.timedelta(days=7))

    assert recurrences[0].start_datetime.strftime('%m-%d') == '05-02'

    # exception wraps around first_event start_datetime

    time_period_exception.end_datetime = first_event.start_datetime + datetime.timedelta(minutes=30)

    time_period_exception.save()

    recurrences = event.get_recurrences(now(), now() + datetime.timedelta(days=7))

    assert recurrences[0].start_datetime.strftime('%m-%d') == '05-03'

    # exception starts after first_event start_datetime

    time_period_exception.start_datetime = first_event.start_datetime + datetime.timedelta(minutes=15)

    time_period_exception.save()

    recurrences = event.get_recurrences(now(), now() + datetime.timedelta(days=7))

    assert recurrences[0].start_datetime.strftime('%m-%d') == '05-02'

    assert recurrences[1].start_datetime.strftime('%m-%d') == '05-03'

    # exception spans multiple days

    time_period_exception.end_datetime = first_event.start_datetime + datetime.timedelta(days=3)

    time_period_exception.save()

    recurrences = event.get_recurrences(now(), now() + datetime.timedelta(days=7))

    assert recurrences[0].start_datetime.strftime('%m-%d') == '05-02'

    assert recurrences[1].start_datetime.strftime('%m-%d') == '05-06'

    # move exception to unavailability calendar

    unavailability_calendar = UnavailabilityCalendar.objects.create(label='Calendar')

    time_period_exception.desk = None

    time_period_exception.unavailability_calendar = unavailability_calendar

    time_period_exception.save()

    recurrences = event.get_recurrences(now(), now() + datetime.timedelta(days=7))

    assert recurrences[0].start_datetime.strftime('%m-%d') == '05-02'

    assert recurrences[1].start_datetime.strftime('%m-%d') == '05-03'

    unavailability_calendar.desks.add(desk)

    recurrences = event.get_recurrences(now(), now() + datetime.timedelta(days=7))

    assert recurrences[0].start_datetime.strftime('%m-%d') == '05-02'

    assert recurrences[1].start_datetime.strftime('%m-%d') == '05-06'

@override_settings(

    EXCEPTIONS_SOURCES={

        'holidays': {'class': 'workalendar.europe.France', 'label': 'Holidays'},

    }

)

def test_recurring_events_manage_exceptions(settings, app, admin_user, freezer):

    freezer.move_to('2021-07-01 12:10')

    app = login(app)

    resp = app.get('/manage/')

    resp = resp.click('New')

    resp.form['label'] = 'Foo bar'

    resp.form['kind'] = 'events'

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

    agenda = Agenda.objects.get(label='Foo bar')

    assert agenda.desk_set.count() == 1

    desk = agenda.desk_set.get(slug='_exceptions_holder')

    event = Event.objects.create(start_datetime=now(), places=10, agenda=agenda)

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

    assert not 'Recurrence exceptions' in resp.text

    event.repeat = 'daily'

    event.save()

    resp = app.get('/manage/agendas/%s/%s/%s/' % (agenda.id, 2021, 7))

    assert len(resp.pyquery.find('.event-info')) == 31

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

    assert 'Recurrence exceptions' in resp.text

    resp = resp.click('Add a time period exception')

    resp.form['start_datetime_0'] = now().strftime('%Y-%m-%d')

    resp.form['start_datetime_1'] = now().strftime('%H:%M')

    resp.form['end_datetime_0'] = (now() + datetime.timedelta(days=7)).strftime('%Y-%m-%d')

    resp.form['end_datetime_1'] = (now() + datetime.timedelta(days=7)).strftime('%H:%M')

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

    assert desk.timeperiodexception_set.count() == 1

    resp = app.get('/manage/agendas/%s/%s/%s/' % (agenda.id, 2021, 7))

    assert len(resp.pyquery.find('.event-info')) == 24

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

    resp = resp.click('Configure', href='exceptions')

    resp = resp.click('enable').follow()

    assert TimePeriodException.objects.count() > 1

    assert 'Bastille Day' in resp.text

    resp = app.get('/manage/agendas/%s/%s/%s/' % (agenda.id, 2021, 7))

    assert len(resp.pyquery.find('.event-info')) == 23

    # add recurrence end date, which lead to recurrences creation

    resp = app.get('/manage/agendas/%s/events/%s/edit' % (agenda.id, event.id))

    resp.form['recurrence_end_date'] = (now() + datetime.timedelta(days=31)).strftime('%Y-%m-%d')

    resp = resp.form.submit()

    # recurrences corresponding to exceptions have not been created

    assert Event.objects.count() == 24