from notifications_python_client.base import BaseAPIClient


class NotificationApiClient(BaseAPIClient):
    def __init__(self, base_url=None, client_id=None, secret=None):
        super(self.__class__, self).__init__(base_url=base_url or 'base_url',
                                             client_id=client_id or 'client_id',
                                             secret=secret or 'secret')

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

    def get_all_notifications(self, page=None):
        params = {}
        if page is not None:
            params['page'] = page
        return self.get(
            url='/notifications',
            params=params
        )

    def get_notifications_for_service(self, service_id, job_id=None, page=None):
        params = {}
        if page is not None:
            params['page'] = page
        if job_id:
            return self.get(
                url='/service/{}/job/{}/notifications'.format(service_id, job_id),
                params=params
            )
        else:
            return self.get(
                url='/service/{}/notifications'.format(service_id),
                params=params
            )