mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-06 11:23:48 -05:00
1d3a4e5043
This removes some code which is duplicative and obscure (ie it’s not very clear why we do `"a" * 73` even though there is a Very Good Reason for doing so).
56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
from app.notify_client import NotifyAdminAPIClient, cache
|
|
|
|
|
|
class TemplateFolderAPIClient(NotifyAdminAPIClient):
|
|
|
|
@cache.delete('service-{service_id}-template-folders')
|
|
def create_template_folder(
|
|
self,
|
|
service_id,
|
|
name,
|
|
parent_id=None
|
|
):
|
|
data = {
|
|
'name': name,
|
|
'parent_id': parent_id
|
|
}
|
|
return self.post('/service/{}/template-folder'.format(service_id), data)['data']['id']
|
|
|
|
@cache.set('service-{service_id}-template-folders')
|
|
def get_template_folders(self, service_id):
|
|
return self.get('/service/{}/template-folder'.format(service_id))['template_folders']
|
|
|
|
@cache.delete('service-{service_id}-template-folders')
|
|
@cache.delete('service-{service_id}-templates')
|
|
def move_to_folder(self, service_id, folder_id, template_ids, folder_ids):
|
|
|
|
if folder_id:
|
|
url = '/service/{}/template-folder/{}/contents'.format(service_id, folder_id)
|
|
else:
|
|
url = '/service/{}/template-folder/contents'.format(service_id)
|
|
|
|
self.post(url, {
|
|
'templates': list(template_ids),
|
|
'folders': list(folder_ids),
|
|
})
|
|
|
|
if template_ids:
|
|
self.redis_client.delete(*map(
|
|
'template-{}-version-None'.format,
|
|
template_ids,
|
|
))
|
|
|
|
@cache.delete('service-{service_id}-template-folders')
|
|
def update_template_folder(self, service_id, template_folder_id, name):
|
|
self.post(
|
|
'/service/{}/template-folder/{}'.format(service_id, template_folder_id),
|
|
{"name": name}
|
|
)
|
|
|
|
@cache.delete('service-{service_id}-template-folders')
|
|
def delete_template_folder(self, service_id, template_folder_id):
|
|
self.delete('/service/{}/template-folder/{}'.format(service_id, template_folder_id), {})
|
|
|
|
|
|
template_folder_api_client = TemplateFolderAPIClient()
|