mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 23:41:17 -05:00
2591d3a1df
Lots of changes, all changing the line of code that puts things into queues, and the code that tests that.
35 lines
1.5 KiB
Python
35 lines
1.5 KiB
Python
from flask import current_app
|
|
|
|
from app.config import QueueNames
|
|
from app.dao.services_dao import dao_fetch_service_by_id, dao_fetch_active_users_for_service
|
|
from app.dao.templates_dao import dao_get_template_by_id
|
|
from app.models import EMAIL_TYPE, KEY_TYPE_NORMAL
|
|
from app.notifications.process_notifications import persist_notification, send_notification_to_queue
|
|
|
|
|
|
def send_notification_to_service_users(service_id, template_id, personalisation={}, include_user_fields=[]):
|
|
template = dao_get_template_by_id(template_id)
|
|
service = dao_fetch_service_by_id(service_id)
|
|
active_users = dao_fetch_active_users_for_service(service.id)
|
|
notify_service = dao_fetch_service_by_id(current_app.config['NOTIFY_SERVICE_ID'])
|
|
|
|
for user in active_users:
|
|
personalisation = _add_user_fields(user, personalisation, include_user_fields)
|
|
notification = persist_notification(
|
|
template_id=template.id,
|
|
template_version=template.version,
|
|
recipient=user.email_address if template.template_type == EMAIL_TYPE else user.mobile_number,
|
|
service=notify_service,
|
|
personalisation=personalisation,
|
|
notification_type=template.template_type,
|
|
api_key_id=None,
|
|
key_type=KEY_TYPE_NORMAL
|
|
)
|
|
send_notification_to_queue(notification, False, queue=QueueNames.NOTIFY)
|
|
|
|
|
|
def _add_user_fields(user, personalisation, fields):
|
|
for field in fields:
|
|
personalisation[field] = getattr(user, field)
|
|
return personalisation
|