client.py

import logging
import requests

from urllib.parse import urljoin, urlparse

from .model import Operator

logger = logging.getLogger(__name__)


class Endpoints:
    def __init__(self, base_url):
        self.base_url = base_url

    def _prepend_base_url(self, path):
        return urljoin(self.base_url, path)

    def get_operator(self, identity):
        return urljoin(self._prepend_base_url('/operators/'), identity)

    def get_operators(self):
        return self._prepend_base_url('/operators/')


class TopDeskClient:
    def __init__(self, url, headers=None, rewrite_url=None):
        self.urls = Endpoints(url)
        self.rewrite_url = rewrite_url
        self.headers = headers

    def _build_headers(self, headers):
        request_headers = {}
        for h in (self.headers):
            request_headers[h] = self.headers[h]
        for h in (headers or ()):
            request_headers[h] = headers[h]
        return request_headers

    def call(self,
             method_name,
             url,
             headers=None,
             params=None,
             **kwargs):
        logger.debug('Calling %s %s with params=%r',
                     method_name,
                     urlparse(url).path,
                     params)
        r = requests.request(method_name,
                             (url if self.rewrite_url is None
                              else url.replace(*self.rewrite_url)),
                             headers=(None if headers is None else
                                      self._build_headers(headers)),
                             params=params if params is not None else {},
                             **kwargs)
        logger.debug('GOT HTTP %d: %r', r.status_code, r.content)
        r.raise_for_status()
        return r.json()

    def post(self, url, **kwargs):
        return self.call('POST', url, **kwargs)

    def get(self, url, **kwargs):
        return self.call('GET', url, **kwargs)

    def put(self, url, **kwargs):
        return self.call('PUT', url, **kwargs)

    def get_operator(self, identity):
        url = self.urls.get_operator(identity)
        return Operator.from_dict(self.get(url))

    def list_operators(self):
        for x in self.get(self.urls.get_operators()):
            yield Operator.from_dict(x)


def get_client(config):
    return TopDeskClient(**config)