Produits Entr'ouvert » Chrono

# chrono - agendas system

# Copyright (C) 2020  Entr'ouvert

#

# This program is free software: you can redistribute it and/or modify it

# under the terms of the GNU Affero General Public License as published

# by the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU Affero General Public License for more details.

#

# You should have received a copy of the GNU Affero General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from django.core.management.base import BaseCommand

from chrono.agendas.models import Desk

class Command(BaseCommand):

    help = 'Synchronize time period exceptions from settings'

    def handle(self, **options):

        for desk in Desk.objects.all():

            desk.import_timeperiod_exceptions_from_settings()

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

# Generated by Django 1.11.18 on 2020-03-30 16:03

from __future__ import unicode_literals

from django.db import migrations, models

class Migration(migrations.Migration):

    dependencies = [

        ('agendas', '0040_timeperiod_agenda'),

    ]

    operations = [

        migrations.AddField(

            model_name='timeperiodexceptionsource', name='enabled', field=models.BooleanField(default=True),

        ),

        migrations.AddField(

            model_name='timeperiodexceptionsource',

            name='last_update',

            field=models.DateTimeField(auto_now=True, null=True),

        ),

        migrations.AddField(

            model_name='timeperiodexceptionsource',

            name='settings_label',

            field=models.CharField(max_length=150, null=True),

        ),

        migrations.AddField(

            model_name='timeperiodexceptionsource',

            name='settings_slug',

            field=models.CharField(max_length=150, null=True),

        ),

    ]

from django.utils.module_loading import import_string

from django.utils.translation import ugettext, ugettext_lazy as _

        first_created = not self.pk

        if first_created:

            self.import_timeperiod_exceptions_from_settings(enable=True)

    def import_timeperiod_exceptions_from_settings(self, enable=False):

        start_update = now()

        for slug, source_info in settings.EXCEPTIONS_SOURCES.items():

            label = source_info['label']

            try:

                source = TimePeriodExceptionSource.objects.get(desk=self, settings_slug=slug)

            except TimePeriodExceptionSource.DoesNotExist:

                source = TimePeriodExceptionSource.objects.create(

                    desk=self, settings_slug=slug, enabled=False

                )

            source.settings_label = _(label)

            source.save()

            if enable or source.enabled:  # if already enabled, update anyway

                source.enable()

        TimePeriodExceptionSource.objects.filter(

            desk=self, settings_slug__isnull=False, last_update__lt=start_update

        ).delete()  # source was not in settings anymore

    settings_slug = models.CharField(null=True, max_length=150)

    settings_label = models.CharField(null=True, max_length=150)

    last_update = models.DateTimeField(auto_now=True, null=True)

    enabled = models.BooleanField(default=True)

        if self.settings_label is not None:

            return ugettext(self.settings_label)

    def enable(self):

        source_info = settings.EXCEPTIONS_SOURCES.get(self.settings_slug)

        if not source_info:

            return

        source_class = import_string(source_info['class'])

        calendar = source_class()

        this_year = now().year

        days = [day for year in range(this_year, this_year + 3) for day in calendar.holidays(year)]

        with transaction.atomic():

            self.timeperiodexception_set.all().delete()

            for day, label in days:

                start_datetime = make_aware(datetime.datetime.combine(day, datetime.datetime.min.time()))

                end_datetime = start_datetime + datetime.timedelta(days=1)

                TimePeriodException.objects.create(

                    desk=self.desk,

                    source=self,

                    label=_(label),

                    start_datetime=start_datetime,

                    end_datetime=end_datetime,

                )

            self.enabled = True

            self.save()

    def disable(self):

        self.timeperiodexception_set.all().delete()

        self.enabled = False

        self.save()

          <a {% if not object.enabled %}class="disabled"{% endif %} title="{{ object }}" {% if object.ics_url %}href="{{ object }}"{% endif %}>{% if object.ics_filename %}{{ object|truncatechars:50 }}{% else %}{{ object|truncatechars:50 }}{% endif %}</a>

          {% elif object.ics_url %}

          {% if not object.settings_slug %}

          {% else %}

          <a class="link-action-text" href="{% url 'chrono-manager-time-period-exception-source-toggle' object.pk %}">({{ object.enabled|yesno:_("disable,enable") }})</a>

          {% endif %}

    url(

        r'^time-period-exceptions-source/(?P<pk>\d+)/toggle$',

        views.time_period_exception_source_toggle,

        name='chrono-manager-time-period-exception-source-toggle',

    ),

class TimePeriodExceptionSourceToggleView(ManagedDeskSubobjectMixin, DetailView):

    model = TimePeriodExceptionSource

    def get_object(self, queryset=None):

        source = super().get_object(queryset)

        if source.settings_slug is None:

            raise Http404('This source cannot be enabled nor disabled')

        return source

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

        source = self.get_object()

        if source.enabled:

            source.disable()

            message = _('Exception source %(source)s has been disabled on desk %(desk)s.')

        else:

            source.enable()

            message = _('Exception source %(source)s has been enabled on desk %(desk)s.')

        messages.info(self.request, message % {'source': source, 'desk': source.desk})

        return HttpResponseRedirect(

            reverse('chrono-manager-agenda-settings', kwargs={'pk': source.desk.agenda_id})

        )

time_period_exception_source_toggle = TimePeriodExceptionSourceToggleView.as_view()

_ = lambda s: s

EXCEPTIONS_SOURCES = {

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

}

0 0 1 1 * chrono /usr/bin/chrono-manage -- tenant_command sync_desks_timeperiod_exceptions_from_settings --all-tenants

        'workalendar',

EXCEPTIONS_SOURCES = {}

from django.test import override_settings

@override_settings(

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

)

def test_timeperiodexception_from_settings():

    agenda = Agenda(label=u'Test 1 agenda')

    agenda.save()

    desk = Desk(label='Test 1 desk', agenda=agenda)

    desk.save()

    # first save automatically load exceptions

    source = TimePeriodExceptionSource.objects.get(desk=desk)

    assert source.settings_slug == 'holidays'

    assert source.enabled

    assert TimePeriodException.objects.filter(desk=desk, source=source).exists()

    exception = TimePeriodException.objects.first()

    from workalendar.europe import France

    date, label = France().holidays()[0]

    exception = TimePeriodException.objects.filter(label=label).first()

    assert exception.end_datetime - exception.start_datetime == datetime.timedelta(days=1)

    assert localtime(exception.start_datetime).date() == date

    source.disable()

    assert not source.enabled

    assert not TimePeriodException.objects.filter(desk=desk, source=source).exists()

    source.enable()

    assert source.enabled

    assert TimePeriodException.objects.filter(desk=desk, source=source).exists()

def test_timeperiodexception_from_settings_command():

    setting = {

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

    }

    agenda = Agenda(label=u'Test 1 agenda')

    agenda.save()

    desk1 = Desk(label='Test 1 desk', agenda=agenda)

    desk1.save()

    with override_settings(**setting):

        desk2 = Desk(label='Test 2 desk', agenda=agenda)

        desk2.save()

        desk3 = Desk(label='Test 3 desk', agenda=agenda)

        desk3.save()

        source3 = TimePeriodExceptionSource.objects.get(desk=desk3)

        source3.disable()

        call_command('sync_desks_timeperiod_exceptions_from_settings')

    assert not TimePeriodExceptionSource.objects.get(desk=desk1).enabled

    source2 = TimePeriodExceptionSource.objects.get(desk=desk2)

    assert source2.enabled

    source3.refresh_from_db()

    assert not source3.enabled

    exceptions_count = source2.timeperiodexception_set.count()

    # Alsace Moselle has more holidays

    setting['EXCEPTIONS_SOURCES']['holidays']['class'] = 'workalendar.europe.FranceAlsaceMoselle'

    with override_settings(**setting):

        call_command('sync_desks_timeperiod_exceptions_from_settings')

    source2.refresh_from_db()

    assert exceptions_count < source2.timeperiodexception_set.count()

    setting['EXCEPTIONS_SOURCES'] = {}

    with override_settings(**setting):

        call_command('sync_desks_timeperiod_exceptions_from_settings')

    assert not TimePeriodExceptionSource.objects.exists()

from django.test import override_settings

@override_settings(

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

)

def test_meetings_agenda_time_period_exception_source_from_settings(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)

    )

    assert TimePeriodException.objects.exists()

    login(app)

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

    resp = resp.click('Settings')

    resp = resp.click('upload')

    assert 'Holidays' in resp.text

    assert 'disabled' not in resp.text

    assert 'refresh' not in resp.text

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

    assert not TimePeriodException.objects.exists()

    resp = resp.click('upload')

    assert 'Holidays' in resp.text

    assert 'disabled' in resp.text

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

    assert TimePeriodException.objects.exists()

    resp = resp.click('upload')

    assert 'disabled' not in resp.text

def test_meetings_agenda_time_period_exception_source_try_disable_ics(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)

    )

    source = TimePeriodExceptionSource.objects.create(desk=desk, ics_url='https://example.com/test.ics')

    login(app)

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

    resp = resp.click('Settings')

    resp = resp.click('upload')

    assert 'test.ics' in resp.text

    assert app.get('/manage/time-period-exceptions-source/%s/toggle' % source.pk, status=404)