Files
atst/atst/utils/notification_sender.py
2019-10-31 13:46:04 -04:00

18 lines
558 B
Python

from sqlalchemy import select
from atst.jobs import send_notification_mail
from atst.database import db
from atst.models.notification_recipient import NotificationRecipient
class NotificationSender(object):
EMAIL_SUBJECT = "ATST notification"
def send(self, body, type_=None):
recipients = self._get_recipients(type_)
send_notification_mail.delay(recipients, self.EMAIL_SUBJECT, body)
def _get_recipients(self, type_):
query = select([NotificationRecipient.email])
return db.session.execute(query).fetchone()