Merge branch 'master' of github.com:alphagov/notifications-api into return-service_sms_sender_value

This commit is contained in:
Rebecca Law
2017-10-02 12:36:46 +01:00
7 changed files with 452 additions and 9 deletions

View File

@@ -18,6 +18,11 @@ from notifications_utils.recipients import (
InvalidEmailError
)
from notifications_utils.letter_timings import get_letter_timings
from notifications_utils.template import (
PlainTextEmailTemplate,
SMSMessageTemplate,
LetterDVLATemplate,
)
from app.encryption import (
hashpw,
@@ -531,6 +536,22 @@ class Template(db.Model):
_external=True
)
def _as_utils_template(self):
if self.template_type == EMAIL_TYPE:
return PlainTextEmailTemplate(
{'content': self.content, 'subject': self.subject}
)
if self.template_type == SMS_TYPE:
return SMSMessageTemplate(
{'content': self.content}
)
if self.template_type == LETTER_TYPE:
return LetterDVLATemplate(
{'content': self.content, 'subject': self.subject},
notification_reference=1,
contact_block=self.service.letter_contact_block,
)
def serialize(self):
serialized = {
"id": str(self.id),
@@ -542,6 +563,12 @@ class Template(db.Model):
"body": self.content,
"subject": self.subject if self.template_type != SMS_TYPE else None,
"name": self.name,
"personalisation": {
key: {
'required': True,
}
for key in self._as_utils_template().placeholders
},
}
return serialized
@@ -582,6 +609,9 @@ class TemplateHistory(db.Model):
nullable=False,
default=NORMAL)
def _as_utils_template(self):
return Template._as_utils_template(self)
def serialize(self):
return Template.serialize(self)