add actual_template relationship to notification

also renamed the function to make it apparent that it'll join and grab personalisation
This commit is contained in:
Leo Hemsted
2016-08-09 13:07:48 +01:00
parent c820938ced
commit 46c0728b12
6 changed files with 43 additions and 22 deletions

View File

@@ -8,6 +8,7 @@ from datetime import (
from flask import current_app
from werkzeug.datastructures import MultiDict
from sqlalchemy import (desc, func, Integer, or_, and_, asc)
from sqlalchemy.orm import joinedload
from sqlalchemy.sql.expression import cast
from notifications_utils.template import get_sms_fragment_count
@@ -351,12 +352,12 @@ def get_notifications_for_job(service_id, job_id, filter_dict=None, page=1, page
@statsd(namespace="dao")
def get_notification(service_id, notification_id, key_type=None):
def get_notification_with_personalisation(service_id, notification_id, key_type):
filter_dict = {'service_id': service_id, 'id': notification_id}
if key_type:
filter_dict['key_type'] = key_type
return Notification.query.filter_by(**filter_dict).one()
return Notification.query.filter_by(**filter_dict).options(joinedload('actual_template')).one()
@statsd(namespace="dao")
@@ -374,7 +375,8 @@ def get_notifications_for_service(service_id,
page=1,
page_size=None,
limit_days=None,
key_type=None):
key_type=None,
personalisation=False):
if page_size is None:
page_size = current_app.config['PAGE_SIZE']
filters = [Notification.service_id == service_id]
@@ -388,6 +390,10 @@ def get_notifications_for_service(service_id,
query = Notification.query.filter(*filters)
query = _filter_query(query, filter_dict)
if personalisation:
query.options(
joinedload('actual_template')
)
return query.order_by(desc(Notification.created_at)).paginate(
page=page,
per_page=page_size