tiingo-python/tiingo/restclient.py

# -*- coding: utf-8 -*-

import logging
import requests
from requests.exceptions import HTTPError


# TODO: Possibly print HTTP json response if available?
class RestClientError(Exception):
    "Wrapper around HTTP Errors"
    pass


class RestClient(object):
    def __init__(self, config={}):
        """Base class for interacting with RESTful APIs
        Child class MUST have a ._base_url property!

         Args:
             config (dict): Arbitrary options that child classes can access
        """
        self._config = config

        # The child class should override these properties or else the
        # restclient won't work. Reevalute whether to do this as an abstract
        # base class so it doesn't get used by itself.
        self._headers = {}
        self._base_url = ""

        if config.get("session"):
            self._session = requests.Session()
        else:
            self._session = requests

    def __repr__(self):
        return '<RestClient(url="{}")>'.format(self._base_url)

    def _request(self, method, url, **kwargs):
        """Make HTTP request and return response object

        Args:
            method (str): GET, POST, PUT, DELETE
            url (str): path appended to the base_url to create request
            **kwargs: passed directly to a requests.request object
        """
        resp = self._session.request(
            method, "{}/{}".format(self._base_url, url), headers=self._headers, **kwargs
        )

        try:
            resp.raise_for_status()
        except HTTPError as e:
            logging.error(resp.content)
            raise RestClientError(e)

        return resp