Produits Entr'ouvert » Passerelle

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

# Generated by Django 1.11.18 on 2020-10-08 15:31

from __future__ import unicode_literals

import django.contrib.postgres.fields

from django.db import migrations, models

class Migration(migrations.Migration):

    dependencies = [

        ('ovh', '0010_auto_20201008_1126'),

    ]

    operations = [

        migrations.AddField(

            model_name='ovhsmsgateway',

            name='alert_emails',

            field=django.contrib.postgres.fields.ArrayField(base_field=models.EmailField(blank=True, max_length=254), blank=True, null=True, size=None, verbose_name='Email addresses to send credit alerts to'),

        ),

        migrations.AddField(

            model_name='ovhsmsgateway',

            name='credit_alert_timestamp',

            field=models.DateTimeField(null=True),

        ),

        migrations.AlterField(

            model_name='ovhsmsgateway',

            name='credit_threshold_alert',

            field=models.PositiveIntegerField(default=500, verbose_name='Credit alert threshold'),

        ),

    ]

from datetime import timedelta

from django.contrib.postgres.fields import ArrayField

from django.conf import settings

from django.core.mail import send_mail

from django.template.loader import render_to_string

from django.utils import timezone

                                                         default=500)

    alert_emails = ArrayField(

        models.EmailField(blank=True),

        blank=True,

        null=True,

        verbose_name=_('Email addresses to send credit alerts to'),

    )

    credit_alert_timestamp = models.DateTimeField(null=True)

        self.send_credit_alert_if_needed()

    def send_credit_alert_if_needed(self):

        if self.credit_left >= self.credit_threshold_alert:

            return

        if self.credit_alert_timestamp and self.credit_alert_timestamp > timezone.now() - timedelta(days=1):

            return  # alerts are sent daily

        ctx = {

            'connector': self,

            'connector_url': urljoin(settings.SITE_BASE_URL, self.get_absolute_url()),

        }

        subject = render_to_string('ovh/credit_alert_subject.txt', ctx).strip()

        body = render_to_string('ovh/credit_alert_body.txt', ctx)

        html_body = render_to_string('ovh/credit_alert_body.html', ctx)

        send_mail(

            _('OVH SMS alert: only %s credits left') % self.credit_left,

            body,

            settings.DEFAULT_FROM_EMAIL,

            self.alert_emails,

            html_message=html_body,

        )

        self.credit_alert_timestamp = timezone.now()

        self.save()

        self.logger.warning('credit is too low, alerts were sent to %s', self.alert_emails)

        if self.uses_new_api:

            self.update_credit_left()

            self.send_credit_alert_if_needed()

        if update_credit and self.uses_new_api:

{% extends "emails/body_base.html" %}

{% load i18n %}

{% block content %}

<p>{% trans "Hi," %}</p>

<p>

{% blocktrans trimmed with name=connector.title credit_left=connector.credit_left %}

There are only {{ credit_left }} credits left for connector {{ name }}.

{% endblocktrans %}

</p>

<p>

{% blocktrans trimmed with account=connector.account %}

Please add more credit as soon as possible for OVH account {{ account }}.

{% endblocktrans %}

</p>

{% with _("View connector page") as button_label %}

{% include "emails/button-link.html" with url=connector_url label=button_label %}

{% endwith %}

{% endblock %}

{% extends "emails/body_base.txt" %}

{% load i18n %}

{% block content %}{% autoescape off %}{% trans "Hi," %}

{% blocktrans trimmed with name=connector.title credit_left=connector.credit_left %}

There are only {{ credit_left }} credits left for connector {{ name }}.

{% endblocktrans %}

{% blocktrans trimmed with account=connector.account %}

Please add more credit as soon as possible for OVH account {{ account }}.

{% endblocktrans %}

{% trans "View connector page:" %} {{ connector_url }}

{% endautoescape %}

{% endblock %}

{% extends "emails/subject.txt" %}

{% load i18n %}

{% block email-subject %}{% autoescape off %}{% blocktrans trimmed with credit_left=connector.credit_left %}

OVH SMS alert: only {{ credit_left }} credits left

{% endblocktrans %}{% endautoescape %}{% endblock %}

SITE_BASE_URL = 'http://localhost'

<!DOCTYPE html>

<html>

<head>

  <meta charset="utf-8">

</head>

<body>

  <div style="max-width: 60ex;">

    <div class="content">

      {% block content %}

      {{ content }}

      {% endblock %}

    </div>

  </div>

</body>

</html>

{% block content %}{{ content }}{% endblock %}

<a href="{{ url }}">{{ label }}</a>

{% block email-subject %}{% endblock %}

def test_ovh_alert_emails(app, freezer, mailoutbox):

    connector = OVHSMSGateway.objects.create(

        slug='test-ovh', title='Test OVH', account='sms-test42',

        application_key='RHrTdU2oTsrVC0pu',

        application_secret='CLjtS69tTcPgCKxedeoZlgMSoQGSiXMa',

        consumer_key='iF0zi0MJrbjNcI3hvuvwkhNk8skrigxz',

        credit_threshold_alert=100,

        credit_left=102,

        alert_emails=['test@entrouvert.org'],

    )

    api = ApiUser.objects.create(username='apiuser')

    obj_type = ContentType.objects.get_for_model(connector)

    AccessRight.objects.create(codename='can_send_messages', apiuser=api, resource_type=obj_type,

                               resource_pk=connector.pk)

    payload = {

        'message': 'hello',

        'from': '+33699999999',

        'to': ['+33688888888', '+33677777777'],

    }

    # register job

    freezer.move_to('2019-01-01 00:00:00')

    path = '/%s/%s/send/' % (connector.get_connector_slug(), connector.slug)

    app.post_json(path, params=payload)

    # perform job

    freezer.move_to('2019-01-01 01:00:03')

    resp = {

        'totalCreditsRemoved': 1,

    }

    base_url = connector.API_URL % {'serviceName': 'sms-test42'}

    url = base_url + 'jobs/'

    with utils.mock_url(url, resp, 200) as mocked:

        connector.jobs()

    assert connector.credit_left == 101

    assert len(mailoutbox) == 0

    # register job

    app.post_json(path, params=payload)

    # perform job

    freezer.move_to('2019-01-01 01:00:06')

    resp = {

        'totalCreditsRemoved': 2,

    }

    with utils.mock_url(url, resp, 200) as mocked:

        connector.jobs()

    mail = mailoutbox[0]

    assert mail.recipients() == ['test@entrouvert.org']

    assert mail.subject == 'OVH SMS alert: only 99 credits left'

    for body in (mail.body, mail.alternatives[0][0]):

        assert connector.account in body

        assert connector.title in body

        assert 'http://localhost/ovh/test-ovh/' in body

    mailoutbox.clear()

    # alert should not be send with every SMS

    app.post_json(path, params=payload)

    freezer.move_to('2019-01-01 01:00:09')

    with utils.mock_url(url, resp, 200) as mocked:

        connector.jobs()

    assert len(mailoutbox) == 0

    # alert is sent again daily

    resp = {'creditsLeft': 99}

    ovh_url = connector.API_URL % {'serviceName': 'sms-test42'}

    with utils.mock_url(ovh_url, resp, 200) as mocked:

        connector.hourly()

    assert len(mailoutbox) == 0

    freezer.move_to('2019-01-02 01:00:07')

    with utils.mock_url(ovh_url, resp, 200) as mocked:

        connector.hourly()

    assert len(mailoutbox) == 1