Use the letter_branding logo if it exists otherwise fall back to the dvla_organisation logo.

This commit is contained in:
Rebecca Law
2019-01-23 12:51:09 +00:00
parent f11aa55e0b
commit e4ea208d06
3 changed files with 45 additions and 5 deletions

View File

@@ -52,11 +52,13 @@ from app.models import (
def create_letters_pdf(self, notification_id):
try:
notification = get_notification_by_id(notification_id, _raise=True)
# We only need this while we are migrating to the new letter_branding model
letter_logo_file_name = notification.service.letter_branding.filename if notification.service.letter_branding\
else notification.service.dvla_organisation.filename
pdf_data, billable_units = get_letters_pdf(
notification.template,
contact_block=notification.reply_to_text,
filename=notification.service.dvla_organisation.filename,
filename=letter_logo_file_name,
values=notification.personalisation
)