notifications-admin/app/notify_client/api_client.py

from __future__ import unicode_literals
from flask import url_for
from notifications_python_client.notifications import NotificationsAPIClient
from app.utils import BrowsableItem
from app.notify_client import _attach_current_user


class ServiceAPIClient(NotificationsAPIClient):

    # Fudge assert in the super __init__ so
    # we can set those variables later.
    def __init__(self):
        super(ServiceAPIClient, self).__init__("api_url",
                                               "client",
                                               "secret")

    def init_app(self, application):
        self.base_url = application.config['API_HOST_NAME']
        self.client_id = application.config['ADMIN_CLIENT_USER_NAME']
        self.secret = application.config['ADMIN_CLIENT_SECRET']

    def create_service(self, service_name, active, message_limit, restricted, user_id, email_from):
        """
        Create a service and return the json.
        """
        data = {
            "name": service_name,
            "active": active,
            "message_limit": message_limit,
            "user_id": user_id,
            "restricted": restricted,
            "email_from": email_from
        }
        _attach_current_user(data)
        return self.post("/service", data)['data']['id']

    def delete_service(self, service_id):
        """
        Delete a service.
        """
        endpoint = "/service/{0}".format(service_id)
        data = _attach_current_user({})
        return self.delete(endpoint, data)

    def get_service(self, service_id, *params):
        """
        Retrieve a service.
        """
        return self.get(
            '/service/{0}'.format(service_id))

    def get_services(self, *params):
        """
        Retrieve a list of services.
        """
        return self.get('/service', *params)

    def update_service(self,
                       service_id,
                       service_name,
                       active,
                       message_limit,
                       restricted,
                       users,
                       email_from):
        """
        Update a service.
        """
        data = {
            "id": service_id,
            "name": service_name,
            "active": active,
            "message_limit": message_limit,
            "restricted": restricted,
            "users": users,
            "email_from": email_from
        }
        _attach_current_user(data)
        endpoint = "/service/{0}".format(service_id)
        return self.post(endpoint, data)

    def remove_user_from_service(self, service_id, user_id):
        """
        Remove a user from a service
        """
        endpoint = '/service/{service_id}/users/{user_id}'.format(
            service_id=service_id,
            user_id=user_id)
        data = _attach_current_user({})
        return self.delete(endpoint, data)

    def create_service_template(self, name, type_, content, service_id, subject=None):
        """
        Create a service template.
        """
        data = {
            "name": name,
            "template_type": type_,
            "content": content,
            "service": service_id
        }
        if subject:
            data.update({
                'subject': subject
            })
        _attach_current_user(data)
        endpoint = "/service/{0}/template".format(service_id)
        return self.post(endpoint, data)

    def update_service_template(self, id_, name, type_, content, service_id, subject=None):
        """
        Update a service template.
        """
        data = {
            'id': id_,
            'name': name,
            'template_type': type_,
            'content': content,
            'service': service_id
        }
        if subject:
            data.update({
                'subject': subject
            })
        _attach_current_user(data)
        endpoint = "/service/{0}/template/{1}".format(service_id, id_)
        return self.post(endpoint, data)

    def get_service_template(self, service_id, template_id, *params):
        """
        Retrieve a service template.
        """
        endpoint = '/service/{service_id}/template/{template_id}'.format(
            service_id=service_id,
            template_id=template_id)
        return self.get(endpoint, *params)

    def get_service_templates(self, service_id, *params):
        """
        Retrieve all templates for service.
        """
        endpoint = '/service/{service_id}/template'.format(
            service_id=service_id)
        return self.get(endpoint, *params)

    def delete_service_template(self, service_id, template_id):
        """
        Delete a service template.
        """
        endpoint = "/service/{0}/template/{1}".format(service_id, template_id)
        data = _attach_current_user({})
        return self.delete(endpoint)

    def find_all_service_email_from(self, user_id=None):
        resp = self.get_services(user_id)
        return [x['email_from'] for x in resp['data']]

    # Temp access of service history data. Includes service and api key history
    def get_service_history(self, service_id):
        return self.get('/service/{0}/history'.format(service_id))

    def get_service_usage(self, service_id):
        return self.get('/service/{0}/fragment/aggregate_statistics'.format(service_id))


class ServicesBrowsableItem(BrowsableItem):

    @property
    def title(self):
        return self._item['name']

    @property
    def link(self):
        return url_for('main.service_dashboard', service_id=self._item['id'])

    @property
    def destructive(self):
        return False

    @property
    def hint(self):
        return None