mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 19:03:30 -05:00
f6367f2278
when clients are defined in app/__init__.py, it increases the chance of cyclical imports. By moving module level client singletons out to a separate extensions file, we stop cyclical imports, but keep the same code flow - the clients are still initialised in `create_app` in `__init__.py`. The redis client in particular is no longer separate - previously redis was set up on the `NotifyAdminAPIClient` base class, but now there's one singleton in `app.extensions`. This was done so that we can access redis from outside of the existing clients.
57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
from app.extensions import redis_client
|
|
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:
|
|
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()
|