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 urllib.parse import urljoin

from django.conf import settings as django_settings

from django.core.mail import send_mail

from django.core.management.base import BaseCommand

from django.db.transaction import atomic

from django.utils.translation import ugettext_lazy as _

from chrono.agendas.models import AgendaNotificationsSettings

class Command(BaseCommand):

    help = 'Send email notifications'

    def handle(self, **options):

        notifications_settings = AgendaNotificationsSettings.objects.all()

        for settings in notifications_settings.select_related('agenda'):

            for setting, recipients in settings:

                if not recipients:

                    continue

                status = setting.replace('_event', '')

                events = settings.agenda.event_set.filter(**{status: True, 'was_' + status: False})

                for event in events:

                    self.send_notification(event, status, recipients)

    def send_notification(self, event, status, recipients):

        with atomic():

            setattr(event, 'was_' + status, True)

            event.save()

            subject = _('Alert: event "%(event)s" is %(status)s') % {

                'event': event,

                'status': status.replace('_', ' '),

            }

            body = _('You can view it here: %s.') % urljoin(

                django_settings.SITE_BASE_URL, event.get_absolute_view_url()

            )

            send_mail(subject, body, django_settings.DEFAULT_FROM_EMAIL, recipients)

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

# Generated by Django 1.11.18 on 2020-07-16 13:15

from __future__ import unicode_literals

import django.contrib.postgres.fields

from django.db import migrations, models

import django.db.models.deletion

class Migration(migrations.Migration):

    dependencies = [

        ('agendas', '0053_event_almost_full'),

    ]

    operations = [

        migrations.CreateModel(

            name='AgendaNotificationsSettings',

            fields=[

                (

                    'id',

                    models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID'),

                ),

                (

                    'almost_full_event',

                    models.CharField(

                        blank=True,

                        choices=[

                            ('edit-role', 'Edit Role'),

                            ('view-role', 'View Role'),

                            ('use-email-field', 'Specify email addresses manually'),

                        ],

                        max_length=16,

                        verbose_name='Almost full event (90%)',

                    ),

                ),

                (

                    'almost_full_event_emails',

                    django.contrib.postgres.fields.ArrayField(

                        base_field=models.EmailField(max_length=254), blank=True, null=True, size=None

                    ),

                ),

                (

                    'full_event',

                    models.CharField(

                        blank=True,

                        choices=[

                            ('edit-role', 'Edit Role'),

                            ('view-role', 'View Role'),

                            ('use-email-field', 'Specify email addresses manually'),

                        ],

                        max_length=16,

                        verbose_name='Full event',

                    ),

                ),

                (

                    'full_event_emails',

                    django.contrib.postgres.fields.ArrayField(

                        base_field=models.EmailField(max_length=254), blank=True, null=True, size=None

                    ),

                ),

                (

                    'cancelled_event',

                    models.CharField(

                        blank=True,

                        choices=[

                            ('edit-role', 'Edit Role'),

                            ('view-role', 'View Role'),

                            ('use-email-field', 'Specify email addresses manually'),

                        ],

                        max_length=16,

                        verbose_name='Cancelled event',

                    ),

                ),

                (

                    'cancelled_event_emails',

                    django.contrib.postgres.fields.ArrayField(

                        base_field=models.EmailField(max_length=254), blank=True, null=True, size=None

                    ),

                ),

                (

                    'agenda',

                    models.OneToOneField(

                        on_delete=django.db.models.deletion.CASCADE,

                        related_name='notifications_settings',

                        to='agendas.Agenda',

                    ),

                ),

            ],

        ),

        migrations.AddField(

            model_name='event', name='was_almost_full', field=models.BooleanField(default=False),

        ),

        migrations.AddField(

            model_name='event', name='was_cancelled', field=models.BooleanField(default=False),

        ),

        migrations.AddField(model_name='event', name='was_full', field=models.BooleanField(default=False),),

    ]

from django.contrib.postgres.fields import ArrayField

            if hasattr(self, 'notifications_settings'):

                agenda['notifications_settings'] = self.notifications_settings.export_json()

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

                AgendaNotificationsSettings.objects.filter(agenda=agenda).delete()

            if notifications_settings:

                notifications_settings['agenda'] = agenda

                AgendaNotificationsSettings.import_json(notifications_settings).save()

    # flags for detecting changes

    was_almost_full = models.BooleanField(default=False)

    was_full = models.BooleanField(default=False)

    was_cancelled = models.BooleanField(default=False)

    def get_absolute_view_url(self):

        return reverse('chrono-manager-event-view', kwargs={'pk': self.agenda.id, 'event_pk': self.id})

class AgendaNotificationsSettings(models.Model):

    EMAIL_FIELD = 'use-email-field'

    VIEW_ROLE = 'view-role'

    EDIT_ROLE = 'edit-role'

    CHOICES = [

        (EDIT_ROLE, _('Edit Role')),

        (VIEW_ROLE, _('View Role')),

        (EMAIL_FIELD, _('Specify email addresses manually')),

    ]

    agenda = models.OneToOneField(Agenda, on_delete=models.CASCADE, related_name='notifications_settings')

    almost_full_event = models.CharField(

        max_length=16, blank=True, choices=CHOICES, verbose_name=_('Almost full event (90%)')

    )

    almost_full_event_emails = ArrayField(models.EmailField(), blank=True, null=True)

    full_event = models.CharField(max_length=16, blank=True, choices=CHOICES, verbose_name=_('Full event'))

    full_event_emails = ArrayField(models.EmailField(), blank=True, null=True)

    cancelled_event = models.CharField(

        max_length=16, blank=True, choices=CHOICES, verbose_name=_('Cancelled event')

    )

    cancelled_event_emails = ArrayField(models.EmailField(), blank=True, null=True)

    def __iter__(self):

        for field in self.get_setting_fields():

            yield (field.name, self.get_recipients(field.name))

    @classmethod

    def get_setting_fields(cls):

        return [field for field in cls._meta.get_fields() if isinstance(field, models.CharField)]

    @classmethod

    def get_email_field_names(cls):

        return [field.name for field in cls._meta.get_fields() if isinstance(field, ArrayField)]

    def get_recipients(self, setting):

        value = getattr(self, setting)

        if not value:

            return []

        if value == self.EMAIL_FIELD:

            return getattr(self, setting + '_emails')

        role = self.get_role_from_choice(value)

        if not role or not hasattr(role, 'role'):

            return []

        emails = role.role.emails

        if role.role.emails_to_members:

            emails.extend(role.user_set.values_list('email', flat=True))

        return emails

    @property

    def display_info(self):

        for field in self.get_setting_fields():

            choice = getattr(self, field.name)

            if not choice:

                continue

            if choice == self.EMAIL_FIELD:

                emails = getattr(self, field.name + '_emails')

                yield (field.verbose_name, ', '.join(emails))

            else:

                role = self.get_role_from_choice(choice)

                if role:

                    display_name = getattr(self, 'get_' + field.name + '_display')()

                    yield (field.verbose_name, '%s (%s)' % (display_name, role))

    def get_role_from_choice(self, choice):

        if choice == self.EDIT_ROLE:

            return self.agenda.edit_role

        elif choice == self.VIEW_ROLE:

            return self.agenda.view_role

    @classmethod

    def import_json(cls, data):

        data = clean_import_data(cls, data)

        return cls(**data)

    def export_json(self):

        return {

            'almost_full_event': self.almost_full_event,

            'almost_full_event_emails': self.almost_full_event_emails,

            'full_event': self.full_event,

            'full_event_emails': self.full_event_emails,

            'cancelled_event': self.cancelled_event,

            'cancelled_event_emails': self.cancelled_event_emails,

        }

    AgendaNotificationsSettings,

class AgendaNotificationsForm(forms.ModelForm):

    class Meta:

        model = AgendaNotificationsSettings

        fields = '__all__'

        widgets = {

            'agenda': forms.HiddenInput(),

        }

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

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

        for email_field in AgendaNotificationsSettings.get_email_field_names():

            self.fields[email_field].widget.attrs['size'] = 80

            self.fields[email_field].label = ''

            self.fields[email_field].help_text = _('Enter a comma separated list of email addresses.')

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

{% load i18n %}

{% block breadcrumb %}

{{ block.super }}

<a href="">{% trans "Notification settings" %}</a>

{% endblock %}

{% block appbar %}

<h2>{% trans "Notification settings" %}</h2>

{% endblock %}

{% block content %}

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

  {% csrf_token %}

  {{ form.as_p }}

  <div class="buttons">

    <button class="submit-button">{% trans "Save" %}</button>

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

  </div>

  <script>

  $('select').change(function(){

    role_field_id = $(this).attr('id')

    email_field_id = '#' + role_field_id + '_emails'

    if ($(this).val() == 'use-email-field')

      $(email_field_id).parent('p').show();

    else

      $(email_field_id).parent('p').hide();

  });

  $('select').trigger('change');

  </script>

</form>

{% endblock %}

<div class="section">

<h3>{% trans "Notifications" %}</h3>

<div>

<ul>

{% for setting, value in object.notifications_settings.display_info %}

  <li>

  {% blocktrans %}

  {{ setting }}: {{ value }} will be notified.

  {% endblocktrans %}

  </li>

{% empty %}

{% trans "Notifications are disabled for this agenda." %}

{% endfor %}

</ul>

<a rel="popup" href="{% url 'chrono-manager-agenda-notifications-settings' pk=object.id %}">{% trans 'Configure' %}</a>

</div>

</div>

    url(

        r'^agendas/(?P<pk>\d+)/notifications$',

        views.agenda_notifications_settings,

        name='chrono-manager-agenda-notifications-settings',

    ),

from django.db.models import Q, F

    AgendaNotificationsSettings,

    AgendaNotificationsForm,

class AgendaNotificationsSettingsView(ManagedAgendaMixin, UpdateView):

    template_name = 'chrono/manager_agenda_notifications_form.html'

    model = AgendaNotificationsSettings

    form_class = AgendaNotificationsForm

    def get_object(self):

        try:

            return self.agenda.notifications_settings

        except AgendaNotificationsSettings.DoesNotExist:

            # prevent old events from sending notifications

            statuses = ('almost_full', 'full', 'cancelled')

            kwargs = {'was_' + status: F(status) for status in statuses}

            self.agenda.event_set.update(**kwargs)

            return AgendaNotificationsSettings.objects.create(agenda=self.agenda)

agenda_notifications_settings = AgendaNotificationsSettingsView.as_view()

SITE_BASE_URL = 'https://example.com'

from django.contrib.auth.models import Group, User

    AgendaNotificationsSettings,

@mock.patch('django.contrib.auth.models.Group.role', create=True)

@pytest.mark.parametrize(

    'emails_to_members,emails',

    [(False, []), (False, ['test@entrouvert.com']), (True, []), (True, ['test@entrouvert.com']),],

)

def test_agenda_notifications_role_email(mocked_role, emails_to_members, emails, mailoutbox):

    group = Group.objects.create(name='group')

    user = User.objects.create(username='user', email='user@entrouvert.com')

    user.groups.add(group)

    mocked_role.emails_to_members = emails_to_members

    mocked_role.emails = emails

    expected_recipients = emails

    if emails_to_members:

        expected_recipients.append(user.email)

    expected_email_count = 1 if emails else 0

    agenda = Agenda.objects.create(label='Foo bar', kind='event', edit_role=group)

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

    settings = AgendaNotificationsSettings.objects.create(agenda=agenda)

    settings.almost_full_event = AgendaNotificationsSettings.EDIT_ROLE

    settings.save()

    # book 9/10 places to reach almost full state

    for i in range(9):

        Booking.objects.create(event=event)

    event.refresh_from_db()

    assert event.almost_full

    call_command('send_email_notifications')

    assert len(mailoutbox) == expected_email_count

    if mailoutbox:

        assert mailoutbox[0].recipients() == expected_recipients

        assert mailoutbox[0].subject == 'Alert: event "Hop" is almost full'

    # no new email on subsequent run

    call_command('send_email_notifications')

    assert len(mailoutbox) == expected_email_count

def test_agenda_notifications_email_list(mailoutbox):

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

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

    settings = AgendaNotificationsSettings.objects.create(agenda=agenda)

    settings.full_event = AgendaNotificationsSettings.EMAIL_FIELD

    settings.full_event_emails = recipients = ['hop@entrouvert.com', 'top@entrouvert.com']

    settings.save()

    for i in range(10):

        Booking.objects.create(event=event)

    event.refresh_from_db()

    assert event.full

    call_command('send_email_notifications')

    assert len(mailoutbox) == 1

    assert mailoutbox[0].recipients() == recipients

    assert mailoutbox[0].subject == 'Alert: event "Hop" is full'

    assert mailoutbox[0].body == 'You can view it here: https://example.com/manage/agendas/%s/events/%s/.' % (

        agenda.pk,

        event.pk,

    )

    # no new email on subsequent run

    call_command('send_email_notifications')

    assert len(mailoutbox) == 1

def test_agenda_notifications_cancelled(mailoutbox):

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

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

    settings = AgendaNotificationsSettings.objects.create(agenda=agenda)

    settings.cancelled_event = AgendaNotificationsSettings.EMAIL_FIELD

    settings.cancelled_event_emails = recipients = ['hop@entrouvert.com', 'top@entrouvert.com']

    settings.save()

    event.cancelled = True

    event.save()

    call_command('send_email_notifications')

    assert len(mailoutbox) == 1

    assert mailoutbox[0].recipients() == recipients

    assert mailoutbox[0].subject == 'Alert: event "Hop" is cancelled'

    # no new email on subsequent run

    call_command('send_email_notifications')

    assert len(mailoutbox) == 1

    AgendaNotificationsSettings,

def test_import_export_notification_settings():

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

    settings = AgendaNotificationsSettings.objects.create(

        agenda=agenda,

        almost_full_event=AgendaNotificationsSettings.EDIT_ROLE,

        full_event=AgendaNotificationsSettings.VIEW_ROLE,

        cancelled_event=AgendaNotificationsSettings.EMAIL_FIELD,

        cancelled_event_emails=['hop@entrouvert.com', 'top@entrouvert.com'],

    )

    output = get_output_of_command('export_site')

    payload = json.loads(output)

    agenda.delete()

    assert not AgendaNotificationsSettings.objects.exists()

    import_site(payload)

    agenda = Agenda.objects.first()

    AgendaNotificationsSettings.objects.get(

        agenda=agenda,

        almost_full_event=AgendaNotificationsSettings.EDIT_ROLE,

        full_event=AgendaNotificationsSettings.VIEW_ROLE,

        cancelled_event=AgendaNotificationsSettings.EMAIL_FIELD,

        cancelled_event_emails=['hop@entrouvert.com', 'top@entrouvert.com'],

    )

from django.core.management import call_command

def managers_group():

    group, _ = Group.objects.get_or_create(name='Managers')

    return group

@pytest.fixture

def manager_user(managers_group):

    user.groups.set([managers_group])

def test_agenda_notifications(app, admin_user, managers_group):

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

    login(app)

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

    assert 'Notifications' in resp.text

    assert 'Notifications are disabled' in resp.text

    resp = resp.click('Configure')

    resp.form['almost_full_event'] = 'edit-role'

    resp.form['full_event'] = 'view-role'

    resp.form['cancelled_event'] = 'use-email-field'

    resp.form['cancelled_event_emails'] = 'hop@entrouvert.com, top@entrouvert.com'

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

    settings = agenda.notifications_settings

    assert settings.almost_full_event == 'edit-role'

    assert settings.full_event == 'view-role'

    assert settings.cancelled_event == 'use-email-field'

    assert settings.cancelled_event_emails == ['hop@entrouvert.com', 'top@entrouvert.com']

    assert 'Cancelled event: hop@entrouvert.com, top@entrouvert.com will be notified' in resp.text

    assert not 'Full event:' in resp.text

    assert not 'Almost full event (90%):' in resp.text

    agenda.view_role = managers_group

    agenda.edit_role = Group.objects.create(name='hop')

    agenda.save()

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

    assert 'Almost full event (90%): Edit Role (hop) will be notified' in resp.text

    assert 'Full event: View Role (Managers) will be notified' in resp.text

def test_agenda_notifications_no_old_events(app, admin_user, mailoutbox):

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

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

    event.cancelled = True

    event.save()

    login(app)

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

    resp = resp.click('Configure')

    resp.form['cancelled_event'] = 'use-email-field'

    resp.form['cancelled_event_emails'] = 'hop@entrouvert.com'

    resp.form.submit()

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

    event.cancelled = True

    event.save()

    call_command('send_email_notifications')

    # no notification is sent for old event

    assert len(mailoutbox) == 1

    assert 'New event' in mailoutbox[0].subject