Signall for send email

Author: elesbom

dbaas/maintenance/models.py

@@ -2,13 +2,19 @@
 from __future__ import absolute_import, unicode_literals
 import logging
 import simple_audit
-from dateutil import rrule
+from datetime import datetime
+
 from copy import copy
+
 from django.db import models
-from django.core.exceptions import ValidationError
 from django.utils.translation import ugettext_lazy as _
-from dateutil import tz
-from datetime import datetime
+from dateutil import rrule, tz
+from django.db.models.signals import post_save
+from django.db.models.signals import pre_delete
+from django.dispatch import receiver
+from celery.task import control
+
+
 from account.models import Team
 from backup.models import BackupGroup, Snapshot
 from logical.models import Database, Project
@@ -17,13 +23,10 @@
     Offering, EnginePatch)
 from notification.models import TaskHistory
 from util.models import BaseModel
-from django.db.models.signals import post_save
-from django.db.models.signals import pre_delete
-from django.dispatch import receiver
-from celery.task import control
 from maintenance.tasks import execute_scheduled_maintenance
 from .registered_functions.functools import _get_registered_functions
 from .managers import DatabaseMaintenanceTaskManager
+from util.email_notifications import schedule_task_notification
 
 
 LOG = logging.getLogger(__name__)
@@ -945,11 +948,13 @@ def __unicode__(self):
 simple_audit.register(HostMigrate)
 simple_audit.register(DatabaseMigrate)
 simple_audit.register(DatabaseUpgradePatch)
+simple_audit.register(TaskSchedule)
 
 
-#########
-# SIGNALS#
-#########
+#########################################################
+#                       SIGNALS                         #
+#########################################################
+
 
 @receiver(pre_delete, sender=Maintenance)
 def maintenance_pre_delete(sender, **kwargs):
@@ -991,3 +996,10 @@ def maintenance_post_save(sender, **kwargs):
 
             maintenance.celery_task_id = task.task_id
             maintenance.save()
+
+
+@receiver(post_save, sender=TaskSchedule)
+def task_schedule_post_save(sender, **kwargs):
+    task = kwargs.get("instance")
+    is_new = kwargs.get("created")
+    schedule_task_notification(task.database, task, is_new)

dbaas/notification/templates/email_extras/schedule_task_notification.html

@@ -0,0 +1,17 @@
+{% if is_new %}
+<h2>The SSL of database {{ database }} is about to expire.</h2><br/>
+That will expire at {{ ssl_expire_at }}.<br/>
+We schedule the update in your maintenance window at {{ scheduled_for }}.<br/>
+If you want change that date go to {{ database_url }}.<br/>
+<br/>
+{% else %}
+The schedule task of database {{ database }} was changed for {{ scheduled_for }}.<br/>
+If you want change that date go to {{ database_url }}<br/>
+<br/>
+{% endif %}
+You are receiving this email because in our records you are in team {{ database.team.name }}.<br/>
+If this is not right, contact the Dbaas system administrators.<br/>
+<br><br><br>
+Regards,<br>
+Dbaas notification robot<br>
+{{domain}}

dbaas/notification/templates/email_extras/schedule_task_notification.txt

@@ -0,0 +1,17 @@
+{% if is_new %}
+The SSL of database {{ database }} is about to expire.
+That will expire at {{ ssl_expire_at }}.
+We schedule the update in your maintenance window at {{ scheduled_for }}.
+If you want change that date go to {{ database_url }}.
+
+{% else %}
+The schedule task of database {{ database }} was changed for {{ scheduled_for }}.
+If you want change that date go to {{ database_url }}
+
+{% endif %}
+You are receiving this email because in our records you are in team {{ database.team.name }}.
+If this is not right, contact the Dbaas system administrators.
+
+Regards,
+Dbaas notification robot
+{{domain}}

dbaas/util/email_notifications.py

@@ -45,8 +45,10 @@ def notify_new_user_creation(user=None):
     LOG.debug("user: %s | addr_from: %s | addr_to: %s" %
               (user, addr_from, addr_to))
     if user and addr_from and addr_to:
-        send_mail_template(subject, template, addr_from, addr_to,
-                           fail_silently=False, attachments=None, context=context)
+        send_mail_template(
+            subject, template, addr_from, addr_to,
+            fail_silently=False, attachments=None, context=context
+        )
     else:
         LOG.warning("could not send email for new user creation")
 
@@ -64,8 +66,10 @@ def notify_team_change_for(user=None):
         context['url'] = domain
         context['teams'] = [team.name for team in user.team_set.all()]
         if user and addr_from and addr_to:
-            send_mail_template(subject, template, addr_from, addr_to,
-                               fail_silently=False, attachments=None, context=context)
+            send_mail_template(
+                subject, template, addr_from, addr_to,
+                fail_silently=False, attachments=None, context=context
+            )
         else:
             LOG.warning("could not send email for team change")
     else:
@@ -149,3 +153,30 @@ def disk_resize_notification(database, new_disk, usage_percentage):
         subject, template, email_from(), email_to(database.team),
         fail_silently=False, attachments=None, context=context
     )
+
+
+def schedule_task_notification(database, scheduled_task, is_new):
+
+    subject = _('[DBaaS] Automatic Task {} for Database {}'.format(
+        'created' if is_new else 'updated',
+        database.name,
+    ))
+
+    template = "schedule_task_notification"
+    domain = get_domain()
+
+    context = {
+        'database': database,
+        'ssl_expire_at': database.infra.earliest_ssl_expire_at,
+        'scheduled_for': scheduled_task.scheduled_for,
+        'database_url': "{}{}".format(domain, reverse(
+            'admin:logical_database_maintenance', kwargs={'id': database.id}
+        )),
+        'is_new': is_new,
+        'domain': domain
+    }
+
+    send_mail_template(
+        subject, template, email_from(), email_to(database.team),
+        fail_silently=False, attachments=None, context=context
+    )