notifications-admin/app/utils/branding.py

from app.models.organization import Organization

NHS_EMAIL_BRANDING_ID = 'a7dc4e56-660b-4db7-8cff-12c37b12b5ea'


def get_email_choices(service):
    organization_branding_id = service.organization.email_branding_id if service.organization else None

    if (
        service.organization_type == Organization.TYPE_FEDERAL
        and service.email_branding_id is not None  # GOV.UK is not current branding
        and organization_branding_id is None  # no default to supersede it (GOV.UK)
    ):
        yield ('govuk', 'GOV.UK')

    if (
        service.organization_type == Organization.TYPE_FEDERAL
        and service.organization
        and organization_branding_id is None  # don't offer both if org has default
        and service.email_branding_name.lower() != f'GOV.UK and {service.organization.name}'.lower()
    ):
        yield ('govuk_and_org', f'GOV.UK and {service.organization.name}')