diff options
Diffstat (limited to 'src/notifier')
-rw-r--r-- | src/notifier/manager.py | 29 | ||||
-rw-r--r-- | src/notifier/models.py | 7 | ||||
-rw-r--r-- | src/notifier/tasks.py | 18 |
3 files changed, 29 insertions, 25 deletions
diff --git a/src/notifier/manager.py b/src/notifier/manager.py index 8ea8021..e2afdec 100644 --- a/src/notifier/manager.py +++ b/src/notifier/manager.py @@ -8,9 +8,8 @@ # http://www.apache.org/licenses/LICENSE-2.0 ############################################################################## import os -from notifier.models import Notification, Emailed +from notifier.models import Notification, Emailed, Email -from django.core.mail import send_mail from django.template.loader import render_to_string from django.utils import timezone @@ -99,14 +98,8 @@ class NotificationHandler(object): # render email template message = render_to_string(template_name, context) - # finally, send the email - send_mail( - "Your Booking is Ready", - message, - os.environ.get("DEFAULT_FROM_EMAIL", "opnfv@laas-dashboard"), - [user.userprofile.email_addr], - fail_silently=False - ) + # finally, queue email for sending + Email.objects.create(title="Your Booking is Ready", message=message, recipient=user.userprofile.email_addr) @classmethod def email_booking_over(cls, booking): @@ -124,13 +117,7 @@ class NotificationHandler(object): message = render_to_string(template_name, context) - send_mail( - "Your Booking has Expired", - message, - os.environ.get("DEFAULT_FROM_EMAIL", "opnfv@laas-dashboard"), - [user.userprofile.email_addr], - fail_silently=False - ) + Email.objects.create(title="Your Booking has Expired", message=message, recipient=user.userprofile.email_addr) @classmethod def email_booking_expiring(cls, booking): @@ -148,13 +135,7 @@ class NotificationHandler(object): message = render_to_string(template_name, context) - send_mail( - "Your Booking is Expiring", - message, - os.environ.get("DEFAULT_FROM_EMAIL", "opnfv@laas-dashboard"), - [user.userprofile.email_addr], - fail_silently=False - ) + Email.objects.create(title="Your Booking is Expiring", message=message, recipient=user.userprofile.email_addr) @classmethod def task_updated(cls, task): diff --git a/src/notifier/models.py b/src/notifier/models.py index 0af748b..03e23b3 100644 --- a/src/notifier/models.py +++ b/src/notifier/models.py @@ -47,3 +47,10 @@ class Emailed(models.Model): on_delete=models.CASCADE, related_name="over_mail" ) + + +class Email(models.Model): + sent = models.BooleanField(default=False) + title = models.CharField(max_length=150) + message = models.TextField() + recipient = models.CharField(max_length=150) diff --git a/src/notifier/tasks.py b/src/notifier/tasks.py index 474d64d..389750a 100644 --- a/src/notifier/tasks.py +++ b/src/notifier/tasks.py @@ -13,8 +13,11 @@ from celery import shared_task from django.utils import timezone from django.conf import settings from booking.models import Booking -from notifier.models import Emailed +from notifier.models import Emailed, Email from notifier.manager import NotificationHandler +from django.core.mail import send_mail + +import os @shared_task @@ -33,3 +36,16 @@ def notify_expiring(): continue NotificationHandler.notify_booking_expiring(booking) Emailed.objects.create(almost_end_booking=booking) + + +@shared_task +def dispatch_emails(): + for email in Email.objects.filter(sent=False): + email.sent = True + email.save() + send_mail( + email.title, + email.message, + os.environ.get("DEFAULT_FROM_EMAIL", "opnfv@laas-dashboard"), + email.recipient, + fail_silently=False) |