mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-09-11 10:28:41 -04:00
notify-api-412 use black to enforce python coding style
This commit is contained in:
@@ -2,38 +2,37 @@ from app.notify_client import NotifyAdminAPIClient, cache
|
||||
|
||||
|
||||
class EmailBrandingClient(NotifyAdminAPIClient):
|
||||
|
||||
@cache.set('email_branding-{branding_id}')
|
||||
@cache.set("email_branding-{branding_id}")
|
||||
def get_email_branding(self, branding_id):
|
||||
return self.get(url='/email-branding/{}'.format(branding_id))
|
||||
return self.get(url="/email-branding/{}".format(branding_id))
|
||||
|
||||
@cache.set('email_branding')
|
||||
@cache.set("email_branding")
|
||||
def get_all_email_branding(self, sort_key=None):
|
||||
brandings = self.get(url='/email-branding')['email_branding']
|
||||
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
|
||||
|
||||
@cache.delete('email_branding')
|
||||
@cache.delete("email_branding")
|
||||
def create_email_branding(self, logo, name, text, colour, brand_type):
|
||||
data = {
|
||||
"logo": logo,
|
||||
"name": name,
|
||||
"text": text,
|
||||
"colour": colour,
|
||||
"brand_type": brand_type
|
||||
"brand_type": brand_type,
|
||||
}
|
||||
return self.post(url="/email-branding", data=data)
|
||||
|
||||
@cache.delete('email_branding')
|
||||
@cache.delete('email_branding-{branding_id}')
|
||||
@cache.delete("email_branding")
|
||||
@cache.delete("email_branding-{branding_id}")
|
||||
def update_email_branding(self, branding_id, logo, name, text, colour, brand_type):
|
||||
data = {
|
||||
"logo": logo,
|
||||
"name": name,
|
||||
"text": text,
|
||||
"colour": colour,
|
||||
"brand_type": brand_type
|
||||
"brand_type": brand_type,
|
||||
}
|
||||
return self.post(url="/email-branding/{}".format(branding_id), data=data)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user