import uuid from datetime import datetime from flask import current_app from gds_metrics import Histogram from notifications_utils.clients import redis from notifications_utils.recipients import ( format_email_address, get_international_phone_info, validate_and_format_phone_number, ) from notifications_utils.template import ( LetterPrintTemplate, PlainTextEmailTemplate, SMSMessageTemplate, ) from app import redis_store from app.celery import provider_tasks from app.celery.letters_pdf_tasks import get_pdf_for_templated_letter from app.config import QueueNames from app.dao.notifications_dao import ( dao_create_notification, dao_delete_notifications_by_id, ) from app.models import ( EMAIL_TYPE, INTERNATIONAL_POSTAGE_TYPES, KEY_TYPE_TEST, LETTER_TYPE, NOTIFICATION_CREATED, SMS_TYPE, Notification, ) from app.v2.errors import BadRequestError REDIS_GET_AND_INCR_DAILY_LIMIT_DURATION_SECONDS = Histogram( 'redis_get_and_incr_daily_limit_duration_seconds', 'Time taken to get and possibly incremement the daily limit cache key', ) def create_content_for_notification(template, personalisation): if template.template_type == EMAIL_TYPE: template_object = PlainTextEmailTemplate( { 'content': template.content, 'subject': template.subject, 'template_type': template.template_type, }, personalisation, ) if template.template_type == SMS_TYPE: template_object = SMSMessageTemplate( { 'content': template.content, 'template_type': template.template_type, }, personalisation, ) if template.template_type == LETTER_TYPE: template_object = LetterPrintTemplate( { 'content': template.content, 'subject': template.subject, 'template_type': template.template_type, }, personalisation, contact_block=template.reply_to_text, ) check_placeholders(template_object) return template_object def check_placeholders(template_object): if template_object.missing_data: message = 'Missing personalisation: {}'.format(", ".join(template_object.missing_data)) raise BadRequestError(fields=[{'template': message}], message=message) def persist_notification( *, template_id, template_version, recipient, service, personalisation, notification_type, api_key_id, key_type, created_at=None, job_id=None, job_row_number=None, reference=None, client_reference=None, notification_id=None, simulated=False, created_by_id=None, status=NOTIFICATION_CREATED, reply_to_text=None, billable_units=None, postage=None, document_download_count=None, updated_at=None ): notification_created_at = created_at or datetime.utcnow() if not notification_id: notification_id = uuid.uuid4() notification = Notification( id=notification_id, template_id=template_id, template_version=template_version, to=recipient, service_id=service.id, personalisation=personalisation, notification_type=notification_type, api_key_id=api_key_id, key_type=key_type, created_at=notification_created_at, job_id=job_id, job_row_number=job_row_number, client_reference=client_reference, reference=reference, created_by_id=created_by_id, status=status, reply_to_text=reply_to_text, billable_units=billable_units, document_download_count=document_download_count, updated_at=updated_at ) if notification_type == SMS_TYPE: formatted_recipient = validate_and_format_phone_number(recipient, international=True) recipient_info = get_international_phone_info(formatted_recipient) notification.normalised_to = formatted_recipient notification.international = recipient_info.international notification.phone_prefix = recipient_info.country_prefix notification.rate_multiplier = recipient_info.billable_units elif notification_type == EMAIL_TYPE: notification.normalised_to = format_email_address(notification.to) elif notification_type == LETTER_TYPE: notification.postage = postage notification.international = postage in INTERNATIONAL_POSTAGE_TYPES notification.normalised_to = ''.join(notification.to.split()).lower() # if simulated create a Notification model to return but do not persist the Notification to the dB if not simulated: dao_create_notification(notification) # Only keep track of the daily limit for trial mode services. if service.restricted and key_type != KEY_TYPE_TEST: if redis_store.get(redis.daily_limit_cache_key(service.id)): redis_store.incr(redis.daily_limit_cache_key(service.id)) current_app.logger.info( "{} {} created at {}".format(notification_type, notification_id, notification_created_at) ) return notification def send_notification_to_queue_detached( key_type, notification_type, notification_id, research_mode, queue=None ): if research_mode or key_type == KEY_TYPE_TEST: queue = QueueNames.RESEARCH_MODE if notification_type == SMS_TYPE: if not queue: queue = QueueNames.SEND_SMS deliver_task = provider_tasks.deliver_sms if notification_type == EMAIL_TYPE: if not queue: queue = QueueNames.SEND_EMAIL deliver_task = provider_tasks.deliver_email if notification_type == LETTER_TYPE: if not queue: queue = QueueNames.CREATE_LETTERS_PDF deliver_task = get_pdf_for_templated_letter try: deliver_task.apply_async([str(notification_id)], queue=queue) except Exception: dao_delete_notifications_by_id(notification_id) raise current_app.logger.debug( "{} {} sent to the {} queue for delivery".format(notification_type, notification_id, queue)) def send_notification_to_queue(notification, research_mode, queue=None): send_notification_to_queue_detached( notification.key_type, notification.notification_type, notification.id, research_mode, queue ) def simulated_recipient(to_address, notification_type): if notification_type == SMS_TYPE: formatted_simulated_numbers = [ validate_and_format_phone_number(number) for number in current_app.config['SIMULATED_SMS_NUMBERS'] ] return to_address in formatted_simulated_numbers else: return to_address in current_app.config['SIMULATED_EMAIL_ADDRESSES']