diff --git a/app/schemas.py b/app/schemas.py index 5e020ddf1..b4d34ab62 100644 --- a/app/schemas.py +++ b/app/schemas.py @@ -314,9 +314,14 @@ class NotificationModelSchema(BaseSchema): class BaseTemplateSchema(BaseSchema): + reply_to = fields.Method("get_reply_to", allow_none=True) + + def get_reply_to(self, template): + return template.reply_to + class Meta: model = models.Template - exclude = ("service_id", "jobs") + exclude = ("service_id", "jobs", "service_letter_contact_id") strict = True @@ -339,9 +344,14 @@ class TemplateSchema(BaseTemplateSchema): class TemplateHistorySchema(BaseSchema): + reply_to = fields.Method("get_reply_to", allow_none=True) + created_by = fields.Nested(UserSchema, only=['id', 'name', 'email_address'], dump_only=True) created_at = field_for(models.Template, 'created_at', format='%Y-%m-%d %H:%M:%S.%f') + def get_reply_to(self, template): + return template.reply_to + class Meta: model = models.TemplateHistory