mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-02 04:39:25 -04:00
3fda171f80
Added banner_colour, single_id_colour and domain to email branding view. Now able to set the fields. However, the new fields are not being used yet.
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
from app.notify_client import NotifyAdminAPIClient
|
|
|
|
|
|
class EmailBrandingClient(NotifyAdminAPIClient):
|
|
|
|
def __init__(self):
|
|
super().__init__("a" * 73, "b")
|
|
|
|
def get_email_branding(self, branding_id):
|
|
return self.get(url='/email-branding/{}'.format(branding_id))
|
|
|
|
def get_all_email_branding(self, sort_key=None):
|
|
brandings = self.get(url='/email-branding')['email_branding']
|
|
if sort_key and sort_key in brandings[0]:
|
|
brandings.sort(key=lambda branding: branding[sort_key].lower())
|
|
return brandings
|
|
|
|
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):
|
|
data = {
|
|
"logo": logo,
|
|
"name": name,
|
|
"text": text,
|
|
"colour": colour,
|
|
"banner_colour": banner_colour,
|
|
"single_id_colour": single_id_colour,
|
|
"domain": domain,
|
|
}
|
|
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):
|
|
data = {
|
|
"logo": logo,
|
|
"name": name,
|
|
"text": text,
|
|
"colour": colour,
|
|
"banner_colour": banner_colour,
|
|
"single_id_colour": single_id_colour,
|
|
"domain": domain,
|
|
}
|
|
return self.post(url="/email-branding/{}".format(branding_id), data=data)
|