Add redact_personalisation to TemplateSchema

This commit is contained in:
Rebecca Law
2017-06-29 18:16:03 +01:00
parent 362652ec47
commit 49917ccafd
3 changed files with 19 additions and 8 deletions

View File

@@ -297,15 +297,14 @@ class BaseTemplateSchema(BaseSchema):
strict = True
class TemplateRedactedSchema(BaseSchema):
class Meta:
model = models.TemplateRedacted
class TemplateSchema(BaseTemplateSchema):
created_by = field_for(models.Template, 'created_by', required=True)
process_type = field_for(models.Template, 'process_type')
redact_personalisation = fields.Method("redact")
def redact(self, template):
return template.redact_personalisation
@validates_schema
def validate_type(self, data):