Add brand type to email branding.

Removed banner_colour and single_id_colour.
We only really need one colour now.

app/main/forms.py

@@ -728,6 +728,16 @@ class ServiceSelectEmailBranding(StripWhitespaceForm):
 
 
 class ServiceUpdateEmailBranding(StripWhitespaceForm):
+    def __init__(self, *args, **kwargs):
+
+        super().__init__(*args, **kwargs)
+
+        self.brand_type.choices = filter(None, [
+            ('govuk', 'GOV.UK only'),
+            ('both', 'GOV.UK and branding'),
+            ('org', 'Branding only'),
+            ('org_banner', 'Branding banner'),
+        ])
 
     name = StringField('Name of brand')
     text = StringField('Text')
@@ -738,19 +748,10 @@ class ServiceUpdateEmailBranding(StripWhitespaceForm):
             Regexp(regex="^$|^#(?:[0-9a-fA-F]{3}){1,2}$", message='Must be a valid color hex code')
         ]
     )
-    banner_colour = StringField(
-        'Banner colour',
-        validators=[
-            Regexp(regex="^$|^#(?:[0-9a-fA-F]{3}){1,2}$", message='Must be a valid color hex code')
-        ]
-    )
-    single_id_colour = StringField(
-        'Single identity colour',
-        validators=[
-            Regexp(regex="^$|^#(?:[0-9a-fA-F]{3}){1,2}$", message='Must be a valid color hex code')
-        ]
-    )
     file = FileField_wtf('Upload a PNG logo', validators=[FileAllowed(['png'], 'PNG Images only!')])
+    brand_type = RadioField(
+        "Brand type"
+    )
 
 
 class CreateOrUpdateOrganisation(StripWhitespaceForm):

app/main/views/email_branding.py

@@ -51,7 +51,13 @@ def email_branding():
 def update_email_branding(branding_id, logo=None):
     email_branding = email_branding_client.get_email_branding(branding_id)['email_branding']
 
-    form = ServiceUpdateEmailBranding()
+    form = ServiceUpdateEmailBranding(
+        name=email_branding['name'],
+        text=email_branding['text'],
+        colour=email_branding['colour'],
+        domain=email_branding['domain'],
+        brand_type=email_branding['brand_type']
+    )
 
     logo = logo if logo else email_branding.get('logo') if email_branding else None
 
@@ -80,20 +86,12 @@ def update_email_branding(branding_id, logo=None):
             name=form.name.data,
             text=form.text.data,
             colour=form.colour.data,
-            banner_colour=form.banner_colour.data,
-            single_id_colour=form.single_id_colour.data,
             domain=form.domain.data,
+            brand_type=form.brand_type.data,
         )
 
         return redirect(url_for('.email_branding', branding_id=branding_id))
 
-    form.name.data = email_branding['name']
-    form.text.data = email_branding['text']
-    form.colour.data = email_branding['colour']
-    form.banner_colour.data = email_branding['banner_colour']
-    form.single_id_colour.data = email_branding['single_id_colour']
-    form.domain.data = email_branding['domain']
-
     return render_template(
         'views/email-branding/manage-branding.html',
         form=form,
@@ -134,9 +132,8 @@ def create_email_branding(logo=None):
             name=form.name.data,
             text=form.text.data,
             colour=form.colour.data,
-            banner_colour=form.banner_colour.data,
-            single_id_colour=form.single_id_colour.data,
-            domain=form.domain.data
+            domain=form.domain.data,
+            brand_type=form.brand_type.data,
         )
 
         return redirect(url_for('.email_branding'))

app/notify_client/email_branding_client.py

@@ -18,26 +18,24 @@ class EmailBrandingClient(NotifyAdminAPIClient):
     def get_letter_email_branding(self):
         return self.get(url='/dvla_organisations')
 
-    def create_email_branding(self, logo, name, text, colour, banner_colour, single_id_colour, domain):
+    def create_email_branding(self, logo, name, text, colour, domain, brand_type):
         data = {
             "logo": logo,
             "name": name,
             "text": text,
             "colour": colour,
-            "banner_colour": banner_colour,
-            "single_id_colour": single_id_colour,
             "domain": domain,
+            "brand_type": brand_type
         }
         return self.post(url="/email-branding", data=data)
 
-    def update_email_branding(self, branding_id, logo, name, text, colour, banner_colour, single_id_colour, domain):
+    def update_email_branding(self, branding_id, logo, name, text, colour, domain, brand_type):
         data = {
             "logo": logo,
             "name": name,
             "text": text,
             "colour": colour,
-            "banner_colour": banner_colour,
-            "single_id_colour": single_id_colour,
             "domain": domain,
+            "brand_type": brand_type
         }
         return self.post(url="/email-branding/{}".format(branding_id), data=data)

app/templates/views/email-branding/manage-branding.html

@@ -2,6 +2,7 @@
 {% from "components/file-upload.html" import file_upload %}
 {% from "components/page-footer.html" import page_footer %}
 {% from "components/textbox.html" import textbox %}
+{% from "components/radios.html" import radios %}
 
 {% block service_page_title %}
   {{ '{} email branding'.format('Update' if email_branding else 'Create')}}
@@ -23,9 +24,8 @@
           <div style='margin-top:15px;'>{{textbox(form.name)}}</div>
           <div style='margin-top:15px;'>{{textbox(form.text)}}</div>
           {{ textbox(form.colour, width='1-4', colour_preview=True) }}
-          {{ textbox(form.banner_colour, width='1-4', colour_preview=True) }}
-          {{ textbox(form.single_id_colour, width='1-4', colour_preview=True) }}
            <div style='margin-top:15px;'>{{textbox(form.domain)}}</div>
+           {{ radios(form.brand_type) }}
           {{ page_footer(
             'Save',
             back_link=url_for('.email_branding'),