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

        desks.append(Desk(agenda=agenda, slug='_exceptions_holder'))

    Desk.objects.bulk_create(desks)

class Migration(migrations.Migration):

    dependencies = [

        ('agendas', '0076_event_recurrence_end_date'),

    ]

    operations = [

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

    ]

        created = bool(not self.pk)

        if created and self.kind == 'events':

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

            desk.import_timeperiod_exceptions_from_settings()

            agenda['exceptions_desk'] = self.desk_set.get().export_json()

            exceptions_desk = data.pop('exceptions_desk', None)

            if exceptions_desk:

                exceptions_desk['agenda'] = agenda

                Desk.import_json(exceptions_desk)

        exceptions = self.get_recurrence_exceptions(min_start, max_start)

            events.extend(

                event.get_recurrences(

                    min_start, max_start, excluded_datetimes, exceptions, slug_separator=':'

                )

            )

    def get_recurrence_exceptions(self, min_start, max_start):

        return TimePeriodException.objects.filter(

            Q(desk__slug='_exceptions_holder', desk__agenda=self)

            | Q(

                unavailability_calendar__desks__slug='_exceptions_holder',

                unavailability_calendar__desks__agenda=self,

            ),

            start_datetime__lt=max_start,

            end_datetime__gt=min_start,

        )

    def get_recurrences(

        self, min_datetime, max_datetime, excluded_datetimes=None, exceptions=None, slug_separator='--'

    ):

        if exceptions is None:

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

        for exception in exceptions:

            exception_start = localtime(exception.start_datetime)

            event_start = localtime(self.start_datetime)

            if event_start.time() < exception_start.time():

                exception_start += datetime.timedelta(days=1)

            exception_start = exception_start.replace(

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

            )

            rrule_set.exrule(

                rrule(

                    freq=DAILY,

                    dtstart=make_naive(exception_start),

                    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 %}

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

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

                recurrence_rule__isnull=False

            ).exists()

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

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

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

                end_datetime__gt=now(),

            )

            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.get(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'

        assert len(ctx.captured_queries) == 6

    exceptions_desk = Desk.objects.get(agenda=agenda_events, slug='_exceptions_holder')

    TimePeriodException.objects.create(desk=exceptions_desk, start_datetime=tpx_start, end_datetime=tpx_end)

    assert TimePeriodException.objects.get(desk__agenda__kind='meetings').start_datetime == tpx_start

    assert TimePeriodException.objects.get(desk__agenda__kind='meetings').end_datetime == tpx_end

    assert TimePeriodException.objects.get(desk__agenda__kind='events').start_datetime == tpx_start

    assert TimePeriodException.objects.get(desk__agenda__kind='events').end_datetime == tpx_end

@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