Adding new data models for letter branding.

This commit is contained in:
Rebecca Law
2019-01-22 17:27:00 +00:00
parent 52a1b534ee
commit f11aa55e0b
10 changed files with 204 additions and 21 deletions

View File

@@ -213,7 +213,8 @@ class ServiceSchema(BaseSchema):
letter_contact_block = fields.Method(serialize="get_letter_contact")
def get_letter_logo_filename(self, service):
return service.dvla_organisation.filename
return service.letter_branding.filename if service.letter_branding\
else service.dvla_organisation.filename
def service_permissions(self, service):
return [p.permission for p in service.permissions]