From 49cfe56edd6d25fafd7ca6a6604731cb38ace990 Mon Sep 17 00:00:00 2001 From: Ed Woodward Date: Wed, 7 Jul 2021 14:13:50 -0500 Subject: [PATCH] Partner types only returned if they have a partner --- pages/models.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/pages/models.py b/pages/models.py index 8734b794c..3022d1b5f 100644 --- a/pages/models.py +++ b/pages/models.py @@ -2253,7 +2253,14 @@ def field_name_mapping(): @staticmethod def partner_type_choices(): - return [x.display_name for x in PartnerTypeMapping.objects.all()] + partner_types_array = [] + partner_type_mappings = PartnerTypeMapping.objects.all() + types_from_partners = Partner.objects.values_list('partner_type', flat=True).exclude(partner_type__isnull=True) + for partner_type in partner_type_mappings: + if any(p_type.lower().startswith(partner_type.display_name.lower()) for p_type in types_from_partners): + partner_types_array.append(partner_type.display_name) + + return partner_types_array content_panels = [ FieldPanel('title', classname='full title', help_text="Internal name for page."),