<reponame>raigad/simple-stripe-client
import requests
import logging
import json

try:
    # python 3
    import http.client as http_client
    from urllib.parse import urlparse, urlunparse, urlencode
except ImportError:
    # python 2
    import httplib as http_client
    from urlparse import urlparse, urlunparse
    from urllib import urlencode

API_BASE_URL = 'https://api.stripe.com'
API_VERSION = 'v1'


class Api(object):
    def __init__(self, api_key, debug_http=False, timeout=None, base_url=None, api_version=None):
        self.api_key = api_key
        self._debug_http = debug_http
        self._timeout = timeout
        self.url = ''

        if base_url:
            self.base_url = base_url
        else:
            self.base_url = API_BASE_URL

        if api_version:
            self.base_url = '/'.join([self.base_url, api_version])
        else:
            self.base_url = '/'.join([self.base_url, API_VERSION])

        if debug_http:
            http_client.HTTPConnection.debuglevel = 1
            logging.basicConfig()  # initialize logging
            logging.getLogger().setLevel(logging.DEBUG)
            requests_log = logging.getLogger("requests.packages.urllib3")
            requests_log.setLevel(logging.DEBUG)
            requests_log.propagate = True

        self._session = requests.Session()
        self._session.headers.update({'Authorization': 'Bearer ' + self.api_key})

    def __getattr__(self, attribute):
        """
        Intercepts every call to attribute that does not exists

        """
        self.url = '/'.join([self.url, attribute])
        return self

    def __repr__(self):
        return "{cls}(api_key={api_key},url={url})".format(cls=self.__class__, api_key=self.api_key, url=self.url)

    def id(self, value):
        self.url = '/'.join([self.url, value])
        return self

    def get(self, **kwargs):
        return self._make_request('GET', self.get_and_reset_url(), data=kwargs)

    def post(self, **kwargs):
        return self._make_request('POST', self.get_and_reset_url(), data=kwargs)

    def put(self, **kwargs):
        return self._make_request('PUT', self.get_and_reset_url(), data=kwargs)

    def delete(self):
        return self._make_request('DELETE', self.get_and_reset_url())

    def get_and_reset_url(self):
        url = self.url
        self.url = ''
        return url

    def _make_request(self, http_method, url, data=None, json=None):
        """
        :param http_method: http method i.e. GET, POST, PUT etc
        :param url: api endpoint url
        :param data: dictionary of key/value params
        :param json: json data
        :return:
        """
        if not data:
            data = {}

        data = self.__class__._prepare_nested_data(data)

        response = {}
        if http_method == 'GET':
            response = self._session.get(self._build_url(url, extra_params=data), timeout=self._timeout)
        elif http_method == 'POST':
            response = self._session.post(self._build_url(url), data=data, timeout=self._timeout)
        elif http_method == 'PUT':
            response = self._session.put(self._build_url(url), data=data, timeout=self._timeout)
        elif http_method == 'DELETE':
            response = self._session.delete(self._build_url(url), timeout=self._timeout)

        try:
            response = self.__class__._parse_json_data(response.content.decode('utf-8'))
        except AttributeError as e:
            response = {"Error": "Unknown error while parsing response"}

        return response

    def _build_url(self, url, path_elements=None, extra_params=None):
        url = self.base_url + url
        (scheme, netloc, path, params, query, fragment) = urlparse(url)

        # Add extra_parameters to the query
        if extra_params and len(extra_params) > 0:
            extra_query = self.__class__._encode_parameters(extra_params)
            if query:
                query += '&' + extra_query
            else:
                query = extra_query
        return urlunparse((scheme, netloc, path, params, query, fragment))

    @staticmethod
    def _encode_parameters(parameters):
        """
        Return url encoded string in 'key=value&key=value' format
        """
        if not parameters or not isinstance(parameters, dict):
            return None
        return urlencode(parameters)

    @staticmethod
    def _parse_json_data(json_data):
        try:
            data = json.loads(json_data)
        except ValueError:
            data = json.loads('{"Error": "Unknown error while parsing response"}')
        return data

    @staticmethod
    def _prepare_nested_data(data):
        """
        Converted nested dict into one key=value dict
        { 'A' :  65, 'B' : 66, 'nested' : { 'C' : 67 , 'D' : 68  }  }
        To
        { 'A' :  65, 'B' : 66, 'nested[C]' :  67 , 'nested[D]' : 68  }

        """
        processed_data = {}
        for key, value in data.items():
            if isinstance(value, dict):
                nested_data = {}
                for k, v in value.items():
                    nested_key = "{key}[{k}]".format(key=key, k=k)
                    nested_data[nested_key] = v
                processed_data.update(Api._prepare_nested_data(nested_data))
            else:
                processed_data.update({key: value})
        return processed_data
